Split up remove_expired_reponses() into remove() and reset_expiration() methods, with more granular arguments

1 files changed, 2 insertions, 5 deletions

diff --git a/examples/generate_test_db.py b/examples/generate_test_db.py
index a68b7ee..93822fd 100755
--- a/examples/generate_test_db.py
+++ b/examples/generate_test_db.py
@@ -86,13 +86,13 @@ def get_randomized_response(i=0):
     return new_response
 
 
-def remove_expired_responses(expire_after=None):
+def remove_expired_responses():
     logger.setLevel('DEBUG')
     session = CachedSession(CACHE_NAME)
     total_responses = len(session.cache.responses)
 
     start = time()
-    session.remove_expired_responses(expire_after=expire_after)
+    session.cache.remove(expired=True)
     elapsed = time() - start
     n_removed = total_responses - len(session.cache.responses)
     logger.info(
@@ -121,6 +121,3 @@ if __name__ == '__main__':
 
     # Remove some responses (with randomized expiration)
     # remove_expired_responses()
-
-    # Expire and remove all responses
-    # remove_expired_responses(expire_after=1)