From a301093ce9ea825606db29bd1f1f40e1971897dd Mon Sep 17 00:00:00 2001
From: Alexander Kabui
Date: Mon, 5 Apr 2021 14:34:01 +0300
Subject: fix for fetching dataset traits data

---
 tests/unit/computations/test_datasets.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'tests/unit/computations/test_datasets.py')

diff --git a/tests/unit/computations/test_datasets.py b/tests/unit/computations/test_datasets.py
index b696f70..44ff527 100644
--- a/tests/unit/computations/test_datasets.py
+++ b/tests/unit/computations/test_datasets.py
@@ -29,6 +29,8 @@ class TestDatasets(TestCase):
         dataset_id = "HC_M2_0606_P&"
         dataset_type = "Publish"
 
+        database = mock.Mock()
+
         dataset = {
             "id": dataset_id,
             "type": dataset_type,
@@ -40,7 +42,7 @@ class TestDatasets(TestCase):
         mock_fetch_sample_results.return_value = fetch_results
 
         results = retrieve_trait_sample_data(
-            dataset, trait_name)
+            dataset, trait_name, database)
         self.assertEqual(mock_fetch_sample_results.call_count, 1)
         self.assertEqual(results, fetch_results)
 
-- 
cgit v1.2.3