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")