Unverified Commit 97344add authored by Aaron Hill's avatar Aaron Hill
Browse files

Don't sleep when replaying recorded data

parent a7d21baf
......@@ -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():
......
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()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment