Skip to content

Commit bdab866

Browse files
authored
Fix workflowEventConsumer test (#25442)
1 parent 8470c1a commit bdab866

1 file changed

Lines changed: 11 additions & 8 deletions

File tree

openmetadata-service/src/test/java/org/openmetadata/service/governance/workflows/WorkflowEventConsumerTest.java

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434

3535
import java.lang.reflect.Method;
3636
import java.sql.SQLException;
37+
import java.util.Collections;
3738
import java.util.UUID;
3839
import java.util.concurrent.atomic.AtomicInteger;
3940
import org.apache.ibatis.exceptions.PersistenceException;
@@ -89,7 +90,7 @@ void testSendMessage_SkipsGovernanceBotEvents() throws Exception {
8990
try (MockedStatic<WorkflowHandler> mockedHandler = mockStatic(WorkflowHandler.class)) {
9091
mockedHandler.when(WorkflowHandler::getInstance).thenReturn(workflowHandler);
9192

92-
assertDoesNotThrow(() -> consumer.sendMessage(event));
93+
assertDoesNotThrow(() -> consumer.sendMessage(event, Collections.emptySet()));
9394

9495
verify(workflowHandler, never()).triggerWithSignal(anyString(), anyMap());
9596
}
@@ -103,7 +104,7 @@ void testSendMessage_SkipsGovernanceBotImpersonatedEvents() throws Exception {
103104
try (MockedStatic<WorkflowHandler> mockedHandler = mockStatic(WorkflowHandler.class)) {
104105
mockedHandler.when(WorkflowHandler::getInstance).thenReturn(workflowHandler);
105106

106-
assertDoesNotThrow(() -> consumer.sendMessage(event));
107+
assertDoesNotThrow(() -> consumer.sendMessage(event, Collections.emptySet()));
107108

108109
verify(workflowHandler, never()).triggerWithSignal(anyString(), anyMap());
109110
}
@@ -124,7 +125,7 @@ void testSendMessage_SuccessfulTrigger() throws Exception {
124125

125126
doNothing().when(workflowHandler).triggerWithSignal(anyString(), anyMap());
126127

127-
assertDoesNotThrow(() -> consumer.sendMessage(event));
128+
assertDoesNotThrow(() -> consumer.sendMessage(event, Collections.emptySet()));
128129

129130
verify(workflowHandler, times(1)).triggerWithSignal(eq("table-entityUpdated"), anyMap());
130131
}
@@ -157,7 +158,7 @@ void testSendMessage_RetriesOnDeadlock() throws Exception {
157158
.when(workflowHandler)
158159
.triggerWithSignal(anyString(), anyMap());
159160

160-
assertDoesNotThrow(() -> consumer.sendMessage(event));
161+
assertDoesNotThrow(() -> consumer.sendMessage(event, Collections.emptySet()));
161162

162163
assertEquals(3, callCount.get());
163164
}
@@ -189,7 +190,7 @@ void testSendMessage_RetriesOnOptimisticLockingFailure() throws Exception {
189190
.when(workflowHandler)
190191
.triggerWithSignal(anyString(), anyMap());
191192

192-
assertDoesNotThrow(() -> consumer.sendMessage(event));
193+
assertDoesNotThrow(() -> consumer.sendMessage(event, Collections.emptySet()));
193194

194195
assertEquals(2, callCount.get());
195196
}
@@ -212,7 +213,8 @@ void testSendMessage_DoesNotRetryNonTransientErrors() throws Exception {
212213
.when(workflowHandler)
213214
.triggerWithSignal(anyString(), anyMap());
214215

215-
assertThrows(EventPublisherException.class, () -> consumer.sendMessage(event));
216+
assertThrows(
217+
EventPublisherException.class, () -> consumer.sendMessage(event, Collections.emptySet()));
216218

217219
verify(workflowHandler, times(1)).triggerWithSignal(anyString(), anyMap());
218220
}
@@ -241,7 +243,8 @@ void testSendMessage_FailsAfterMaxRetries() throws Exception {
241243
.when(workflowHandler)
242244
.triggerWithSignal(anyString(), anyMap());
243245

244-
assertThrows(EventPublisherException.class, () -> consumer.sendMessage(event));
246+
assertThrows(
247+
EventPublisherException.class, () -> consumer.sendMessage(event, Collections.emptySet()));
245248

246249
assertEquals(3, callCount.get());
247250
}
@@ -254,7 +257,7 @@ void testSendMessage_SkipsInvalidEventTypes() throws Exception {
254257
try (MockedStatic<WorkflowHandler> mockedHandler = mockStatic(WorkflowHandler.class)) {
255258
mockedHandler.when(WorkflowHandler::getInstance).thenReturn(workflowHandler);
256259

257-
assertDoesNotThrow(() -> consumer.sendMessage(event));
260+
assertDoesNotThrow(() -> consumer.sendMessage(event, Collections.emptySet()));
258261

259262
verify(workflowHandler, never()).triggerWithSignal(anyString(), anyMap());
260263
}

0 commit comments

Comments
 (0)