Updated tests to use statement in order to clean api_user table after each test run. This is done because of returning relationship into User model.

Author: dmytrotsko

integrations/acquisition/covid_hosp/facility/test_scenarios.py

@@ -43,7 +43,7 @@ def setUp(self):
             with db.new_cursor() as cur:
                 cur.execute("truncate table covid_hosp_facility")
                 cur.execute("truncate table covid_hosp_meta")
-                cur.execute("truncate table api_user")
+                cur.execute("delete from api_user")
                 cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
 
     @freeze_time("2021-03-16")

integrations/acquisition/covid_hosp/state_daily/test_scenarios.py

@@ -45,7 +45,7 @@ def setUp(self):
             with db.new_cursor() as cur:
                 cur.execute("truncate table covid_hosp_state_timeseries")
                 cur.execute("truncate table covid_hosp_meta")
-                cur.execute("truncate table api_user")
+                cur.execute("delete from api_user")
                 cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
 
     @freeze_time("2021-03-16")

integrations/acquisition/covid_hosp/state_timeseries/test_scenarios.py

@@ -45,7 +45,7 @@ def setUp(self):
             with db.new_cursor() as cur:
                 cur.execute("truncate table covid_hosp_state_timeseries")
                 cur.execute("truncate table covid_hosp_meta")
-                cur.execute("truncate table api_user")
+                cur.execute("delete from api_user")
                 cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
 
     @freeze_time("2021-03-17")

integrations/acquisition/covidcast_nowcast/test_csv_uploading.py

@@ -43,7 +43,7 @@ def setUp(self):
         )
         cur = cnx.cursor()
         cur.execute("truncate table covidcast_nowcast")
-        cur.execute("truncate table api_user")
+        cur.execute("delete from api_user")
         cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
         cnx.commit()
         cur.close()

integrations/client/test_nowcast.py

@@ -25,7 +25,7 @@ def setUp(self):
         cur = cnx.cursor()
 
         cur.execute("truncate table covidcast_nowcast")
-        cur.execute("truncate table api_user")
+        cur.execute("delete from api_user")
         cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
 
         cnx.commit()

integrations/server/test_covid_hosp.py

@@ -30,8 +30,8 @@ def setUp(self):
             with db.new_cursor() as cur:
                 cur.execute("truncate table covid_hosp_state_timeseries")
                 cur.execute("truncate table covid_hosp_meta")
-                # cur.execute("truncate table api_user")
-                # cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
+                cur.execute("delete from api_user")
+                cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
 
     def insert_issue(self, cur, issue, value, record_type):
         so_many_nulls = ", ".join(["null"] * 57)

integrations/server/test_covidcast_nowcast.py

@@ -23,7 +23,7 @@ def setUp(self):
         )
         cur = cnx.cursor()
         cur.execute("truncate table covidcast_nowcast")
-        cur.execute("truncate table api_user")
+        cur.execute("delete from api_user")
         cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
         cnx.commit()
         cur.close()

integrations/server/test_fluview.py

@@ -34,7 +34,7 @@ def setUp(self):
         )
         cur = cnx.cursor()
         cur.execute("truncate table fluview")
-        cur.execute("truncate table api_user")
+        cur.execute("delete from api_user")
         cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
         cnx.commit()
         cur.close()

integrations/server/test_fluview_meta.py

@@ -34,7 +34,7 @@ def setUp(self):
         )
         cur = cnx.cursor()
         cur.execute("truncate table fluview")
-        cur.execute("truncate table api_user")
+        cur.execute("delete from api_user")
         cur.execute('insert into api_user(api_key, tracking, registered) values ("key", 1, 1)')
         cnx.commit()
         cur.close()