def test_saveSearch_should_call_storeValue(self): sys.modules["__main__"].settings.getSetting.return_value = "0" storage = BlipTVStorage() storage.retrieve = Mock() storage.store = Mock() storage.retrieve.return_value = ["some_search", "some_search2", "some_search3"] storage.saveSearch({"scraper":"search","search": "some_search2"}) storage.store.assert_called_with({"scraper":"search", "search": "some_search2"}, ['some_search2', 'some_search', 'some_search3'])
def test_editStoredSearch_should_ask_user_for_new_search_phrase(self): sys.modules["__main__"].settings.getSetting.return_value = "0" sys.modules["__main__"].language.return_value = "some_title" sys.modules["__main__"].common.getUserInput.return_value = "some_search3" storage = BlipTVStorage() storage.retrieve = Mock() storage.retrieve.return_value = ["some_search2"] storage.saveSearch = Mock() storage.editStoredSearch({"search": "some_search1"}) sys.modules["__main__"].common.getUserInput.assert_called_with('some_title', 'some_search1')
def test_editStoredSearch_should_exit_cleanly_if_search_param_is_missing(self): sys.modules["__main__"].settings.getSetting.return_value = "0" sys.modules["__main__"].common.getUserInput.return_value = "some_search3" storage = BlipTVStorage() storage.retrieve = Mock() storage.retrieve.return_value = ["some_search2"] storage.saveSearch = Mock() storage.editStoredSearch({}) assert(storage.saveSearch.call_count == 0) assert(storage.retrieve.call_count == 0)
def test_editStoredSearch_should_save_search(self): sys.modules["__main__"].settings.getSetting.return_value = "0" sys.modules["__main__"].language.return_value = "some_title" sys.modules["__main__"].common.getUserInput.return_value = "some_search3" storage = BlipTVStorage() storage.retrieve = Mock() storage.retrieve.return_value = ["some_search2"] storage.saveSearch = Mock() storage.editStoredSearch({"search": "some_search1", "action": "edit_search"}) sys.modules["__main__"].common.getUserInput.assert_called_with('some_title', 'some_search1') storage.saveSearch.assert_called_with({'search': 'some_search3', 'feed': 'search'})