diff --git a/Lambdas/Lists/Item/test/TestItemGetter.java b/Lambdas/Lists/Item/test/TestItemGetter.java index 022c2bf..0e1302c 100644 --- a/Lambdas/Lists/Item/test/TestItemGetter.java +++ b/Lambdas/Lists/Item/test/TestItemGetter.java @@ -1,4 +1,6 @@ -import org.junit.jupiter.api.Test; +import org.junit.Test; +import org.mockito.Mockito; +import org.mockito.Mockito.*; import java.math.BigDecimal; import java.sql.SQLException; @@ -12,15 +14,15 @@ public class TestItemGetter { @Test public void testItemGetterValid() { - conductItemGetterTest(false); + conductItemGetterTestMock(false); } @Test public void testItemGetterError() { - conductItemGetterTest(true); + conductItemGetterTestMock(true); } - public void conductItemGetterTest(boolean shouldThrow) { + public void conductItemGetterTestMock(boolean shouldThrow) { ArrayList rsReturns = new ArrayList<>(); rsReturns.add(1);//ProductID rsReturns.add(2);//chainID @@ -37,7 +39,7 @@ public class TestItemGetter { } catch (SQLException throwables) { throwables.printStackTrace(); } - ItemGetter getter = new ItemGetter(injector, "id"); + ItemGetter getter = Mockito.spy(new ItemGetter(injector, "id")); Map ignore = new HashMap<>(); HashMap queryParams = TestInputUtils.addQueryParams(ignore); queryParams.put("id", "1");