From 8385888075b8ddb17279ced8a754c0c26ce3341f Mon Sep 17 00:00:00 2001
From: Adam Ding <ding183@purdue.edu>
Date: Tue, 1 Dec 2020 21:00:10 -0500
Subject: [PATCH] ChainGetter tests added

---
 Lambdas/Lists/Chain/test/TestChainGetter.java | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 Lambdas/Lists/Chain/test/TestChainGetter.java

diff --git a/Lambdas/Lists/Chain/test/TestChainGetter.java b/Lambdas/Lists/Chain/test/TestChainGetter.java
new file mode 100644
index 0000000..b217e2f
--- /dev/null
+++ b/Lambdas/Lists/Chain/test/TestChainGetter.java
@@ -0,0 +1,43 @@
+import org.junit.Test;
+import org.mockito.Mockito;
+
+import java.sql.SQLException;
+import java.util.HashMap;
+import java.util.Map;
+
+public class TestChainGetter {
+
+    @Test
+    public void testChainGetterValid() {
+        testChainGetter(false);
+    }
+
+    @Test
+    public void testChainGetterError() {
+        testChainGetter(true);
+    }
+
+    public void testChainGetter(boolean shouldThrow) {
+        StatementInjector injector;
+        try {
+            injector = new StatementInjector(null, null, shouldThrow);
+        } catch (SQLException throwables) {
+            throwables.printStackTrace();
+            assert false; //Error in test infrastructure
+            return;
+        }
+        ChainGetter chainGetter = Mockito.spy(new ChainGetter(injector, "cognitoID"));
+        Map<String, Object> ignore = new HashMap<>();
+        Map<String, Object> body = TestInputUtils.addBody(ignore);
+        ignore.put("id", 1);    //in ChainGetter.java uses ignore map for id parameter
+
+        try {
+            Object rawIDReturn = chainGetter.conductAction(body, TestInputUtils.addQueryParams(ignore), "cognitoID");
+            assert !shouldThrow;
+            assert (rawIDReturn != null);
+        } catch (SQLException throwables) {
+            assert shouldThrow;
+            throwables.printStackTrace();
+        }
+    }
+}