Skip to content

Commit 2e99df8

Browse files
fix unit test
1 parent a1e9c09 commit 2e99df8

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

temporal-sdk/src/test/java/io/temporal/internal/sync/QueryDispatcherTest.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ public void testQuerySuccess() {
4949

5050
// Invoke functionality under test, expect no exceptions for an existing query.
5151
Optional<Payloads> queryResult =
52-
dispatcher.handleQuery(null, "QueryB", Header.empty(), Optional.empty());
52+
dispatcher.handleQuery(
53+
mock(SyncWorkflowContext.class), "QueryB", Header.empty(), Optional.empty());
5354
assertTrue(queryResult.isPresent());
5455
}
5556

@@ -61,7 +62,8 @@ public void testQueryDispatcherException() {
6162
assertThrows(
6263
IllegalArgumentException.class,
6364
() -> {
64-
dispatcher.handleQuery(null, "QueryC", Header.empty(), null);
65+
dispatcher.handleQuery(
66+
mock(SyncWorkflowContext.class), "QueryC", Header.empty(), null);
6567
});
6668
assertEquals("Unknown query type: QueryC, knownTypes=[QueryA, QueryB]", exception.getMessage());
6769
}

0 commit comments

Comments
 (0)