private void importAlbums(Collection<PhotoAlbum> albums, TempPhotosData tempPhotosData) { for (PhotoAlbum album : albums) { tempPhotosData.addTempAlbumMapping(album.getId(), album); } }
photoData.addIdMapping(album.getId(), folderId); jobStore.update(jobId, createCacheKey(), photoData);
ContinuationData continuationData = new ContinuationData(null); for (PhotoAlbum album : photosContainerResource.getAlbums()) { continuationData.addContainerResource(new IdOnlyContainerResource(album.getId()));
@VisibleForTesting void importSingleAlbum(UUID jobId, TokensAndUrlAuthData authData, PhotoAlbum inputAlbum) throws IOException { // Set up album GoogleAlbum googleAlbum = new GoogleAlbum(); googleAlbum.setTitle(COPY_PREFIX + inputAlbum.getName()); GoogleAlbum responseAlbum = getOrCreatePhotosInterface(authData).createAlbum(googleAlbum); TempPhotosData tempPhotosData = jobStore.findData(jobId, TEMP_PHOTOS_KEY, TempPhotosData.class); if (tempPhotosData == null) { tempPhotosData = new TempPhotosData(jobId); jobStore.create(jobId, TEMP_PHOTOS_KEY, tempPhotosData); } tempPhotosData.addAlbumId(inputAlbum.getId(), responseAlbum.getId()); jobStore.update(jobId, TEMP_PHOTOS_KEY, tempPhotosData); }
@VisibleForTesting void importSingleAlbum(UUID jobId, PhotoAlbum inputAlbum, SmugMugInterface smugMugInterface) throws IOException { SmugMugAlbumResponse response = smugMugInterface.createAlbum(inputAlbum.getName()); // Put new album ID in job store so photos can be assigned to correct album // TODO(olsona): thread safety! TempPhotosData tempPhotosData = jobStore.findData(jobId, createCacheKey(), TempPhotosData.class); if (tempPhotosData == null) { tempPhotosData = new TempPhotosData(jobId); jobStore.create(jobId, createCacheKey(), tempPhotosData); } tempPhotosData.addAlbumId(inputAlbum.getId(), response.getUri()); jobStore.update(jobId, createCacheKey(), tempPhotosData); }
Assert.assertEquals("folder1", album.getId()); Assert.assertEquals("Test Picture Folder", album.getName()); Assert.assertEquals("image1", photo1.getDataId()); Assert.assertEquals("image/png", photo1.getMediaType()); Assert.assertEquals(album.getId(), photo1.getAlbumId()); Assert.assertEquals("image2", photo2.getDataId()); Assert.assertEquals("image/jpeg", photo2.getMediaType()); Assert.assertEquals(album.getId(), photo2.getAlbumId());