Fix mocking in a unit test

lutovich · lutovich · commit d67a5b34235a · 2018-09-28T10:57:59.000+02:00
diff --git a/driver/src/test/java/org/neo4j/driver/internal/async/inbound/InboundMessageDispatcherTest.java b/driver/src/test/java/org/neo4j/driver/internal/async/inbound/InboundMessageDispatcherTest.java
@@ -67,7 +67,7 @@ void shouldFailWhenCreatedWithNullChannel()
     @Test
     void shouldFailWhenCreatedWithNullLogging()
     {
-        assertThrows( NullPointerException.class, () -> new InboundMessageDispatcher( mock( Channel.class ), null ) );
+        assertThrows( NullPointerException.class, () -> new InboundMessageDispatcher( newChannelMock(), null ) );
     }
 
     @Test
@@ -109,7 +109,7 @@ void shouldDequeHandlerOnFailure()
     @Test
     void shouldSendResetOnFailure()
     {
-        Channel channel = mock( Channel.class );
+        Channel channel = newChannelMock();
         InboundMessageDispatcher dispatcher = newDispatcher( channel );
 
         dispatcher.enqueue( mock( ResponseHandler.class ) );

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@ void shouldFailWhenCreatedWithNullChannel()`
`67`	`67`	`@Test`
`68`	`68`	`void shouldFailWhenCreatedWithNullLogging()`
`69`	`69`	`{`
`70`		`- assertThrows( NullPointerException.class, () -> new InboundMessageDispatcher( mock( Channel.class ), null ) );`
	`70`	`+ assertThrows( NullPointerException.class, () -> new InboundMessageDispatcher( newChannelMock(), null ) );`
`71`	`71`	`}`
`72`	`72`
`73`	`73`	`@Test`
`@@ -109,7 +109,7 @@ void shouldDequeHandlerOnFailure()`
`109`	`109`	`@Test`
`110`	`110`	`void shouldSendResetOnFailure()`
`111`	`111`	`{`
`112`		`- Channel channel = mock( Channel.class );`
	`112`	`+ Channel channel = newChannelMock();`
`113`	`113`	`InboundMessageDispatcher dispatcher = newDispatcher( channel );`
`114`	`114`
`115`	`115`	`dispatcher.enqueue( mock( ResponseHandler.class ) );`