[tests] Add unit tests for the store class

This commit is contained in:
Tobi-wan Kenobi 2017-03-05 13:04:20 +01:00
parent 00849aa4fa
commit c7621caf78

26
tests/test_store.py Normal file
View file

@ -0,0 +1,26 @@
# pylint: disable=C0103,C0111,W0703
import unittest
from bumblebee.store import Store
class TestStore(unittest.TestCase):
def setUp(self):
self.store = Store()
self.anyKey = "some-key"
self.anyValue = "some-value"
self.unsetKey = "invalid-key"
def test_set_value(self):
self.store.set(self.anyKey, self.anyValue)
self.assertEquals(self.store.get(self.anyKey), self.anyValue)
def test_get_invalid_value(self):
result = self.store.get(self.unsetKey)
self.assertEquals(result, None)
def test_get_invalid_with_default_value(self):
result = self.store.get(self.unsetKey, self.anyValue)
self.assertEquals(result, self.anyValue)
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4