aboutsummaryrefslogtreecommitdiff
path: root/wqflask/tests
diff options
context:
space:
mode:
authorFrederick Muriuki Muriithi2022-09-22 05:11:16 +0300
committerFrederick Muriuki Muriithi2022-09-22 05:11:56 +0300
commitd4f00edc48e4b0b467440cb7566f26920be99d95 (patch)
tree8526227a98d330156b76ebbae9674c370f66d421 /wqflask/tests
parent1292d94abbf2e10698344dd98b7d887ea6bc5221 (diff)
downloadgenenetwork2-d4f00edc48e4b0b467440cb7566f26920be99d95.tar.gz
Update tests: pass in missing arguments.
Diffstat (limited to 'wqflask/tests')
-rw-r--r--wqflask/tests/unit/base/test_data_set.py13
1 files changed, 9 insertions, 4 deletions
diff --git a/wqflask/tests/unit/base/test_data_set.py b/wqflask/tests/unit/base/test_data_set.py
index fa0024f7..f5f26614 100644
--- a/wqflask/tests/unit/base/test_data_set.py
+++ b/wqflask/tests/unit/base/test_data_set.py
@@ -149,9 +149,10 @@ class TestDataSetTypes(unittest.TestCase):
"""
with app.app_context():
redis_mock = mock.Mock()
+ cursor_mock = mock.Mock()
redis_mock.get.return_value = self.test_dataset
self.assertEqual(DatasetType(redis_mock)
- ("All Phenotypes"), "Publish")
+ ("All Phenotypes", redis_mock, cursor_mock), "Publish")
redis_mock.get.assert_called_once_with("dataset_structure")
@mock.patch('base.data_set.datasettype.requests.get')
@@ -163,11 +164,15 @@ class TestDataSetTypes(unittest.TestCase):
with app.app_context():
request_mock.return_value.content = GEN_MENU_JSON
redis_mock = mock.Mock()
+ cursor_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")
+ self.assertEqual(data_set("BXDGeno", redis_mock, cursor_mock),
+ "Geno")
+ self.assertEqual(data_set("BXDPublish", redis_mock, cursor_mock),
+ "Publish")
+ self.assertEqual(data_set("HLC_0311", redis_mock, cursor_mock),
+ "ProbeSet")
redis_mock.set.assert_called_once_with(
"dataset_structure",