unittest cache server

test_cache_server

import unittest
import cache_server
import time


TEST_KEY = "test_key"
TEST_VALUE = "test_value"

TEST_DATA = {TEST_KEY: {"value": TEST_VALUE,
                        "creation_time": time.time(),
                        "timeout": 60,
                        },
             }


class CacheServerTest(unittest.TestCase):
    def setUp(self):
        self.app = cache_server.CacheServer()

    def test_get_returned_value(self):
        self.app.cached_values = TEST_DATA
        self.assertEqual(TEST_VALUE, self.app.get(TEST_KEY))

    def test_delete_key(self):
        self.app.cached_values = TEST_DATA
        self.assertEqual(None, self.app.delete(TEST_KEY))