From d14383ff2ec82ba84c148a6c5bdc80ddd5326d05 Mon Sep 17 00:00:00 2001
From: NMerz <nmerz@icloud.com>
Date: Thu, 8 Oct 2020 19:07:02 -0400
Subject: [PATCH] Test id retrieval

---
 Lambdas/Lists/List/test/TestListGetter.java | 42 +++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/Lambdas/Lists/List/test/TestListGetter.java b/Lambdas/Lists/List/test/TestListGetter.java
index 9f9ce56..2c563d4 100644
--- a/Lambdas/Lists/List/test/TestListGetter.java
+++ b/Lambdas/Lists/List/test/TestListGetter.java
@@ -14,6 +14,16 @@ public class TestListGetter {
         conductListGetterTest(false);
     }
 
+    @Test
+    public void testListIDGetterValid() {
+        conductListIDGetterTest(false);
+    }
+
+    @Test
+    public void testListIDGetterError() {
+        conductListIDGetterTest(false);
+    }
+
     @Test
     public void testListGetterError() {
         conductListGetterTest(true);
@@ -60,4 +70,36 @@ public class TestListGetter {
             assert shouldThrow;
         }
     }
+
+    public void conductListIDGetterTest(boolean shouldThrow) {
+
+
+        ArrayList<Object> rsReturns = new ArrayList<>();
+        rsReturns.add(1);
+        rsReturns.add(2);
+        rsReturns.add(3);
+        rsReturns.add(4);
+
+        StatementInjector injector = null;
+        try {
+            injector = new StatementInjector(null, rsReturns, shouldThrow);
+        } catch (SQLException throwables) {
+            throwables.printStackTrace();
+        }
+        ListGetter getter = new ListGetter(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() == ArrayList.class);
+            ArrayList<Integer> listIDsReturn = (ArrayList<Integer>) conductReturn;
+            System.out.println(listIDsReturn.toString());
+            assert (listIDsReturn.toString().equals("[1, 2, 3, 4]"));
+        } catch (SQLException throwables) {
+            throwables.printStackTrace();
+            assert shouldThrow;
+        }
+    }
 }