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 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 ignore = new HashMap<>(); + HashMap 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 listIDsReturn = (ArrayList) conductReturn; + System.out.println(listIDsReturn.toString()); + assert (listIDsReturn.toString().equals("[1, 2, 3, 4]")); + } catch (SQLException throwables) { + throwables.printStackTrace(); + assert shouldThrow; + } + } }