Stan's addToPlaylistTest Sample

AddToPlaylistTaskOperatorTests.java

public class AddToPlaylistTaskOperatorTests {
+    private AddToPlaylistTaskOperator addToPlaylistTaskOperator;
+
+    private APIManager mockAPIManager;
+    private Playlist mockPlaylist;
+
+    @Before
+    public void setup() {
+        mockAPIManager = mock(APIManager.class);
+        mockPlaylist = mock(Playlist.class);
+
+        when(mockPlaylist.getId()).thenReturn("55688");
+
+        addToPlaylistTaskOperator = new AddToPlaylistTaskOperator(mockAPIManager, mockPlaylist);
+    }
+
+    @Test
+    public void testAddSong() {
+        Song song = new Song();
+        song.setId("5566");
+
+        List<String> originSongs = new ArrayList<String>();
+        originSongs.add("222");
+        when(mockPlaylist.getSongIds()).thenReturn(originSongs);
+        when(mockAPIManager.changePlaylistSongsOrder(anyString(), anyString())).thenReturn(Observable.<Response<ResponseBody>>empty());
+
+        addToPlaylistTaskOperator.addSong(song);
+
+        verify(mockAPIManager).changePlaylistSongsOrder("55688", "222,5566");
+    }
+
+    @Test
+    public void testAddSongs() {
+        Song song1 = new Song();
+        song1.setId("5566");
+        Song song2 = new Song();
+        song2.setId("7788");
+        List<Song> addedSongs = new ArrayList<>();
+        addedSongs.add(song1);
+        addedSongs.add(song2);
+
+        Song originSong = new Song();
+        originSong.setId("222");
+        List<Song> originSongs = new ArrayList<>();
+        originSongs.add(originSong);
+        when(mockPlaylist.getSongs()).thenReturn(originSongs);
+        when(mockAPIManager.changePlaylistSongsOrder(anyString(), anyString())).thenReturn(Observable.<Response<ResponseBody>>empty());
+
+        addToPlaylistTaskOperator.addSongs(addedSongs);
+
+        verify(mockAPIManager).changePlaylistSongsOrder("55688", "222,5566,7788");
+    }
+
+    @Test
+    public void testAddSongIds() {
+        List<String> addedSongs = new ArrayList<>();
+        addedSongs.add("5566");
+        addedSongs.add("7788");
+
+        List<String> originSongs = new ArrayList<>();
+        originSongs.add("222");
+        when(mockPlaylist.getSongIds()).thenReturn(originSongs);
+        when(mockAPIManager.changePlaylistSongsOrder(anyString(), anyString())).thenReturn(Observable.<Response<ResponseBody>>empty());
+
+        addToPlaylistTaskOperator.addSongIds(addedSongs);
+
+        verify(mockAPIManager).changePlaylistSongsOrder("55688", "222,5566,7788");
+    }
+}