def test_state_handler_by_mock(self):
client = MagicMock()
state_handler = StateHandler(client)
client.get_storage.assert_not_called()
client.update_storage = MagicMock(return_value=dict(result='success'))
state_handler.put('key', [1, 2, 3])
client.update_storage.assert_called_with(
dict(storage=dict(key='[1, 2, 3]')))
val = state_handler.get('key')
client.get_storage.assert_not_called()
self.assertEqual(val, [1, 2, 3])
# force us to get non-cached values
client.get_storage = MagicMock(return_value=dict(
result='success', storage=dict(non_cached_key='[5]')))
val = state_handler.get('non_cached_key')
client.get_storage.assert_called_with({'keys': ['non_cached_key']})
self.assertEqual(val, [5])
# value must already be cached
client.get_storage = MagicMock()
val = state_handler.get('non_cached_key')
client.get_storage.assert_not_called()
self.assertEqual(val, [5])
def test_state_handler(self):
client = MagicMock()
state_handler = StateHandler(client)
client.get_storage.assert_not_called()
client.update_storage = MagicMock(return_value=dict(result='success'))
state_handler.put('key', [1, 2, 3])
client.update_storage.assert_called_with(dict(storage=dict(key='[1, 2, 3]')))
val = state_handler.get('key')
client.get_storage.assert_not_called()
self.assertEqual(val, [1, 2, 3])
# force us to get non-cached values
client.get_storage = MagicMock(return_value=dict(
result='success',
storage=dict(non_cached_key='[5]')))
val = state_handler.get('non_cached_key')
client.get_storage.assert_called_with(keys=('non_cached_key',))
self.assertEqual(val, [5])
# value must already be cached
client.get_storage = MagicMock()
val = state_handler.get('non_cached_key')
client.get_storage.assert_not_called()
self.assertEqual(val, [5])
def test_state_handler_by_mock(self):
client = MagicMock()
state_handler = StateHandler(client)
client.get_storage.assert_not_called()
client.update_storage = MagicMock(return_value=dict(result="success"))
state_handler.put("key", [1, 2, 3])
client.update_storage.assert_called_with(dict(storage=dict(key="[1, 2, 3]")))
val = state_handler.get("key")
client.get_storage.assert_not_called()
self.assertEqual(val, [1, 2, 3])
# force us to get non-cached values
client.get_storage = MagicMock(
return_value=dict(result="success", storage=dict(non_cached_key="[5]"))
)
val = state_handler.get("non_cached_key")
client.get_storage.assert_called_with({"keys": ["non_cached_key"]})
self.assertEqual(val, [5])
# value must already be cached
client.get_storage = MagicMock()
val = state_handler.get("non_cached_key")
client.get_storage.assert_not_called()
self.assertEqual(val, [5])
def test_state_handler(self):
client = FakeClient()
state_handler = StateHandler(client)
state_handler.put('key', [1, 2, 3])
val = state_handler.get('key')
self.assertEqual(val, [1, 2, 3])
# force us to get non-cached values
state_handler = StateHandler(client)
val = state_handler.get('key')
self.assertEqual(val, [1, 2, 3])
def test_state_handler(self):
client = FakeClient()
state_handler = StateHandler(client)
state_handler.put('key', [1, 2, 3])
val = state_handler.get('key')
self.assertEqual(val, [1, 2, 3])
# force us to get non-cached values
state_handler = StateHandler(client)
val = state_handler.get('key')
self.assertEqual(val, [1, 2, 3])
