lavatech/rana-cli
2
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

tests for connection issues with session caching

Browse source
This commit is contained in:
Alan Hamlett 2015-09-27 01:59:39 -07:00
parent 9af0c110ca
commit dd5678a919
1 changed files with 27 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/test_session_cache.py
View file

@ -10,10 +10,9 @@ class SessionCacheTestCase(utils.TestCase):
def test_can_crud_session(self): def test_can_crud_session(self):
with tempfile.NamedTemporaryFile() as fh: with tempfile.NamedTemporaryFile() as fh:
db_file = fh.name
cache = SessionCache() cache = SessionCache()
cache.DB_FILE = db_file cache.DB_FILE = fh.name
session = cache.get() session = cache.get()
session.headers.update({'x-test': 'abc'}) session.headers.update({'x-test': 'abc'})
cache.save(session) cache.save(session)
@ -22,3 +21,20 @@ class SessionCacheTestCase(utils.TestCase):
cache.delete() cache.delete()
session = cache.get() session = cache.get()
self.assertEquals(session.headers.get('x-test'), None) self.assertEquals(session.headers.get('x-test'), None)
def test_get_handles_connection_error(self):
with tempfile.NamedTemporaryFile() as fh:
cache = SessionCache()
cache.DB_FILE = fh.name
with utils.mock.patch('wakatime.session_cache.SessionCache.connect') as mock_connect:
mock_connect.side_effect = OSError('')
session = cache.get()
session.headers.update({'x-test': 'abc'})
cache.save(session)
session = cache.get()
self.assertEquals(session.headers.get('x-test'), None)
cache.delete()
session = cache.get()
self.assertEquals(session.headers.get('x-test'), None)