From 363424aac6a1ab14e0d1abd672237a75f117686e Mon Sep 17 00:00:00 2001
From: Thibstars <thibault.helsmoortel@gmail.com>
Date: Sun, 13 Aug 2023 13:52:57 +0200
Subject: [PATCH] Add BaseJavaCsvMigrationTest

---
 pom.xml                                       | 10 +++
 sit-csvloader-core/pom.xml                    |  1 -
 sit-csvloader-flyway/pom.xml                  | 12 +++
 .../csv/flyway/BaseJavaCsvMigrationTest.java  | 73 +++++++++++++++++++
 4 files changed, 95 insertions(+), 1 deletion(-)
 create mode 100644 sit-csvloader-flyway/src/test/java/io/sitoolkit/csv/flyway/BaseJavaCsvMigrationTest.java

diff --git a/pom.xml b/pom.xml
index 42888be..525bb67 100644
--- a/pom.xml
+++ b/pom.xml
@@ -155,5 +155,15 @@
 
   </profiles>
 
+  <dependencyManagement>
+    <dependencies>
+      <dependency>
+        <groupId>org.junit.jupiter</groupId>
+        <artifactId>junit-jupiter-engine</artifactId>
+        <version>5.7.0</version>
+        <scope>test</scope>
+      </dependency>
+    </dependencies>
+  </dependencyManagement>
 
 </project>
diff --git a/sit-csvloader-core/pom.xml b/sit-csvloader-core/pom.xml
index 2a2b379..80784ca 100644
--- a/sit-csvloader-core/pom.xml
+++ b/sit-csvloader-core/pom.xml
@@ -22,7 +22,6 @@
     <dependency>
       <groupId>org.junit.jupiter</groupId>
       <artifactId>junit-jupiter-engine</artifactId>
-      <version>5.7.0</version>
       <scope>test</scope>
     </dependency>
     <dependency>
diff --git a/sit-csvloader-flyway/pom.xml b/sit-csvloader-flyway/pom.xml
index 989d021..c64c952 100644
--- a/sit-csvloader-flyway/pom.xml
+++ b/sit-csvloader-flyway/pom.xml
@@ -15,6 +15,7 @@
 
   <properties>
     <flyway.version>7.3.2</flyway.version>
+    <mockito.version>5.4.0</mockito.version>
   </properties>
 
   <dependencies>
@@ -28,5 +29,16 @@
       <artifactId>sit-csvloader-core</artifactId>
       <version>${project.version}</version>
     </dependency>
+    <dependency>
+      <groupId>org.junit.jupiter</groupId>
+      <artifactId>junit-jupiter-engine</artifactId>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.mockito</groupId>
+      <artifactId>mockito-core</artifactId>
+      <version>${mockito.version}</version>
+      <scope>test</scope>
+    </dependency>
   </dependencies>
 </project>
diff --git a/sit-csvloader-flyway/src/test/java/io/sitoolkit/csv/flyway/BaseJavaCsvMigrationTest.java b/sit-csvloader-flyway/src/test/java/io/sitoolkit/csv/flyway/BaseJavaCsvMigrationTest.java
new file mode 100644
index 0000000..bbf2b5b
--- /dev/null
+++ b/sit-csvloader-flyway/src/test/java/io/sitoolkit/csv/flyway/BaseJavaCsvMigrationTest.java
@@ -0,0 +1,73 @@
+package io.sitoolkit.csv.flyway;
+
+import io.sitoolkit.csv.core.CsvLoader;
+import io.sitoolkit.csv.core.LogCallback;
+import io.sitoolkit.csv.core.ResourceFinder;
+import io.sitoolkit.csv.core.TableDataResource;
+import java.sql.Connection;
+import java.util.List;
+import org.flywaydb.core.api.Location;
+import org.flywaydb.core.api.migration.Context;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.ArgumentMatchers;
+import org.mockito.Mockito;
+
+class BaseJavaCsvMigrationTest {
+
+    private static final List<String> RES_DIR_PATHS = List.of("/classes", "/files");
+
+    private BaseJavaCsvMigration migration;
+
+    @BeforeEach
+    void setUp() {
+        this.migration = new V0__1_MigrationTest();
+    }
+
+    @Test
+    void migrateTest() throws Exception {
+        try (var resourceFinderMockedStatic = Mockito.mockStatic(ResourceFinder.class)) {
+
+            try (var csvLoaderMockedStatic = Mockito.mockStatic(CsvLoader.class)) {
+                Context context = Mockito.mock(Context.class, Mockito.RETURNS_DEEP_STUBS);
+                Connection connection = Mockito.mock(Connection.class, Mockito.RETURNS_DEEP_STUBS);
+                Mockito.when(context.getConnection()).thenReturn(connection);
+
+                mockLocations(context);
+
+                List<TableDataResource> tableDataResources = List.of(new TableDataResource());
+
+                resourceFinderMockedStatic.when(() -> ResourceFinder.findTableDataResources(
+                        ArgumentMatchers.eq(V0__1_MigrationTest.class),
+                        ArgumentMatchers.eq(RES_DIR_PATHS),
+                        ArgumentMatchers.any(LogCallback.class))
+                ).thenReturn(tableDataResources);
+
+                migration.migrate(context);
+
+                csvLoaderMockedStatic.verify(() -> CsvLoader.load(
+                        ArgumentMatchers.eq(connection),
+                        ArgumentMatchers.eq(tableDataResources),
+                        ArgumentMatchers.any(LogCallback.class)
+                ));
+            }
+        }
+    }
+
+    private static void mockLocations(Context context) {
+        Location[] locations = new Location[2];
+        Location classPathLocation = Mockito.mock(Location.class);
+        Mockito.when(classPathLocation.isClassPath()).thenReturn(true);
+        Mockito.when(classPathLocation.getPath()).thenReturn(RES_DIR_PATHS.get(0).substring(1));
+        locations[0] = classPathLocation;
+        Location fileSystemLocation = Mockito.mock(Location.class);
+        Mockito.when(fileSystemLocation.isClassPath()).thenReturn(false);
+        Mockito.when(fileSystemLocation.getPath()).thenReturn(RES_DIR_PATHS.get(1));
+        locations[1] = fileSystemLocation;
+        Mockito.when(context.getConfiguration().getLocations()).thenReturn(locations);
+    }
+
+    static class V0__1_MigrationTest extends BaseJavaCsvMigration {
+
+    }
+}
\ No newline at end of file