tests: mark all tests no rate limit

sgratzl · sgratzl · commit bb77905dfb4d · 2021-08-13T09:38:28.000-04:00
diff --git a/integrations/acquisition/covid_hosp/facility/test_scenarios.py b/integrations/acquisition/covid_hosp/facility/test_scenarios.py
@@ -40,7 +40,7 @@ def setUp(self):
         cur.execute('truncate table covid_hosp_facility')
         cur.execute('truncate table covid_hosp_meta')
         cur.execute("truncate table api_user")
-        cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+        cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
 
   @freeze_time("2021-03-16")
   def test_acquire_dataset(self):
diff --git a/integrations/acquisition/covid_hosp/state_daily/test_scenarios.py b/integrations/acquisition/covid_hosp/state_daily/test_scenarios.py
@@ -45,7 +45,7 @@ def setUp(self):
         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, roles, tracking) values("key", "", 1)')
+        cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
 
   @freeze_time("2021-03-16")
   def test_acquire_dataset(self):
diff --git a/integrations/acquisition/covid_hosp/state_timeseries/test_scenarios.py b/integrations/acquisition/covid_hosp/state_timeseries/test_scenarios.py
@@ -41,7 +41,7 @@ def setUp(self):
         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, roles, tracking) values("key", "", 1)')
+        cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
 
   @freeze_time("2021-03-17")
   def test_acquire_dataset(self):
diff --git a/integrations/acquisition/covidcast/test_covidcast_meta_caching.py b/integrations/acquisition/covidcast/test_covidcast_meta_caching.py
@@ -43,7 +43,7 @@ def setUp(self):
     # clear the `covidcast` table
     cur.execute('truncate table covidcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     # reset the `covidcast_meta_cache` table (it should always have one row)
     cur.execute('update covidcast_meta_cache set timestamp = 0, epidata = ""')
     cnx.commit()
diff --git a/integrations/acquisition/covidcast/test_csv_uploading.py b/integrations/acquisition/covidcast/test_csv_uploading.py
@@ -34,7 +34,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table covidcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/acquisition/covidcast/test_fill_is_latest_issue.py b/integrations/acquisition/covidcast/test_fill_is_latest_issue.py
@@ -31,7 +31,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table covidcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/acquisition/covidcast_nowcast/test_csv_uploading.py b/integrations/acquisition/covidcast_nowcast/test_csv_uploading.py
@@ -42,7 +42,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table covidcast_nowcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/client/test_delphi_epidata.py b/integrations/client/test_delphi_epidata.py
@@ -45,7 +45,7 @@ def setUp(self):
     cur.execute('truncate table covidcast')
     cur.execute('truncate table covidcast_nowcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/server/test_covid_hosp.py b/integrations/server/test_covid_hosp.py
@@ -29,7 +29,7 @@ def setUp(self):
         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, roles, tracking) values("key", "", 1)')
+        cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
 
 
   def insert_issue(self, cur, issue, value, record_type):
diff --git a/integrations/server/test_covidcast.py b/integrations/server/test_covidcast.py
@@ -29,7 +29,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table covidcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/server/test_covidcast_endpoints.py b/integrations/server/test_covidcast_endpoints.py
@@ -111,7 +111,7 @@ def setUp(self):
         cur = cnx.cursor()
         cur.execute("truncate table covidcast")
         cur.execute("truncate table api_user")
-        cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+        cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
         cur.execute('update covidcast_meta_cache set timestamp = 0, epidata = ""')
         cnx.commit()
         cur.close()
diff --git a/integrations/server/test_covidcast_meta.py b/integrations/server/test_covidcast_meta.py
@@ -31,7 +31,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table covidcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cur.execute('update covidcast_meta_cache set timestamp = 0, epidata = ""')
     cnx.commit()
     cur.close()
diff --git a/integrations/server/test_covidcast_nowcast.py b/integrations/server/test_covidcast_nowcast.py
@@ -27,7 +27,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table covidcast_nowcast')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/server/test_fluview.py b/integrations/server/test_fluview.py
@@ -33,7 +33,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table fluview')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/integrations/server/test_fluview_meta.py b/integrations/server/test_fluview_meta.py
@@ -33,7 +33,7 @@ def setUp(self):
     cur = cnx.cursor()
     cur.execute('truncate table fluview')
     cur.execute("truncate table api_user")
-    cur.execute('insert into api_user(api_key, roles, tracking) values("key", "", 1)')
+    cur.execute('insert into api_user(api_key, roles, tracking, registered) values("key", "", 1, 1)')
     cnx.commit()
     cur.close()
 
diff --git a/src/server/_security.py b/src/server/_security.py
@@ -267,6 +267,8 @@ def deduct_on_success(response: Response) -> bool:
 
 @limiter.request_filter
 def _no_rate_limit() -> bool:
+    if app.config.get('TESTING', False) or _is_public_route():
+        return False
     # no rate limit if the user is registered
     user = _get_current_user()
-    return _is_public_route() or user is not None and not user.is_rate_limited()
+    return user is not None and not user.is_rate_limited()
