def test_can_crud_session(self): with utils.NamedTemporaryFile() as fh: cache = SessionCache() cache.DB_FILE = fh.name session = cache.get() session.headers.update({'x-test': 'abc'}) cache.save(session) session = cache.get() self.assertEquals(session.headers.get('x-test'), 'abc') cache.delete() session = cache.get() self.assertEquals(session.headers.get('x-test'), None)
def test_can_crud_session(self): with tempfile.NamedTemporaryFile() as fh: cache = SessionCache() cache.DB_FILE = fh.name session = cache.get() session.headers.update({'x-test': 'abc'}) cache.save(session) session = cache.get() self.assertEquals(session.headers.get('x-test'), 'abc') cache.delete() session = cache.get() self.assertEquals(session.headers.get('x-test'), None)
def test_can_crud_session(self): with utils.NamedTemporaryFile() as fh: cache = SessionCache() with utils.mock.patch('wakatime.session_cache.SessionCache._get_db_file') as mock_dbfile: mock_dbfile.return_value = fh.name session = cache.get() session.headers.update({'x-test': 'abc'}) cache.save(session) session = cache.get() self.assertEquals(session.headers.get('x-test'), 'abc') cache.delete() session = cache.get() self.assertEquals(session.headers.get('x-test'), None)
def test_handles_connection_exception(self): with utils.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)
def test_handles_connection_exception(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)
def test_can_crud_session(self): with utils.NamedTemporaryFile() as fh: cache = SessionCache() with utils.mock.patch( 'wakatime.session_cache.SessionCache._get_db_file' ) as mock_dbfile: mock_dbfile.return_value = fh.name session = cache.get() session.headers.update({'x-test': 'abc'}) cache.save(session) session = cache.get() self.assertEquals(session.headers.get('x-test'), 'abc') cache.delete() session = cache.get() self.assertEquals(session.headers.get('x-test'), None)