Jira Iterator HasNext test

Signed-off-by: Maxwell Brown <mxwelwbr@amazon.com>
opensearch-project · Nov 8, 2024 · 7a85756 · 7a85756
1 parent b9c935a
commit 7a85756
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...source/src/test/java/org/opensearch/dataprepper/plugins/source/jira/JiraIteratorTest.java b/...source/src/test/java/org/opensearch/dataprepper/plugins/source/jira/JiraIteratorTest.java
@@ -99,6 +99,18 @@ void testItemInfoQueueNotEmpty() {
         jiraIterator.setCrawlerQWaitTimeMillis(1);
         assertTrue(jiraIterator.hasNext());
         assertNotNull(jiraIterator.next());
+    }
+
+    @Test
+    void testItemInfoQueueEmpty(){
+        jiraIterator = createObjectUnderTest();
+        List<IssueBean> mockIssues = new ArrayList<>();
+        when(mockSearchResults.getIssues()).thenReturn(mockIssues);
+        when(mockSearchResults.getTotal()).thenReturn(0);
+        doReturn(mockSearchResults).when(jiraRestClient).getAllIssues(any(StringBuilder.class), anyInt(), any(JiraSourceConfig.class));
+
+        jiraIterator.initialize(Instant.ofEpochSecond(0));
+        jiraIterator.setCrawlerQWaitTimeMillis(1);
         assertFalse(jiraIterator.hasNext());
         assertThrows(NoSuchElementException.class, () -> jiraIterator.next());
     }