extending tests

Author: sjanssen2

qiita_db/test/test_user.py

@@ -72,7 +72,10 @@ def setUp(self):
             'pass_reset_code': None,
             'pass_reset_timestamp': None,
             'user_verify_code': None,
-            'receive_processing_job_emails': True
+            'receive_processing_job_emails': True,
+            'social_orcid': None,
+            'social_researchgate': None,
+            'social_googlescholar': None
         }
 
     def tearDown(self):
@@ -125,7 +128,10 @@ def test_create_user(self):
             'address': None,
             'user_level_id': 5,
             'receive_processing_job_emails': False,
-            'email': '[email protected]'}
+            'email': '[email protected]',
+            'social_orcid': None,
+            'social_researchgate': None,
+            'social_googlescholar': None}
         self._check_correct_info(obs, exp)
 
         # Make sure new system messages are linked to user
@@ -162,7 +168,10 @@ def test_create_user_info(self):
             'user_verify_code': '',
             'user_level_id': 5,
             'receive_processing_job_emails': True,
-            'email': '[email protected]'}
+            'email': '[email protected]',
+            'social_orcid': None,
+            'social_researchgate': None,
+            'social_googlescholar': None}
         self._check_correct_info(obs, exp)
 
     def test_create_user_column_not_allowed(self):
@@ -229,7 +238,10 @@ def test_get_info(self):
             'pass_reset_timestamp': None,
             'user_verify_code': None,
             'receive_processing_job_emails': False,
-            'phone': '222-444-6789'
+            'phone': '222-444-6789',
+            'social_orcid': None,
+            'social_researchgate': None,
+            'social_googlescholar': None
         }
         self.assertEqual(self.user.info, expinfo)
 

qiita_db/test/test_util.py

@@ -92,7 +92,8 @@ def test_get_table_cols(self):
         obs = qdb.util.get_table_cols("qiita_user")
         exp = {"email", "user_level_id", "password", "name", "affiliation",
                "address", "phone", "user_verify_code", "pass_reset_code",
-               "pass_reset_timestamp", "receive_processing_job_emails"}
+               "pass_reset_timestamp", "receive_processing_job_emails",
+               "social_orcid", "social_researchgate", "social_googlescholar"}
         self.assertEqual(set(obs), exp)
 
     def test_exists_table(self):