From 769e87ce0aaa6ce37514e4a3bf9c1d859dfab979 Mon Sep 17 00:00:00 2001
From: Parjanya Mudunuri <137906335+parj-hsbc@users.noreply.github.com>
Date: Sun, 11 Aug 2024 19:11:36 +0000
Subject: [PATCH] Added postgres test container

---
 pom.xml                                       | 28 +++++++++--
 .../engineering/PostgresConnectionTest.java   | 50 +++++++++++++++++++
 2 files changed, 75 insertions(+), 3 deletions(-)
 create mode 100644 src/test/java/com/hsbc/engineering/PostgresConnectionTest.java

diff --git a/pom.xml b/pom.xml
index 371b44c..d1e9c1a 100644
--- a/pom.xml
+++ b/pom.xml
@@ -52,6 +52,10 @@
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
     <maven.compiler.source>11</maven.compiler.source>
     <maven.compiler.target>11</maven.compiler.target>
+    <testcontainers.version>1.20.1</testcontainers.version>
+    <junit.version>5.10.2</junit.version>
+    <h2.version>2.2.224</h2.version>
+    <postgres.version>42.7.3</postgres.version>
   </properties>
 
   <dependencyManagement>
@@ -59,7 +63,7 @@
       <dependency>
         <groupId>org.junit</groupId>
         <artifactId>junit-bom</artifactId>
-        <version>5.10.2</version>
+        <version>${junit.version}</version>
         <type>pom</type>
         <scope>import</scope>
       </dependency>
@@ -78,7 +82,25 @@
     <dependency>
       <groupId>com.h2database</groupId>
       <artifactId>h2</artifactId>
-      <version>2.2.224</version>
+      <version>${h2.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.testcontainers</groupId>
+      <artifactId>postgresql</artifactId>
+      <version>${testcontainers.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.testcontainers</groupId>
+      <artifactId>junit-jupiter</artifactId>
+      <version>${testcontainers.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.postgresql</groupId>
+      <artifactId>postgresql</artifactId>
+      <version>${postgres.version}</version>
       <scope>test</scope>
     </dependency>
   </dependencies>
@@ -306,4 +328,4 @@
       <url>https://s01.oss.sonatype.org/service/local/staging/deploy/maven2/</url>
     </repository>
   </distributionManagement>
-</project>
\ No newline at end of file
+</project>
diff --git a/src/test/java/com/hsbc/engineering/PostgresConnectionTest.java b/src/test/java/com/hsbc/engineering/PostgresConnectionTest.java
new file mode 100644
index 0000000..79350f1
--- /dev/null
+++ b/src/test/java/com/hsbc/engineering/PostgresConnectionTest.java
@@ -0,0 +1,50 @@
+package com.hsbc.engineering;
+
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import java.util.Map;
+
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.testcontainers.containers.PostgreSQLContainer;
+import org.testcontainers.junit.jupiter.Container;
+import org.testcontainers.junit.jupiter.Testcontainers;
+
+@Testcontainers
+public class PostgresConnectionTest {
+    final String user       = "hello";
+    final String password   = "moo";
+    final String database   = "foo";
+
+    private Map<String, String> getConnectionMap() {
+        Map<String, String> args = Map.of(
+        Arguments.CLASS_NAME,   "org.postgresql.Driver",
+        Arguments.USER,         user,
+        Arguments.PWD,          password,
+        Arguments.URL,          pSqlContainer.getJdbcUrl(),
+        Arguments.SQL,          "SELECT 2");
+
+        return args;
+    }
+
+    @Container
+    private PostgreSQLContainer pSqlContainer = new PostgreSQLContainer<>("postgres:16")
+        .withDatabaseName(database)
+        .withUsername(user)
+        .withPassword(password);
+
+    @Test
+    void checkOK() {
+        assertTrue(pSqlContainer.isRunning());
+    }
+
+     @Test
+    void mainFunctionTest() {
+        Assertions.assertDoesNotThrow(() -> JDBCInquirer.runSimpleConnectionTest(getConnectionMap()));
+    }
+
+    @Test
+    void perfFunctionTest() {
+        Assertions.assertDoesNotThrow(() -> JDBCInquirer.timeExtractionTest(getConnectionMap()));
+    }
+}