add fetch dataset strain id,strain name and unittests

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/integration/test_datasets.py b/tests/integration/test_datasets.py
index 255d998..cb08ad1 100644
--- a/tests/integration/test_datasets.py
+++ b/tests/integration/test_datasets.py
@@ -26,3 +26,12 @@ class DatasetIntegrationTests(TestCase):
         results = response.get_json()["dataset"]
         self.assertEqual(results[1], "ProbeSet")
         self.assertEqual(response.status_code, 200)
+
+    def test_fetch_traits_data(self):
+        """test api/dataset/fetch_traits_data/d_name/d_type"""
+
+        response = self.app.get(
+            "/api/dataset/fetch_traits_data/Aging-Brain-UCIPublish/Publish")
+
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.get_json(), {})