Index: src/test/java/com/napa/pulse/dao/impl/SessionDAOImplUnitTest.java =================================================================== diff -u -refffdd086e6950d3a3c1ba77ec94c87d8bb6a83c -rd6fe4609bafd690e3ee4121da7784812957b1bfe --- src/test/java/com/napa/pulse/dao/impl/SessionDAOImplUnitTest.java (.../SessionDAOImplUnitTest.java) (revision efffdd086e6950d3a3c1ba77ec94c87d8bb6a83c) +++ src/test/java/com/napa/pulse/dao/impl/SessionDAOImplUnitTest.java (.../SessionDAOImplUnitTest.java) (revision d6fe4609bafd690e3ee4121da7784812957b1bfe) @@ -142,9 +142,8 @@ row.put("isStockEditOff", "true"); expectedResult.add(row); when(jdbcTemplate.queryForList(anyString())).thenReturn(expectedResult); - List> result = sessionDAOImplMock.getMinMaxSessionUnselectedProducts(sessionId, siteId); - assertNotNull(result); - assertEquals(expectedResult, result); + List> result = sessionDAOImplMock.getMinMaxSessionUnselectedProducts(sessionId, siteId);assertNotNull(result); + assertEquals(expectedResult, result); verify(jdbcTemplate, times(1)).queryForList(anyString()); } @@ -274,15 +273,14 @@ Map keyMap = new HashMap<>(); keyMap.put("SESSION_ID", BigDecimal.valueOf(1234)); - // Act + doAnswer(invocation -> { KeyHolder keyHolderArg = invocation.getArgument(1); - // You can set a dummy key here to simulate a successful insertion + keyHolderArg.getKeyList().add(keyMap); - // keyHolderArg.getKeyList().add(Collections.singletonMap("SESSION_ID", BigDecimal.valueOf(1234))); // Simulated session ID return 1;