예제 #1
0
def mock_storage(target = None):
        if target:
            target._ws.stop()

        sh = StorageHandler(STORAGE)
        sh._ws = Mock('send')
        return sh
예제 #2
0
 def testWrongMessage(self):
     self.store = StorageHandler()
     # The first should not raise, but catch and log the error
     self.store.set("storage.command.set.foo", { 'omg_wrong_key' : 'deep' })
     with self.assertRaises(KeyError):
         # But this one should raise
         self.store._store['foo']
예제 #3
0
 def testEmptyStorage(self):
     self.store = StorageHandler()
     self.store._ws = Mock('send')
     self.assertEqual(self.store._store, {})
예제 #4
0
class StorageHandlerTests(unittest.TestCase):
    def setUp(self):
        self.store = mock_storage()

    def tearDown(self):
        self.store.stop()

    def testGet(self):
        self.store.set("storage.command.set.foo", { 'value' : 'baz' })
        self.store.get("storage.command.set.foo", "")
        self.assertEqual(self.store._ws.times('send'), 1)
        self.assertEqual(self.store._ws.args('send'), ["storage.event.foo", { 'value' : 'baz' } ])

    def testGetComplex(self):
        self.store.set("storage.command.set.foo.fizz", { 'value' : 'buz' })
        self.store.get("storage.command.set.foo", "")
        self.assertEqual(self.store._ws.times('send'), 1)
        self.assertEqual(self.store._ws.args('send'), ["storage.event.foo",{ 'value' : { 'fizz' : 'buz' }} ])

    def testSet(self):
        self.store.set("storage.command.set.foo", { 'value' : 'baz' })
        self.assertEqual(self.store._store['foo'], 'baz')

    def testSetNested(self):
        self.store.set("storage.command.set.foo", { 'value' : {'bar': { 'fizz' : 'fuzz' }}})
        self.assertEqual(self.store._store['foo']['bar']['fizz'], 'fuzz')

    def testSetDeep(self):
        self.store.set("storage.command.set.foo.bar.fizz", { 'value' : 'deep' })
        self.assertEqual(self.store._store['foo']['bar']['fizz'], 'deep')

    def testSetMerges(self):
        self.store.set("storage.command.set.foo.bar.fizz", { 'value' : 'deep' })
        self.store.set("storage.command.set.foo.bar.fuzz", { 'value' : 'bizz' })
        self.store.set("storage.command.set.foo.bar", { 'value' : { 'fizz' : 'overriden' }})
        self.store.set("storage.command.set.foo.bar.baz", { 'value' : 'baz' })
        from pprint import pprint
        pprint(self.store._store)
        self.assertEqual(self.store._store['foo']['bar']['fizz'], 'overriden')
        self.assertEqual(self.store._store['foo']['bar']['fuzz'], 'bizz')
        self.assertEqual(self.store._store['foo']['bar']['baz'], 'baz')

    def testSetPersistent(self):
        self.store.set("storage.command.set.foo",
                       {'value': 'fizz',
                        'persistent': True})

        self.store = mock_storage(self.store)
        self.assertEqual(self.store._store['foo'], 'fizz')

    def testSetNestedPersistent(self):
        self.store.set("storage.command.set.foo",
                       { 'value' : {'bar': { 'fizz' : 'fuzz' }},
                        'persistent': True})

        self.store = mock_storage(self.store)
        self.assertEqual(self.store._store['foo']['bar']['fizz'], 'fuzz')

    def testEmptyStorage(self):
        self.store = StorageHandler()
        self.store._ws = Mock('send')
        self.assertEqual(self.store._store, {})

    def testWrongMessage(self):
        self.store = StorageHandler()
        # The first should not raise, but catch and log the error
        self.store.set("storage.command.set.foo", { 'omg_wrong_key' : 'deep' })
        with self.assertRaises(KeyError):
            # But this one should raise
            self.store._store['foo']