From 97344add2ffbe78e60e47bc29af40bb6d909ee51 Mon Sep 17 00:00:00 2001 From: Aaron Hill Date: Fri, 4 Nov 2016 19:06:15 -0400 Subject: [PATCH] Don't sleep when replaying recorded data --- tests/config.py | 2 +- tests/test_api.py | 13 +++++++++---- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/tests/config.py b/tests/config.py index 3f8c24c..165909d 100644 --- a/tests/config.py +++ b/tests/config.py @@ -33,7 +33,7 @@ class TweepyTestCase(unittest.TestCase): self.auth = create_auth() self.api = API(self.auth) self.api.retry_count = 2 - self.api.retry_delay = 5 + self.api.retry_delay = 0 if use_replay else 5 def create_auth(): diff --git a/tests/test_api.py b/tests/test_api.py index e38c4a0..db3b298 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -1,7 +1,7 @@ import unittest import random import shutil -from time import sleep +import time import os from ast import literal_eval @@ -440,7 +440,7 @@ class TweepyAPITests(TweepyTestCase): class TweepyCacheTests(unittest.TestCase): - timeout = 2.0 + timeout = 0.5 memcache_servers = ['127.0.0.1:11211'] # must be running for test to pass def _run_tests(self, do_cleanup=True): @@ -450,14 +450,14 @@ class TweepyCacheTests(unittest.TestCase): 'Stored value does not match retrieved value') # test timeout - sleep(self.timeout) + sleep(self.timeout, True) self.assertEqual(self.cache.get('testkey'), None, 'Cache entry should have expired') # test cleanup if do_cleanup: self.cache.store('testkey', 'testvalue') - sleep(self.timeout) + sleep(self.timeout, True) self.cache.cleanup() self.assertEqual(self.cache.count(), 0, 'Cache cleanup failed') @@ -484,6 +484,11 @@ class TweepyCacheTests(unittest.TestCase): if os.path.exists('cache_test_dir'): shutil.rmtree('cache_test_dir') +old_sleep = time.sleep + +def sleep(t, override=False): + if not use_replay or override: + old_sleep(t) if __name__ == '__main__': unittest.main() -- 2.25.1