update tests

Author: squirrelo

qiita_pet/handlers/study_handlers/listing_handlers.py

@@ -83,10 +83,11 @@ def _build_single_study_info(study, info, study_proc, proc_samples):
     del info["principal_investigator_id"]
     del info["email"]
     # Build the proc data info list for the child row in datatable
+    info["proc_data_info"] = []
     for data_type, proc_datas in viewitems(study_proc[study.id]):
-        info["proc_data_info"] = [
+        info["proc_data_info"].extend([
             _build_single_proc_data_info(pd_id, data_type, proc_samples[pd_id])
-            for pd_id in proc_datas]
+            for pd_id in proc_datas])
     return info
 
 

qiita_pet/test/test_study_handlers.py

@@ -76,7 +76,7 @@ def test_get_shared_links_for_study(self):
         self.assertEqual(obs, exp)
 
     def test_build_single_study_info(self):
-        study_proc = {1: [1]}
+        study_proc = {1: {'18S': [1]}}
         proc_samples = {1: self.proc_data_exp['samples']}
         study_info = {
             'study_id': 1,
@@ -105,7 +105,8 @@ def test_build_single_study_info(self):
         self.assertEqual(obs, self.single_exp)
 
     def test_build_single_proc_data_info(self):
-        obs = _build_single_proc_data_info(1, self.proc_data_exp['samples'])
+        obs = _build_single_proc_data_info(1, '18S',
+                                           self.proc_data_exp['samples'])
         self.assertEqual(obs, self.proc_data_exp)
 
     def test_build_study_info(self):