670 - fix tests

Author: monikakuster

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/AbstractGoogleDriveActionTest.java

@@ -43,7 +43,7 @@ public abstract class AbstractGoogleDriveActionTest {
     protected ArgumentCaptor<File> fileArgumentCaptor = ArgumentCaptor.forClass(File.class);
     protected ArgumentCaptor<String> fileIdArgumentCaptor = ArgumentCaptor.forClass(String.class);
     protected MockedStatic<GoogleServices> googleServicesMockedStatic;
-    protected ActionContext mockedContext = mock(ActionContext.class);
+    protected ActionContext mockedActionContext = mock(ActionContext.class);
     protected Drive.Files.Create mockedCreate = mock(Drive.Files.Create.class);
     protected Drive.Files.Copy mockedCopy = mock(Drive.Files.Copy.class);
     protected Drive mockedDrive = mock(Drive.class);

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveCopyFileActionTest.java

@@ -53,7 +53,8 @@ void testPerform() throws IOException {
         when(mockedCopy.execute())
             .thenReturn(testFile);
 
-        File copiedFile = GoogleDriveCopyFileAction.perform(mockInputParameters, mockInputParameters, mockedContext);
+        File copiedFile =
+            GoogleDriveCopyFileAction.perform(mockInputParameters, mockInputParameters, mockedActionContext);
 
         verify(mockedDrive.files()).get("originalFileId");
         verify(mockedFiles).copy("originalFileId", testFile);

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveCreateNewFolderActionTest.java

@@ -46,7 +46,7 @@ void testPerform() throws IOException {
         when(mockedCreate.execute())
             .thenReturn(mockedGoogleFile);
 
-        File result = GoogleDriveCreateNewFolderAction.perform(mockedParameters, mockedParameters, mockedContext);
+        File result = GoogleDriveCreateNewFolderAction.perform(mockedParameters, mockedParameters, mockedActionContext);
 
         assertEquals(mockedGoogleFile, result);
 

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveCreateNewTextFileActionTest.java

@@ -53,7 +53,8 @@ void testPerform() throws IOException {
         when(mockedCreate.execute())
             .thenReturn(mockedGoogleFile);
 
-        File result = GoogleDriveCreateNewTextFileAction.perform(mockedParameters, mockedParameters, mockedContext);
+        File result =
+            GoogleDriveCreateNewTextFileAction.perform(mockedParameters, mockedParameters, mockedActionContext);
 
         assertEquals(mockedGoogleFile, result);
 

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveDeleteFileActionTest.java

@@ -49,7 +49,7 @@ void testPerform() throws IOException {
             .when(mockedDelete)
             .execute();
 
-        GoogleDriveDeleteFileAction.perform(mockedParameters, mockedParameters, mockedContext);
+        GoogleDriveDeleteFileAction.perform(mockedParameters, mockedParameters, mockedActionContext);
 
         assertEquals("testId", fileIdArgumentCaptor.getValue());
 
@@ -66,7 +66,7 @@ void testPerformThrowsIOException() throws IOException {
             .execute();
 
         assertThrows(IOException.class,
-            () -> GoogleDriveDeleteFileAction.perform(mockedParameters, mockedParameters, mockedContext));
+            () -> GoogleDriveDeleteFileAction.perform(mockedParameters, mockedParameters, mockedActionContext));
 
         assertEquals("testId", fileIdArgumentCaptor.getValue());
     }

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveDownloadFileActionTest.java

@@ -16,6 +16,7 @@
 
 package com.bytechef.component.google.drive.action;
 
+import static com.bytechef.commons.util.MimeTypeUtils.MIME_APPLICATION_MSWORD_2007;
 import static com.bytechef.component.google.drive.constant.GoogleDriveConstants.FILE_ID;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.mockito.ArgumentMatchers.any;
@@ -27,10 +28,8 @@
 import com.bytechef.component.test.definition.MockParametersFactory;
 import com.google.api.services.drive.Drive;
 import com.google.api.services.drive.model.File;
-import com.google.api.services.drive.model.FileList;
 import java.io.IOException;
 import java.io.InputStream;
-import java.util.List;
 import java.util.Map;
 import org.junit.jupiter.api.Test;
 import org.mockito.ArgumentCaptor;
@@ -42,37 +41,56 @@
 class GoogleDriveDownloadFileActionTest extends AbstractGoogleDriveActionTest {
 
     private final InputStream mockedInputStream = mock(InputStream.class);
-    private final Drive.Files.List mockedList = mock(Drive.Files.List.class);
-    private final FileList mockedFileList = mock(FileList.class);
     private final FileEntry mockedFileEntry = mock(FileEntry.class);
     private final Parameters mockedParameters = MockParametersFactory.create(Map.of(FILE_ID, "fileId"));
-    private final ArgumentCaptor<String> qArgumentCaptor = ArgumentCaptor.forClass(String.class);
+    private final Drive.Files.Export mockedExport = mock(Drive.Files.Export.class);
+    private final ArgumentCaptor<String> exportMimeTypeArgumentCaptor = ArgumentCaptor.forClass(String.class);
 
     @Test
-    void testPerform() throws IOException {
+    void testPerformWithGoogleDoc() throws IOException {
         when(mockedFiles.get(fileIdArgumentCaptor.capture()))
             .thenReturn(mockedGet);
+        when(mockedGet.execute())
+            .thenReturn(new File().setName("testDoc")
+                .setMimeType("application/vnd.google-apps.document"));
+
+        when(mockedFiles.export(fileIdArgumentCaptor.capture(), exportMimeTypeArgumentCaptor.capture()))
+            .thenReturn(mockedExport);
+        when(mockedExport.executeMediaAsInputStream())
+            .thenReturn(mockedInputStream);
+
+        when(mockedActionContext.file(any()))
+            .thenReturn(mockedFileEntry);
+
+        FileEntry result =
+            GoogleDriveDownloadFileAction.perform(mockedParameters, mockedParameters, mockedActionContext);
+
+        assertEquals(mockedFileEntry, result);
+
+        assertEquals("fileId", fileIdArgumentCaptor.getValue());
+        assertEquals(MIME_APPLICATION_MSWORD_2007, exportMimeTypeArgumentCaptor.getValue());
+    }
+
+    @Test
+    void testPerformWithNonGoogleDoc() throws IOException {
+        when(mockedFiles.get(fileIdArgumentCaptor.capture()))
+            .thenReturn(mockedGet);
+        when(mockedGet.execute())
+            .thenReturn(new File().setName("testDoc")
+                .setMimeType("application/pdf")
+                .setFileExtension("pdf"));
+
         when(mockedGet.executeMediaAsInputStream())
             .thenReturn(mockedInputStream);
 
-        when(mockedFiles.list())
-            .thenReturn(mockedList);
-        when(mockedList.setQ(qArgumentCaptor.capture()))
-            .thenReturn(mockedList);
-        when(mockedList.execute())
-            .thenReturn(mockedFileList);
-        when(mockedFileList.getFiles())
-            .thenReturn(List.of(new File().setId("fileId")
-                .setName("fileName")));
-
-        when(mockedContext.file(any()))
+        when(mockedActionContext.file(any()))
             .thenReturn(mockedFileEntry);
 
-        FileEntry result = GoogleDriveDownloadFileAction.perform(mockedParameters, mockedParameters, mockedContext);
+        FileEntry result =
+            GoogleDriveDownloadFileAction.perform(mockedParameters, mockedParameters, mockedActionContext);
 
         assertEquals(mockedFileEntry, result);
 
         assertEquals("fileId", fileIdArgumentCaptor.getValue());
-        assertEquals("mimeType != 'application/vnd.google-apps.folder'", qArgumentCaptor.getValue());
     }
 }

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveGetFileActionTest.java

@@ -48,7 +48,7 @@ void testPerform() throws IOException {
         when(mockedGet.execute())
             .thenReturn(testFile);
 
-        File retrievedFile = GoogleDriveGetFileAction.perform(mockedParameters, mockedParameters, mockedContext);
+        File retrievedFile = GoogleDriveGetFileAction.perform(mockedParameters, mockedParameters, mockedActionContext);
 
         verify(mockedDrive.files()).get("testId");
 

server/libs/modules/components/google/google-drive/src/test/java/com/bytechef/component/google/drive/action/GoogleDriveUploadFileActionTest.java

@@ -50,15 +50,15 @@ void testPerform() throws IOException {
             .thenReturn("name");
         when(mockedFileEntry.getMimeType())
             .thenReturn("mimeType");
-        when(mockedContext.file(any()))
+        when(mockedActionContext.file(any()))
             .thenReturn(mockedFile);
 
         when(mockedFiles.create(fileArgumentCaptor.capture(), abstractInputStreamContentArgumentCaptor.capture()))
             .thenReturn(mockedCreate);
         when(mockedCreate.execute())
             .thenReturn(mockedGoogleFile);
 
-        File result = GoogleDriveUploadFileAction.perform(mockedParameters, mockedParameters, mockedContext);
+        File result = GoogleDriveUploadFileAction.perform(mockedParameters, mockedParameters, mockedActionContext);
 
         assertEquals(mockedGoogleFile, result);