From 2301b11e8a975f2e6dc7e5144e4b26c34b186501 Mon Sep 17 00:00:00 2001
From: Alexander Kabui
Date: Sun, 4 Apr 2021 13:33:34 +0300
Subject: refactor code for trait data

modify unittest and integration tests for datasets
---
 tests/integration/test_datasets.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'tests/integration')

diff --git a/tests/integration/test_datasets.py b/tests/integration/test_datasets.py
index cb08ad1..1d72234 100644
--- a/tests/integration/test_datasets.py
+++ b/tests/integration/test_datasets.py
@@ -27,11 +27,13 @@ class DatasetIntegrationTests(TestCase):
         self.assertEqual(results[1], "ProbeSet")
         self.assertEqual(response.status_code, 200)
 
-    def test_fetch_traits_data(self):
+    @mock.patch("gn3.api.datasets.get_traits_data")
+    def test_fetch_traits_data(self, mock_get_trait_data):
         """test api/dataset/fetch_traits_data/d_name/d_type"""
 
+        mock_get_trait_data.return_value = {}
         response = self.app.get(
             "/api/dataset/fetch_traits_data/Aging-Brain-UCIPublish/Publish")
 
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(response.get_json(), {})
+        self.assertEqual(response.get_json(), {"results": {}})
-- 
cgit v1.2.3