aboutsummaryrefslogtreecommitdiff
path: root/wqflask/tests/api/test_gen_menu.py
blob: 0379e297df86d0565de3bc99e446f75c81dfb067 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
"""Test cases for wqflask.api.gen_menu"""
import unittest
import mock

from wqflask.api.gen_menu import get_species
from wqflask.api.gen_menu import get_groups
from wqflask.api.gen_menu import get_types
from wqflask.api.gen_menu import phenotypes_exist
from wqflask.api.gen_menu import genotypes_exist
from wqflask.api.gen_menu import build_datasets
from wqflask.api.gen_menu import build_types


class TestGenMenu(unittest.TestCase):
    """Tests for the gen_menu module"""

    def setUp(self):
        self.test_group = {
            'mouse': [
                ['H_T1',
                 'H_T',
                 'Family:DescriptionA'
                 ],
                ['H_T2', "H_T'", 'Family:None']
            ],
            'human': [
                ['BXD', 'BXD', 'Family:None'],
                ['HLC', 'Liver: Normal Gene Expression with Genotypes (Merck)',
                 'Family:Test']
            ]
        }

    @mock.patch('wqflask.api.gen_menu.g')
    def test_get_species(self, db_mock):
        """Test that assertion is raised when dataset and dataset_name are defined"""
        db_mock.db.execute.return_value.fetchall.return_value = (('human', 'Human'),
                                                                 ('mouse', 'Mouse'))
        self.assertEqual(get_species(),
                         [['human', 'Human'], ['mouse', 'Mouse']])
        db_mock.db.execute.assert_called_once_with(
            "SELECT Name, MenuName FROM Species ORDER BY OrderId"
        )

    @mock.patch('wqflask.api.gen_menu.g')
    def test_get_groups(self, db_mock):
        """Test that species groups are grouped correctly"""
        db_mock.db.execute.return_value.fetchall.side_effect = [
            # Mouse
            (('BXD', 'BXD', None),
             ('HLC', 'Liver: Normal Gene Expression with Genotypes (Merck)', 'Test')),
            # Human
            (('H_T1', "H_T", "DescriptionA"),
             ('H_T2', "H_T'", None))
        ]

        self.assertEqual(get_groups([["human", "Human"], ["mouse", "Mouse"]]),
                         self.test_group)

        for name in ["mouse", "human"]:
            db_mock.db.execute.assert_any_call(
                ("SELECT InbredSet.Name, InbredSet.FullName, " +
                 "IFNULL(InbredSet.Family, 'None') " +
                 "FROM InbredSet, Species WHERE Species.Name " +
                 "= '{}' AND InbredSet.SpeciesId = Species.Id GROUP by " +
                 "InbredSet.Name ORDER BY IFNULL(InbredSet.FamilyOrder, " +
                 "InbredSet.FullName) ASC, IFNULL(InbredSet.Family, " +
                 "InbredSet.FullName) ASC, InbredSet.FullName ASC, " +
                 "InbredSet.MenuOrderId ASC").format(name)
            )

    @mock.patch('wqflask.api.gen_menu.g')
    def test_phenotypes_exist_called_with_correct_query(self, db_mock):
        """Test that phenotypes_exist is called with the correct query"""
        db_mock.db.execute.return_value.fetchone.return_value = None
        phenotypes_exist("test")
        db_mock.db.execute.assert_called_with(
            "SELECT Name FROM PublishFreeze WHERE PublishFreeze.Name = 'testPublish'"
        )

    @mock.patch('wqflask.api.gen_menu.g')
    def test_phenotypes_exist_with_falsy_values(self, db_mock):
        """Test that phenotype check returns correctly when given a None value"""
        for x in [None, False, (), [], ""]:
            db_mock.db.execute.return_value.fetchone.return_value = x
            self.assertFalse(phenotypes_exist("test"))

    @mock.patch('wqflask.api.gen_menu.g')
    def test_phenotypes_exist_with_truthy_value(self, db_mock):
        """Test that phenotype check returns correctly when given Truthy """
        for x in ["x", ("result"), ["result"], [1]]:
            db_mock.db.execute.return_value.fetchone.return_value = (x)
            self.assertTrue(phenotypes_exist("test"))

    @mock.patch('wqflask.api.gen_menu.g')
    def test_genotypes_exist_called_with_correct_query(self, db_mock):
        """Test that genotypes_exist is called with the correct query"""
        db_mock.db.execute.return_value.fetchone.return_value = None
        genotypes_exist("test")
        db_mock.db.execute.assert_called_with(
            "SELECT Name FROM GenoFreeze WHERE GenoFreeze.Name = 'testGeno'"
        )

    @mock.patch('wqflask.api.gen_menu.g')
    def test_genotypes_exist_with_falsy_values(self, db_mock):
        """Test that genotype check returns correctly when given a None value"""
        for x in [None, False, (), [], ""]:
            db_mock.db.execute.return_value.fetchone.return_value = x
            self.assertFalse(genotypes_exist("test"))

    @mock.patch('wqflask.api.gen_menu.g')
    def test_genotypes_exist_with_truthy_value(self, db_mock):
        """Test that genotype check returns correctly when given Truthy """
        for x in ["x", ("result"), ["result"], [1]]:
            db_mock.db.execute.return_value.fetchone.return_value = (x)
            self.assertTrue(phenotypes_exist("test"))


    @mock.patch('wqflask.api.gen_menu.g')
    def test_build_datasets_with_type_phenotypes(self, db_mock):
        """Test that correct dataset is returned for a phenotype type"""
        db_mock.db.execute.return_value.fetchall.return_value = (
            (602, "BXDPublish", "BXD Published Phenotypes"),
        )
        self.assertEqual(build_datasets("Mouse", "BXD", "Phenotypes"),
                         [['602', "BXDPublish", "BXD Published Phenotypes"]])
        db_mock.db.execute.assert_called_with(
            "SELECT InfoFiles.GN_AccesionId, PublishFreeze.Name, " +
            "PublishFreeze.FullName FROM InfoFiles, PublishFreeze, " +
            "InbredSet WHERE InbredSet.Name = 'BXD' AND " +
            "PublishFreeze.InbredSetId = InbredSet.Id AND " +
            "InfoFiles.InfoPageName = PublishFreeze.Name " +
            "ORDER BY PublishFreeze.CreateTime ASC"
        )
        self.assertEqual(build_datasets("Mouse", "MDP", "Phenotypes"),
                         [['602', "BXDPublish", "Mouse Phenome Database"]])

        db_mock.db.execute.return_value.fetchall.return_value = ()
        db_mock.db.execute.return_value.fetchone.return_value = (
            "BXDPublish", "Mouse Phenome Database"
        )
        self.assertEqual(build_datasets("Mouse", "MDP", "Phenotypes"),
                         [["None", "BXDPublish", "Mouse Phenome Database"]])

    @mock.patch('wqflask.api.gen_menu.g')
    def test_build_datasets_with_type_genotypes(self, db_mock):
        """Test that correct dataset is returned for a phenotype type"""
        db_mock.db.execute.return_value.fetchone.return_value = (
            635, "HLCPublish", "HLC Published Genotypes"
        )

        self.assertEqual(build_datasets("Mouse", "HLC", "Genotypes"),
                         [["635", "HLCGeno", "HLC Genotypes"]])
        db_mock.db.execute.assert_called_with(
            "SELECT InfoFiles.GN_AccesionId FROM InfoFiles, GenoFreeze, InbredSet " +
            "WHERE InbredSet.Name = 'HLC' AND GenoFreeze.InbredSetId = InbredSet.Id AND " +
            "InfoFiles.InfoPageName = GenoFreeze.ShortName " +
            "ORDER BY GenoFreeze.CreateTime DESC"
        )
        db_mock.db.execute.return_value.fetchone.return_value = ()
        self.assertEqual(build_datasets("Mouse", "HLC", "Genotypes"),
                         [["None", "HLCGeno", "HLC Genotypes"]])

    @mock.patch('wqflask.api.gen_menu.g')
    def test_build_datasets_with_type_mrna(self, db_mock):
        """Test that correct dataset is returned for a mRNA expression/ Probeset"""
        db_mock.db.execute.return_value.fetchall.return_value = (
            (112, "HC_M2_0606_P",
             "Hippocampus Consortium M430v2 (Jun06) PDNN"), )
        self.assertEqual(build_datasets("Mouse", "HLC", "mRNA"), [[
            "112", 'HC_M2_0606_P', "Hippocampus Consortium M430v2 (Jun06) PDNN"
        ]])
        db_mock.db.execute.assert_called_once_with(
            "SELECT ProbeSetFreeze.Id, ProbeSetFreeze.Name, " +
            "ProbeSetFreeze.FullName FROM ProbeSetFreeze, " +
            "ProbeFreeze, InbredSet, Tissue, Species WHERE " +
            "Species.Name = 'Mouse' AND Species.Id = " +
            "InbredSet.SpeciesId AND InbredSet.Name = 'HLC' AND " +
            "ProbeSetFreeze.ProbeFreezeId = ProbeFreeze.Id and " +
            "Tissue.Name = 'mRNA' AND ProbeFreeze.TissueId = " +
            "Tissue.Id and ProbeFreeze.InbredSetId = InbredSet.Id " +
            "ORDER BY ProbeSetFreeze.CreateTime DESC")

    @mock.patch('wqflask.api.gen_menu.build_datasets')
    @mock.patch('wqflask.api.gen_menu.g')
    def test_build_types(self, db_mock, datasets_mock):
        """Test that correct tissue metadata is returned"""
        datasets_mock.return_value = [
            ["112", 'HC_M2_0606_P', "Hippocampus Consortium M430v2 (Jun06) PDNN"]
        ]
        db_mock.db.execute.return_value.fetchall.return_value = (
            ('Mouse Tissue'), ('Human Tissue'), ('Rat Tissue')
        )
        self.assertEqual(build_types('mouse', 'random group'),
                         [['M', 'M', 'Molecular Trait Datasets'],
                          ['H', 'H', 'Molecular Trait Datasets'],
                          ['R', 'R', 'Molecular Trait Datasets']])
        db_mock.db.execute.assert_called_once_with(
            "SELECT DISTINCT Tissue.Name " +
            "FROM ProbeFreeze, ProbeSetFreeze, InbredSet, " +
            "Tissue, Species WHERE Species.Name = 'mouse' " +
            "AND Species.Id = InbredSet.SpeciesId AND " +
            "InbredSet.Name = 'random group' AND " +
            "ProbeFreeze.TissueId = Tissue.Id AND " +
            "ProbeFreeze.InbredSetId = InbredSet.Id AND " +
            "ProbeSetFreeze.ProbeFreezeId = ProbeFreeze.Id " +
            "ORDER BY Tissue.Name"
        )

    @mock.patch('wqflask.api.gen_menu.build_types')
    @mock.patch('wqflask.api.gen_menu.genotypes_exist')
    @mock.patch('wqflask.api.gen_menu.phenotypes_exist')
    def test_get_types_with_existing_genotype_and_phenotypes(
            self,
            phenotypes_exist_mock,
            genotypes_exist_mock,
            build_types_mock):
        """Test that build types are constructed correctly if phenotypes and genotypes
        exist

        """
        phenotypes_exist_mock.return_value = True
        genotypes_exist_mock.return_value = True

        expected_result = {
            'mouse': {
                'H_T2': [('Phenotypes',
                          'Traits and Cofactors',
                          'Phenotypes'),
                         ('Genotypes',
                          'DNA Markers and SNPs',
                          'Genotypes'),
                         ['M', 'M', 'Molecular Trait Datasets']],
                'H_T1': [('Phenotypes',
                          'Traits and Cofactors',
                          'Phenotypes'),
                         ('Genotypes',
                          'DNA Markers and SNPs',
                          'Genotypes'),
                         ['M', 'M', 'Molecular Trait Datasets']]
            },
            'human': {
                'HLC': [('Phenotypes',
                         'Traits and Cofactors',
                         'Phenotypes'),
                        ('Genotypes',
                         'DNA Markers and SNPs',
                         'Genotypes'),
                        ['M', 'M', 'Molecular Trait Datasets']],
                'BXD': [('Phenotypes',
                         'Traits and Cofactors',
                         'Phenotypes'),
                        ('Genotypes',
                         'DNA Markers and SNPs',
                         'Genotypes'),
                        ['M', 'M', 'Molecular Trait Datasets']]
            }
        }

        build_types_mock.return_value = [
            ['M', 'M', 'Molecular Trait Datasets']
        ]
        self.assertEqual(get_types(self.test_group), expected_result)

    @mock.patch('wqflask.api.gen_menu.build_types')
    @mock.patch('wqflask.api.gen_menu.genotypes_exist')
    @mock.patch('wqflask.api.gen_menu.phenotypes_exist')
    def test_get_types_with_buildtype_and_non_existent_genotype_and_phenotypes(
            self,
            phenotypes_exist_mock,
            genotypes_exist_mock,
            build_types_mock):
        """Test that build types are constructed correctly if phenotypes_exist and
        genotypes_exist are false but build_type is falsy

        """
        phenotypes_exist_mock.return_value = False
        genotypes_exist_mock.return_value = False

        build_types_mock.return_value = []
        self.assertEqual(get_types(self.test_group), {
            'mouse': {},
            'human': {}
        })

    @mock.patch('wqflask.api.gen_menu.build_types')
    @mock.patch('wqflask.api.gen_menu.genotypes_exist')
    @mock.patch('wqflask.api.gen_menu.phenotypes_exist')
    def test_get_types_with_non_existent_genotype_phenotypes_and_buildtype(
            self,
            phenotypes_exist_mock,
            genotypes_exist_mock,
            build_types_mock):
        """Test that build types are constructed correctly if phenotypes_exist,
        genotypes_exist and build_types are truthy

        """
        phenotypes_exist_mock.return_value = False
        genotypes_exist_mock.return_value = False

        build_types_mock.return_value = [
            ['M', 'M', 'Molecular Trait Datasets']
        ]
        expected_result = {
            'mouse': {
                'H_T2': [['M', 'M', 'Molecular Trait Datasets']],
                'H_T1': [['M', 'M', 'Molecular Trait Datasets']]},
            'human': {
                'HLC': [['M', 'M', 'Molecular Trait Datasets']],
                'BXD': [['M', 'M', 'Molecular Trait Datasets']]}}
        self.assertEqual(get_types(self.test_group),
                         expected_result)