rana-cli/tests/test_session_cache.py

# -*- coding: utf-8 -*-


import tempfile
from wakatime.session_cache import SessionCache
from . import utils


class SessionCacheTestCase(utils.TestCase):

    def test_can_crud_session(self):
        with tempfile.NamedTemporaryFile() as fh:
            db_file = fh.name

        cache = SessionCache()
        cache.DB_FILE = db_file
        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)