feat: google music release importer (#1474)

The import function for Google music already currently import of
playlist only. The changes will add support for music releases if they
exist in the request.

Tested via demo-server locally.

Exporter added earlier -
#1469

Author: Shashank1116

extensions/data-transfer/portability-data-transfer-google/src/main/java/org/datatransferproject/datatransfer/google/music/GoogleMusicHttpApi.java

@@ -54,6 +54,7 @@
 import org.datatransferproject.datatransfer.google.musicModels.BatchPlaylistItemResponse;
 import org.datatransferproject.datatransfer.google.musicModels.ExportReleaseResponse;
 import org.datatransferproject.datatransfer.google.musicModels.GooglePlaylist;
+import org.datatransferproject.datatransfer.google.musicModels.GoogleRelease;
 import org.datatransferproject.datatransfer.google.musicModels.ImportPlaylistRequest;
 import org.datatransferproject.datatransfer.google.musicModels.PlaylistItemExportResponse;
 import org.datatransferproject.datatransfer.google.musicModels.PlaylistExportResponse;
@@ -167,6 +168,17 @@ BatchPlaylistItemResponse importPlaylistItems(BatchPlaylistItemRequest playlistI
         BatchPlaylistItemResponse.class);
   }
 
+  GoogleRelease createRelease(GoogleRelease googleRelease)
+      throws IOException, InvalidTokenException, PermissionDeniedException {
+    Map<String, Object> googleReleaseMap = createJsonMap(googleRelease);
+    Map<String, String> params = new LinkedHashMap<>();
+    HttpContent content = new JsonHttpContent(jsonFactory, googleReleaseMap);
+
+    return makePostRequest(RELEASE_BASE_URL, Optional.of(params),
+        content,
+        GoogleRelease.class);
+  }
+
   private <T> T makeGetRequest(
       String baseUrl, Optional<Map<String, String>> parameters, Class<T> clazz)
       throws IOException, InvalidTokenException, PermissionDeniedException {

extensions/data-transfer/portability-data-transfer-google/src/main/java/org/datatransferproject/datatransfer/google/music/GoogleMusicImporter.java

@@ -53,6 +53,7 @@
 import org.datatransferproject.types.common.models.music.MusicGroup;
 import org.datatransferproject.types.common.models.music.MusicPlaylist;
 import org.datatransferproject.types.common.models.music.MusicPlaylistItem;
+import org.datatransferproject.types.common.models.music.MusicRelease;
 import org.datatransferproject.types.transfer.auth.TokensAndUrlAuthData;
 
 public class GoogleMusicImporter implements Importer<TokensAndUrlAuthData, MusicContainerResource> {
@@ -117,7 +118,19 @@ public ImportResult importItem(
 
     // TODO: create tracks
 
-    // TODO: create releases
+    for (MusicRelease release: data.getReleases()){
+      monitor.debug(
+          () ->
+              String.format(
+                  "%s: Google Music importing release item: [release title: %s, release icpn: %s]",
+                  jobId,
+                  release.getTitle(),
+                  release.getIcpnCode()));
+      idempotentImportExecutor.executeAndSwallowIOExceptions(
+          release.getTitle(),
+          release.getTitle(),
+          () -> importSingleRelease(jobId, authData, release));
+    }
 
     return ImportResult.OK;
   }
@@ -136,6 +149,20 @@ String importSinglePlaylist(
     return inputPlaylist.getId();
   }
 
+  @VisibleForTesting
+  String importSingleRelease(
+      UUID jobId, TokensAndUrlAuthData authData, MusicRelease musicRelease)
+      throws IOException, CopyException {
+    // Set up Release
+    GoogleRelease googleRelease = new GoogleRelease();
+    googleRelease.setReleaseTitle(musicRelease.getTitle());
+    googleRelease.setIcpn(musicRelease.getIcpnCode());
+
+    getOrCreateMusicInterface(jobId, authData)
+        .createRelease(googleRelease);
+    return musicRelease.getTitle();
+  }
+
   void importPlaylistItems(
       List<MusicPlaylistItem> playlistItems,
       IdempotentImportExecutor executor,

extensions/data-transfer/portability-data-transfer-google/src/test/java/org/datatransferproject/datatransfer/google/music/GoogleMusicImporterTest.java

@@ -22,6 +22,7 @@
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
@@ -30,6 +31,7 @@
 import com.google.common.collect.Lists;
 import com.google.protobuf.util.Durations;
 import com.google.rpc.Code;
+import java.io.IOException;
 import java.util.UUID;
 import org.datatransferproject.api.launcher.Monitor;
 import org.datatransferproject.datatransfer.google.common.GoogleCredentialFactory;
@@ -284,6 +286,87 @@ public void importPlaylistItemsSkippableFauilre() throws Exception {
     assertThat(executor.getErrors()).isEmpty();
   }
 
+  @Test
+  void importSingleRelease() throws Exception {
+    // Set up
+    MusicRelease release = new MusicRelease("icpndevtest", "test-release", null);
+    ImmutableList<MusicRelease> releases = ImmutableList.of(release);
+    MusicContainerResource data = new MusicContainerResource(null, null, null, releases);
+
+    GoogleRelease googleRelease = new GoogleRelease();
+    googleRelease.setIcpn("icpndevtest");
+    googleRelease.setReleaseTitle("test-release");
+    when(googleMusicHttpApi.createRelease(any(GoogleRelease.class)))
+        .thenReturn(googleRelease);
+
+    // Run test
+    ImportResult importResult = googleMusicImporter.importItem(uuid, executor, null, data);
+
+    // Check results
+    ArgumentCaptor<GoogleRelease> releaseArgumentCaptor =
+        ArgumentCaptor.forClass(GoogleRelease.class);
+    verify(googleMusicHttpApi)
+        .createRelease(releaseArgumentCaptor.capture());
+    assertEquals("test-release", releaseArgumentCaptor.getValue().getTitle());
+    assertTrue(executor.isKeyCached("test-release"));
+    assertEquals(importResult, ImportResult.OK);
+  }
+
+  @Test
+  void importMultipleReleases() throws Exception {
+    // Set up
+    MusicRelease release1 = new MusicRelease("icpndevtest1", "test-release1", null);
+    MusicRelease release2 = new MusicRelease("icpndevtest2", "test-release2", null);
+    MusicRelease release3 = new MusicRelease("icpndevtest3", "test-release3", null);
+
+    ImmutableList<MusicRelease> releases = ImmutableList.of(release1, release2, release3);
+    MusicContainerResource data = new MusicContainerResource(null, null, null, releases);
+
+    GoogleRelease googleRelease = new GoogleRelease();
+    googleRelease.setIcpn("icpndevtest");
+    googleRelease.setReleaseTitle("test-release");
+    when(googleMusicHttpApi.createRelease(any(GoogleRelease.class)))
+        .thenReturn(googleRelease);
+
+    // Run test
+    ImportResult importResult = googleMusicImporter.importItem(uuid, executor, null, data);
+
+    // Check results
+    verify(googleMusicHttpApi,times(3))
+        .createRelease(any(GoogleRelease.class));
+    assertTrue(executor.isKeyCached("test-release3"));
+    assertEquals(importResult, ImportResult.OK);
+  }
+
+  @Test
+  void importMultipleReleasesWithOneFailure() throws Exception {
+    // Set up
+    MusicRelease release1 = new MusicRelease("icpndevtest1", "test-release1", null);
+    MusicRelease release2 = new MusicRelease("icpndevtest2", "test-release2", null);
+    MusicRelease release3 = new MusicRelease("icpndevtest3", "test-release3", null);
+
+    ImmutableList<MusicRelease> releases = ImmutableList.of(release1, release2, release3);
+    MusicContainerResource data = new MusicContainerResource(null, null, null, releases);
+
+    GoogleRelease googleRelease = new GoogleRelease();
+    googleRelease.setIcpn("icpndevtest");
+    googleRelease.setReleaseTitle("test-release");
+    when(googleMusicHttpApi.createRelease(any(GoogleRelease.class)))
+        .thenReturn(googleRelease)
+        .thenThrow(new IOException("Not found"))
+        .thenReturn(googleRelease);
+
+    // Run test
+    ImportResult importResult = googleMusicImporter.importItem(uuid, executor, null, data);
+
+    // Check results
+    verify(googleMusicHttpApi,times(3))
+        .createRelease(any(GoogleRelease.class));
+    assertTrue(executor.isKeyCached("test-release3"));
+    assertEquals(1, executor.getErrors().size());
+    assertEquals(importResult, ImportResult.OK);
+  }
+
   private void importPlaylistSetUp(String playlistId, String playlistTitle)
       throws Exception {
     MusicPlaylist playlist = new MusicPlaylist(playlistId, playlistTitle, null, null, null);