bumblebee-status/tests/util/test_store.py

import unittest

import util.store

class store(unittest.TestCase):
    def setUp(self):
        self.store = util.store.Store()

        self.unusedKey = "someRandomUnusedKey"
        self.someKey = "someRandomKey"
        self.someOtherKey = "anotherRandomKey"
        self.someValue = "someRandomValue"
        self.someOtherValue = "anotherRandomValue"

    def test_get_of_unset_key(self):
        self.assertEqual(None, self.store.get(self.unusedKey), 'default value expected to be None')
        self.assertEqual(self.someValue, self.store.get(self.unusedKey, self.someValue), 'wrong user-provided default value returned')

    def test_get_of_set_key(self):
        self.assertNotEqual(self.someValue, None)

        self.store.set(self.someKey, self.someValue)
        self.assertEqual(self.someValue, self.store.get(self.someKey), 'unexpected value for existing key')

    def test_overwrite_set(self):
        self.assertNotEqual(self.someValue, None)
        self.assertNotEqual(self.someOtherValue, self.someValue)

        self.store.set(self.someKey, self.someValue)
        self.store.set(self.someKey, self.someOtherValue)
        self.assertEqual(self.someOtherValue, self.store.get(self.someKey), 'unexpected value for existing key')

    def test_unused_keys(self):
        self.assertNotEqual(self.someKey, self.someOtherKey)

        self.store.set(self.someKey, self.someValue)
        self.store.set(self.someOtherKey, self.someOtherValue)

        self.assertEqual(sorted(self.store.unused_keys()), sorted([self.someKey, self.someOtherKey]))

        self.store.get(self.someKey)
        self.assertEqual(self.store.unused_keys(), [self.someOtherKey])

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4