def test_set_dataset_key_pheno(self, db_mock):
with app.app_context():
db_mock.db.execute.return_value.fetchone.return_value = [1, 2, 3]
redis_mock = mock.Mock()
redis_mock.get.return_value = self.test_dataset
data_set = DatasetType(redis_mock)
data_set.set_dataset_key("pheno", "Test")
self.assertEqual(data_set("Test"), "Publish")
redis_mock.set.assert_called_once_with(
"dataset_structure",
('{"AD-cases-controls-MyersGeno": "Geno", '
'"AD-cases-controls-MyersPublish": "Publish", '
'"AKXDGeno": "Geno", '
'"AXBXAGeno": "Geno", '
'"AXBXAPublish": "Publish", '
'"Aging-Brain-UCIPublish": "Publish", '
'"All Phenotypes": "Publish", '
'"B139_K_1206_M": "ProbeSet", '
'"B139_K_1206_R": "ProbeSet", '
'"Test": "Publish"}'))
db_mock.db.execute.assert_called_with(
("SELECT InfoFiles.GN_AccesionId "
"FROM InfoFiles, PublishFreeze, InbredSet "
"WHERE InbredSet.Name = 'Test' AND "
"PublishFreeze.InbredSetId = InbredSet.Id AND "
"InfoFiles.InfoPageName = PublishFreeze.Name"))
def test_data_set_type(self):
"""Test that DatasetType returns correctly if the Redis Instance is not empty
and the name variable exists in the dictionary
"""
with app.app_context():
redis_mock = mock.Mock()
redis_mock.get.return_value = self.test_dataset
self.assertEqual(
DatasetType(redis_mock)("All Phenotypes"), "Publish")
redis_mock.get.assert_called_once_with("dataset_structure")
def test_set_dataset_key_mrna(self, db_mock):
with app.app_context():
db_mock.db.execute.return_value.fetchone.return_value = [1, 2, 3]
redis_mock = mock.Mock()
redis_mock.get.return_value = self.test_dataset
data_set = DatasetType(redis_mock)
data_set.set_dataset_key("mrna_expr", "Test")
self.assertEqual(data_set("Test"), "ProbeSet")
redis_mock.set.assert_called_once_with(
"dataset_structure",
('{"AD-cases-controls-MyersGeno": "Geno", '
'"AD-cases-controls-MyersPublish": "Publish", '
'"AKXDGeno": "Geno", '
'"AXBXAGeno": "Geno", '
'"AXBXAPublish": "Publish", '
'"Aging-Brain-UCIPublish": "Publish", '
'"All Phenotypes": "Publish", '
'"B139_K_1206_M": "ProbeSet", '
'"B139_K_1206_R": "ProbeSet", '
'"Test": "ProbeSet"}'))
db_mock.db.execute.assert_called_once_with(
("SELECT ProbeSetFreeze.Id FROM ProbeSetFreeze " +
"WHERE ProbeSetFreeze.Name = \"Test\" "))
def test_data_set_type_with_empty_redis(self, request_mock):
"""Test that DatasetType returns correctly if the Redis Instance is empty and
the name variable exists in the dictionary
"""
with app.app_context():
request_mock.return_value.content = gen_menu_json
redis_mock = mock.Mock()
redis_mock.get.return_value = None
data_set = DatasetType(redis_mock)
self.assertEqual(data_set("BXDGeno"), "Geno")
self.assertEqual(data_set("BXDPublish"), "Publish")
self.assertEqual(data_set("HLC_0311"), "ProbeSet")
redis_mock.set.assert_called_once_with(
"dataset_structure", ('{"HLC_0311": "ProbeSet", '
'"HLCPublish": "Publish", '
'"BXDGeno": "Geno", '
'"HC_M2_0606_P": "ProbeSet", '
'"BXDPublish": "Publish"}'))