diff options
Diffstat (limited to 'wqflask')
| -rw-r--r-- | wqflask/tests/wqflask/show_trait/test_export_trait_data.py | 38 |
1 files changed, 19 insertions, 19 deletions
diff --git a/wqflask/tests/wqflask/show_trait/test_export_trait_data.py b/wqflask/tests/wqflask/show_trait/test_export_trait_data.py index 5d0655ce..085c6a30 100644 --- a/wqflask/tests/wqflask/show_trait/test_export_trait_data.py +++ b/wqflask/tests/wqflask/show_trait/test_export_trait_data.py @@ -6,13 +6,13 @@ from wqflask.show_trait.export_trait_data import export_sample_table from wqflask.show_trait.export_trait_data import get_export_metadata -class MockGeneral(object): +class AttributesSetterClass: def __init__(self, obj): for key, value in obj.items(): setattr(self, key, value) -class MockChild(object): +class HelperClass: pass @@ -22,15 +22,15 @@ class TestExportTraits(unittest.TestCase): @mock.patch("wqflask.show_trait.export_trait_data.data_set") def test_get_export_metadata_no_publish(self, mock_dataset, mock_trait): """test for exporting metadata with no publish""" - mock_data_instance = MockGeneral( + mock_dataset_attributes = AttributesSetterClass( {"type": "no_publish", "dataset_name": "Temp", "name": "Temp"}) - group_obj = MockChild() - group_obj.name = "name" - mock_data_instance.group = group_obj - mock_dataset.create_dataset.return_value = mock_data_instance - mock_trait.return_value = MockGeneral({"symbol": "", "description_display": "Description", - "title": "research1", "journal": "", "authors": ""}) + mock_nested_attributes = HelperClass() + mock_nested_attributes.name = "name" + mock_dataset_attributes.group = mock_nested_attributes + mock_dataset.create_dataset.return_value = mock_dataset_attributes + mock_trait.return_value = AttributesSetterClass({"symbol": "", "description_display": "Description", + "title": "research1", "journal": "", "authors": ""}) results = get_export_metadata("random_id", "Temp") expected = [["Record ID: random_id"], @@ -40,22 +40,22 @@ class TestExportTraits(unittest.TestCase): mock_dataset.create_dataset.assert_called_with("Temp") mock_trait.assert_called_with( - dataset=mock_data_instance, name="random_id", cellid=None, get_qtl_info=False) + dataset=mock_dataset_attributes, name="random_id", cellid=None, get_qtl_info=False) self.assertEqual(results, expected) @mock.patch("wqflask.show_trait.export_trait_data.create_trait") @mock.patch("wqflask.show_trait.export_trait_data.data_set") def test_get_export_metadata_with_publish(self, data_mock, trait_mock): """test for exporting metadata with dataset.type=Publish""" - mock_instance = MockGeneral({"type": "Publish", "dataset_name": "Temp", - "name": "Temp", "description_display": "Description goes here"}) - - group_obj = MockChild() - group_obj.name = "name" - mock_instance.group = group_obj - data_mock.create_dataset.return_value = mock_instance - trait_instance = MockGeneral({"symbol": "", "description_display": "Description", - "title": "research1", "journal": "", "authors": ""}) + mock_dataset_attributes = AttributesSetterClass({"type": "Publish", "dataset_name": "Temp", + "name": "Temp", "description_display": "Description goes here"}) + + mock_nested_attributes = HelperClass() + mock_nested_attributes.name = "name" + mock_dataset_attributes.group = mock_nested_attributes + data_mock.create_dataset.return_value = mock_dataset_attributes + trait_instance = AttributesSetterClass({"symbol": "", "description_display": "Description", + "title": "research1", "journal": "", "authors": ""}) trait_mock.return_value = trait_instance results = get_export_metadata( |