def test_store_should_call_getStorageKey_to_fetch_correct_storage_key(self):
storage = BlipTVStorage()
storage.getStorageKey = Mock()
result = storage.store({"store": "pokeystore"})
storage.getStorageKey.assert_called_with({'store': 'pokeystore'}, '', {})
def test_getStorageKey_should_call_getResultSetStorageKey_if_type_is_not_set(self):
storage = BlipTVStorage()
storage._getResultSetStorageKey = Mock()
result = storage.getStorageKey({"some_param": "param_value"})
storage._getResultSetStorageKey.assert_called_with({"some_param": "param_value"})
def test_getStorageKey_should_call_getThumbnailStorageKey_if_type_is_thumbnail(self):
storage = BlipTVStorage()
storage._getThumbnailStorageKey = Mock()
result = storage.getStorageKey({"some_param": "param_value"}, "thumbnail")
storage._getThumbnailStorageKey.assert_called_with({"some_param": "param_value"},{})
def test_retrieve_should_call_getStorageKey_to_fetch_correct_storage_key(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
storage = BlipTVStorage()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "some_key"
storage.retrieve("some_key", "some_value", {"prepend": "true"})
storage.getStorageKey.assert_called_with("some_key", "some_value", {"prepend": "true"})
def test_storeValue_should_call_setsetting_with_correct_params(self):
storage = BlipTVStorage()
storage.storeResultSet = Mock()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "key"
storage.storeValue("some_key", "some_value")
sys.modules["__main__"].settings.setSetting.assert_called_with("some_key","some_value")
def test_store_should_call_storeResultSet_if_type_is_not_set(self):
storage = BlipTVStorage()
storage.storeResultSet = Mock()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "key"
storage.store({}, {"store": "pokeystore"})
storage.storeResultSet.assert_called_with("key", {"store": "pokeystore"})
def test_store_should_call_storeValue_if_type_is_set(self):
storage = BlipTVStorage()
storage.storeValue = Mock()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "key"
result = storage.store({}, {"store": "pokeystore"}, "value")
storage.storeValue.assert_called_with("key", {"store": "pokeystore"})
def test_retrieve_should_call_retrieveResultSet_if_type_is_not_set(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
storage = BlipTVStorage()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "some_key"
storage.retrieveResultSet = Mock()
storage.retrieve("some_key")
storage.retrieveResultSet.assert_called_with("some_key")
