|
34 | 34 | import static org.mockito.Mockito.mock; |
35 | 35 | import static org.mockito.Mockito.never; |
36 | 36 | import static org.mockito.Mockito.only; |
| 37 | +import static org.mockito.Mockito.times; |
37 | 38 | import static org.mockito.Mockito.verify; |
38 | 39 | import static org.mockito.Mockito.when; |
39 | 40 | import static org.neo4j.driver.Values.value; |
@@ -365,24 +366,41 @@ void shouldCreateChannelActivityLoggerAndLogDebugMessageOnMessageHandling(Class< |
365 | 366 | InboundMessageDispatcher dispatcher = new InboundMessageDispatcher(channel, logging); |
366 | 367 | ResponseHandler handler = mock(ResponseHandler.class); |
367 | 368 | dispatcher.enqueue(handler); |
| 369 | + Runnable loggerVerification = () -> {}; |
368 | 370 |
|
369 | 371 | // WHEN |
370 | 372 | if (SuccessMessage.class.isAssignableFrom(message)) { |
371 | 373 | dispatcher.handleSuccessMessage(new HashMap<>()); |
| 374 | + loggerVerification = () -> { |
| 375 | + verify(logger).isDebugEnabled(); |
| 376 | + verify(logger).debug(anyString(), any(Map.class)); |
| 377 | + }; |
372 | 378 | } else if (FailureMessage.class.isAssignableFrom(message)) { |
373 | 379 | dispatcher.handleFailureMessage(FAILURE_CODE, FAILURE_MESSAGE); |
| 380 | + loggerVerification = () -> { |
| 381 | + verify(logger).isDebugEnabled(); |
| 382 | + verify(logger).debug(anyString(), anyString(), anyString()); |
| 383 | + }; |
374 | 384 | } else if (RecordMessage.class.isAssignableFrom(message)) { |
375 | 385 | dispatcher.handleRecordMessage(Values.values()); |
| 386 | + loggerVerification = () -> { |
| 387 | + verify(logger, times(2)).isDebugEnabled(); |
| 388 | + verify(logger).debug(anyString(), anyString()); |
| 389 | + }; |
376 | 390 | } else if (IgnoredMessage.class.isAssignableFrom(message)) { |
377 | 391 | dispatcher.handleIgnoredMessage(); |
| 392 | + loggerVerification = () -> { |
| 393 | + verify(logger).isDebugEnabled(); |
| 394 | + verify(logger).debug(anyString()); |
| 395 | + }; |
378 | 396 | } else { |
379 | 397 | fail("Unexpected message type parameter provided"); |
380 | 398 | } |
381 | 399 |
|
382 | 400 | // THEN |
383 | 401 | assertTrue(dispatcher.getLog() instanceof ChannelActivityLogger); |
384 | 402 | assertTrue(dispatcher.getErrorLog() instanceof ChannelErrorLogger); |
385 | | - verify(logger).debug(anyString(), any(Object.class)); |
| 403 | + loggerVerification.run(); |
386 | 404 | } |
387 | 405 |
|
388 | 406 | @Test |
|
0 commit comments