Add test for Drive migrator

src/test/java/io/huangsam/photohaul/migration/TestDropboxMigrator.java

@@ -40,7 +40,7 @@ void testMigratePhotos() throws DbxException {
 
         List<String> names = List.of("bauerlite.jpg", "salad.jpg", "foobar.jpg");
         PhotoPathVisitor pathVisitor = visitor(getStaticResources(), names);
-        DropboxMigrator migrator = new DropboxMigrator("/Foo", clientMock, new PhotoResolver(List.of()));
+        Migrator migrator = new DropboxMigrator("folderId123", clientMock, new PhotoResolver(List.of()));
         migrator.migratePhotos(pathVisitor.getPhotos());
 
         assertEquals(2, migrator.getSuccessCount());

src/test/java/io/huangsam/photohaul/migration/TestGoogleDriveMigrator.java

@@ -0,0 +1,52 @@
+package io.huangsam.photohaul.migration;
+
+import com.google.api.services.drive.Drive;
+import com.google.api.services.drive.model.File;
+import com.google.api.services.drive.model.FileList;
+import io.huangsam.photohaul.TestPathBase;
+import io.huangsam.photohaul.traversal.PhotoPathVisitor;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import java.io.IOException;
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+public class TestGoogleDriveMigrator extends TestPathBase {
+    @Mock
+    Drive driveMock;
+
+    @Mock
+    Drive.Files filesMock;
+
+    @Mock
+    Drive.Files.List driveListMock;
+
+    @Mock
+    FileList fileListMock;
+
+    @Mock
+    File fileMock;
+
+    @Test
+    void testMigratePhotos() throws IOException {
+        when(driveMock.files()).thenReturn(filesMock);
+        when(filesMock.list()).thenReturn(driveListMock);
+        when(driveListMock.setQ(any())).thenReturn(driveListMock);
+        when(driveListMock.execute()).thenReturn(fileListMock);
+        when(fileListMock.getFiles()).thenReturn(List.of(fileMock));
+
+        List<String> names = List.of("bauerlite.jpg", "salad.jpg", "foobar.jpg");
+        PhotoPathVisitor pathVisitor = visitor(getStaticResources(), names);
+        Migrator migrator = new GoogleDriveMigrator("Foo", driveMock, new PhotoResolver(List.of()));
+        migrator.migratePhotos(pathVisitor.getPhotos());
+
+        assertEquals(3, migrator.getFailureCount());
+    }
+}

src/test/java/io/huangsam/photohaul/migration/TestPathMigrator.java

@@ -14,7 +14,7 @@ public class TestPathMigrator extends TestPathBase {
     void testMigratePhotos() {
         List<String> names = List.of("bauerlite.jpg", "salad.jpg", "foobar.jpg");
         PhotoPathVisitor pathVisitor = visitor(getStaticResources(), names);
-        PathMigrator migrator = new PathMigrator(getTempResources(), new PhotoResolver(List.of()));
+        Migrator migrator = new PathMigrator(getTempResources(), new PhotoResolver(List.of()));
         migrator.migratePhotos(pathVisitor.getPhotos());
 
         assertEquals(2, migrator.getSuccessCount());