Lambda unit test infrastructure

Unit test infrastructure for testing GET and POST Lambdas

Lambdas/Lists/Item/src/Item.java

@@ -1,7 +1,6 @@
 import java.math.BigDecimal;
 import java.sql.ResultSet;
 import java.sql.SQLException;
-import java.time.LocalDateTime;
 
 public class Item {
     Integer productID;
@@ -11,7 +10,7 @@ public class Item {
     BigDecimal price;
     String imageURL;
     String department;
-    LocalDateTime retrievedDate;
+    long retrievedDate;
     Integer fetchCounts;
 
     Item(ResultSet itemRow) throws SQLException {
@@ -29,7 +28,7 @@ public class Item {
         System.out.println(imageURL);
         this.department = itemRow.getString(7);
         System.out.println(department);
-        this.retrievedDate = itemRow.getObject(8, LocalDateTime.class);
+        this.retrievedDate = itemRow.getTimestamp(8).toInstant().toEpochMilli();
         System.out.println(retrievedDate);
         this.fetchCounts = itemRow.getInt(9);
         System.out.println(fetchCounts);
@@ -106,11 +105,11 @@ public class Item {
         this.department = department;
     }
 
-    public LocalDateTime getRetrievedDate() {
+    public long getRetrievedDate() {
         return retrievedDate;
     }
 
-    public void setRetrievedDate(LocalDateTime retrievedDate) {
+    public void setRetrievedDate(long retrievedDate) {
         this.retrievedDate = retrievedDate;
     }
 

Lambdas/Lists/Item/src/ItemGetter.java

@@ -1,5 +1,3 @@
-import com.google.gson.Gson;
-
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
@@ -7,30 +5,25 @@ import java.sql.SQLException;
 import java.util.HashMap;
 import java.util.Map;
 
-public class ItemGetter implements CallHandler{
-    private final DBConnector connector;
+public class ItemGetter implements CallHandler {
+    private final Connection connection;
 
     private final String GET_ITEM = "SELECT * FROM Product WHERE productID = ?;";
 
-    public ItemGetter(DBConnector connector, String cognitoID) {
-        this.connector = connector;
+    public ItemGetter(Connection connection, String cognitoID) {
+        this.connection = connection;
     }
 
     @Override
     public Object conductAction(Map<String, Object> bodyMap, HashMap<String, String> queryMap, String cognitoID) throws SQLException {
-        Connection connection = connector.getConnection();
-        try {
-            PreparedStatement statement = connection.prepareStatement(GET_ITEM);
-            statement.setInt(1, Integer.parseInt(queryMap.get("id")));
-            System.out.println(statement);
-            ResultSet queryResults = statement.executeQuery();
-            queryResults.first();
-            System.out.println(queryResults);
-            Item retrievedItem = new Item(queryResults);
-            System.out.println(retrievedItem);
-            return retrievedItem;
-        } finally {
-            connection.close();
-        }
+        PreparedStatement statement = connection.prepareStatement(GET_ITEM);
+        statement.setInt(1, Integer.parseInt(queryMap.get("id")));
+        System.out.println(statement);
+        ResultSet queryResults = statement.executeQuery();
+        queryResults.first();
+        System.out.println(queryResults);
+        Item retrievedItem = new Item(queryResults);
+        System.out.println(retrievedItem);
+        return retrievedItem;
     }
 }

Lambdas/Lists/Item/test/TestItemGetter.java

@@ -0,0 +1,55 @@
+import org.junit.jupiter.api.Test;
+
+import java.math.BigDecimal;
+import java.sql.SQLException;
+import java.sql.Timestamp;
+import java.time.Instant;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
+
+public class TestItemGetter {
+
+    @Test
+    public void testItemGetterValid() {
+        conductItemGetterTest(false);
+    }
+
+    @Test
+    public void testItemGetterError() {
+        conductItemGetterTest(true);
+    }
+
+    public void conductItemGetterTest(boolean shouldThrow) {
+        ArrayList<Object> rsReturns = new ArrayList<>();
+        rsReturns.add(1);//ProductID
+        rsReturns.add(2);//chainID
+        rsReturns.add("aupc");
+        rsReturns.add("adescription");
+        rsReturns.add(BigDecimal.valueOf(.03));//Price
+        rsReturns.add("animageurl");
+        rsReturns.add("adepartment");
+        rsReturns.add(Timestamp.from(Instant.ofEpochMilli(1602192528688L)));//retrievedDate
+        rsReturns.add(5); // fetchCounts
+        StatementInjector injector = null;
+        try {
+            injector = new StatementInjector(null, rsReturns, shouldThrow);
+        } catch (SQLException throwables) {
+            throwables.printStackTrace();
+        }
+        ItemGetter getter = new ItemGetter(injector, "id");
+        Map<String, Object> ignore = new HashMap<>();
+        HashMap<String, String> queryParams = TestInputUtils.addQueryParams(ignore);
+        queryParams.put("id", "1");
+        try {
+            Object conductReturn = getter.conductAction(TestInputUtils.addBody(ignore), queryParams, "cognitoID");
+            assert !shouldThrow;
+            assert (conductReturn.getClass() == Item.class);
+            Item itemReturn = (Item) conductReturn;
+            assert (itemReturn.toString().equals("Item{productID=1, chainID=2, upc='aupc', description='adescription', price=0.03, imageURL='animageurl', department='adepartment', retrievedDate=1602192528688, fetchCounts=5}"));
+        } catch (SQLException throwables) {
+            throwables.printStackTrace();
+            assert shouldThrow;
+        }
+    }
+}