remove rate-limiting info from redis for each test (#1214)

Author: melange396

src/acquisition/covidcast/test_utils.py

@@ -4,17 +4,19 @@
 import unittest
 
 import pandas as pd
+from redis import Redis
 
 from delphi_utils import Nans
 from delphi.epidata.common.covidcast_row import CovidcastRow
 from delphi.epidata.acquisition.covidcast.database import Database
+from delphi.epidata.server._config import REDIS_HOST, REDIS_PASSWORD
 from delphi.epidata.server.utils.dates import day_to_time_value, time_value_to_day
 import delphi.operations.secrets as secrets
 
 # all the Nans we use here are just one value, so this is a shortcut to it:
 nmv = Nans.NOT_MISSING.value
 
-# TODO replace these real geo_values with fake values, and use patch and mock to mock the return values of 
+# TODO replace these real geo_values with fake values, and use patch and mock to mock the return values of
 # delphi_utils.geomap.GeoMapper().get_geo_values(geo_type) in parse_geo_sets() of _params.py
 
 FIPS = ['04019', '19143', '29063', '36083'] # Example list of valid FIPS codes as strings
@@ -166,19 +168,25 @@ def setUp(self):
         self.localSetUp()
         self._db._connection.commit()
 
+        # clear all rate-limiting info from redis
+        r = Redis(host=REDIS_HOST, password=REDIS_PASSWORD)
+        for k in r.keys("LIMITER/*"):
+            r.delete(k)
+
+
     def tearDown(self):
         # close and destroy conenction to the database
         self.localTearDown()
         self._db.disconnect(False)
         del self._db
 
     def localSetUp(self):
-        # stub; override in subclasses to perform custom setup. 
+        # stub; override in subclasses to perform custom setup.
         # runs after tables have been truncated but before database changes have been committed
         pass
 
     def localTearDown(self):
-        # stub; override in subclasses to perform custom teardown. 
+        # stub; override in subclasses to perform custom teardown.
         # runs after database changes have been committed
         pass
 
@@ -198,4 +206,4 @@ def params_from_row(self, row: CovidcastTestRow, **kwargs):
             'geo_value': row.geo_value,
         }
         ret.update(kwargs)
-        return ret
+        return ret