def test_addToPlaylist_should_call_core_addToAlbum_if_playlist_is_in_params(self):
control = VimeoPlaylistControl()
control.createPlayList = Mock()
control.addToAlbum({"album":"album1"})
sys.modules["__main__"].core.addToAlbum.assert_called_with({'album': 'album1'})
def test_addToPlaylist_should_call_core_add_to_playlist_if_user_has_selected_playlist(self):
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 1
sys.modules["__main__"].feeds.listAll.return_value = ([{"Title":"Album1","album":"album1"},{"Title":"Album2","album":"album2"}])
control = VimeoPlaylistControl()
control.createPlayList = Mock()
control.addToAlbum({})
sys.modules["__main__"].core.addToAlbum.assert_called_with({'album': 'album1'})
def test_addToPlaylist_should_call_createPlaylist_if_user_selects_create_option(self):
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 0
sys.modules["__main__"].feeds.listAll.return_value =([{"Title":"Album1"},{"Title":"Album2"}])
control = VimeoPlaylistControl()
control.createAlbum = Mock()
control.addToAlbum({})
control.createAlbum.assert_called_with({})
def test_addToPlaylist_should_call_list_all_if_playlist_is_not_in_params(self):
sys.modules["__main__"].feeds.listAll.return_value = ([])
control = VimeoPlaylistControl()
control.createPlayList = Mock()
control.addToAlbum({})
sys.modules["__main__"].feeds.listAll.assert_called_with({'api': 'my_albums', 'login': '******', 'folder': 'album'})
assert(sys.modules["__main__"].feeds.listAll.call_count == 1)
def test_addToPlaylist_should_ask_user_for_album_if_album_is_not_in_params(self):
sys.modules["__main__"].feeds.listAll.return_value = ([{"Title":"Album1"},{"Title":"Album2"}])
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 0
control = VimeoPlaylistControl()
control.createPlayList = Mock()
control.addToAlbum({})
assert(sys.modules["__main__"].xbmcgui.Dialog.call_count == 2)
assert(sys.modules["__main__"].xbmcgui.Dialog().select.call_count == 1)
