db test fix

SoftwareDesignLab · Nov 28, 2023 · 704e7bd · 704e7bd
1 parent d003381
commit 704e7bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/db/src/test/java/edu/rit/se/nvip/db/repositories/RawDescriptionRepositoryTest.java b/db/src/test/java/edu/rit/se/nvip/db/repositories/RawDescriptionRepositoryTest.java
@@ -222,7 +222,7 @@ public void markGarbageTest() {
     @SneakyThrows
     public void getUsedRawVulnerabilitiesTest() {
         when(mockRS.next()).thenReturn(true, true, false);
-        when(mockRS.getInt(anyString())).thenReturn(1);
+        when(mockRS.getInt(anyString())).thenReturn(1,2);
         when(mockRS.getString(anyString())).thenReturn("desc");
         when(mockRS.getTimestamp(anyString())).thenReturn(new Timestamp(System.currentTimeMillis()));