datastore-postgres testFixtures

Author: stephane-airbyte

airbyte-cdk/java/airbyte-cdk/core/src/test/kotlin/io/airbyte/cdk/db/jdbc/TestDefaultJdbcDatabase.kt

@@ -132,19 +132,19 @@ internal class TestDefaultJdbcDatabase {
                 Jsons.jsonNode(ImmutableMap.of("id", 3, "name", "vash"))
             )
 
-        private var PSQL_DB: PostgreSQLContainer<*>? = null
+        private lateinit var PSQL_DB: PostgreSQLContainer<Nothing>
 
         @JvmStatic
         @BeforeAll
         fun init(): Unit {
             PSQL_DB = PostgreSQLContainer<Nothing>("postgres:13-alpine")
-            PSQL_DB!!.start()
+            PSQL_DB.start()
         }
 
         @JvmStatic
         @AfterAll
         fun cleanUp(): Unit {
-            PSQL_DB!!.close()
+            PSQL_DB.close()
         }
     }
 }

airbyte-cdk/java/airbyte-cdk/core/src/test/kotlin/io/airbyte/cdk/db/jdbc/TestJdbcUtils.kt

@@ -334,7 +334,7 @@ internal class TestJdbcUtils {
                 Jsons.jsonNode(ImmutableMap.of("id", 3, "name", "vash"))
             )
 
-        private var PSQL_DB: PostgreSQLContainer<*>? = null
+        private lateinit var PSQL_DB: PostgreSQLContainer<Nothing>
 
         private val sourceOperations: JdbcSourceOperations = JdbcUtils.defaultSourceOperations
 

airbyte-cdk/java/airbyte-cdk/core/src/test/kotlin/io/airbyte/cdk/db/jdbc/TestStreamingJdbcDatabase.kt

@@ -162,7 +162,7 @@ internal class TestStreamingJdbcDatabase {
     }
 
     companion object {
-        private var PSQL_DB: PostgreSQLContainer<*>? = null
+        private lateinit var PSQL_DB: PostgreSQLContainer<Nothing>
 
         @JvmStatic
         @BeforeAll

airbyte-cdk/java/airbyte-cdk/datastore-postgres/src/testFixtures/java/io/airbyte/cdk/testutils/PostgreSQLContainerHelper.java

@@ -0,0 +1,33 @@
+/*
+ * Copyright (c) 2023 Airbyte, Inc., all rights reserved.
+ */
+package io.airbyte.cdk.testutils
+
+import java.io.IOException
+import java.util.*
+import org.testcontainers.containers.PostgreSQLContainer
+import org.testcontainers.utility.MountableFile
+
+object PostgreSQLContainerHelper {
+    @JvmStatic
+    fun runSqlScript(file: MountableFile?, db: PostgreSQLContainer<Nothing>) {
+        try {
+            val scriptPath = "/etc/" + UUID.randomUUID() + ".sql"
+            db.copyFileToContainer(file, scriptPath)
+            db.execInContainer(
+                "psql",
+                "-d",
+                db.databaseName,
+                "-U",
+                db.username,
+                "-a",
+                "-f",
+                scriptPath
+            )
+        } catch (e: InterruptedException) {
+            throw RuntimeException(e)
+        } catch (e: IOException) {
+            throw RuntimeException(e)
+        }
+    }
+}