def test_retrieve_should_call_retrieveValue_if_type_is_set(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
storage = YouTubeStorage()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "some_key"
storage.retrieveValue = Mock()
storage.retrieve("some_key", "thumbnail")
storage.retrieveValue.assert_called_with("some_key")
def test_retrieve_should_call_retrieveValue_if_type_is_set(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
storage = YouTubeStorage()
storage.getStorageKey = Mock()
storage.getStorageKey.return_value = "some_key"
storage.retrieveValue = Mock()
storage.retrieve("some_key", "thumbnail")
storage.retrieveValue.assert_called_with("some_key")
def test_retrieveValue_should_call_cache_get_with_correct_params(self):
storage = YouTubeStorage()
storage.retrieveValue("some_key")
sys.modules["__main__"].cache.get.assert_called_with("some_key")
def test_retrieveValue_should_call_cache_get_with_correct_params(self):
storage = YouTubeStorage()
storage.retrieveValue("some_key")
sys.modules["__main__"].cache.get.assert_called_with("some_key")
