@Test public void writeInitializeMessageTest() throws IOException, InterruptedException, ExecutionException { Future<Boolean> future = this.messageWriter.writeInitializeMessage(InitializationInput.builder().shardId(shardId).build()); future.get(); verify(this.stream, Mockito.atLeastOnce()).write(Mockito.any(byte[].class), Mockito.anyInt(), Mockito.anyInt()); verify(this.stream, Mockito.atLeastOnce()).flush(); }
Future<Boolean> writeInitializeMessage(InitializationInput initializationInput) { return writeMessage(new InitializeMessage(initializationInput)); }
MessageWriter { Future<Boolean> writeInitializeMessage(InitializationInput initializationInput) { return writeMessage(new InitializeMessage(initializationInput)); } }
MessageWriter { Future<Boolean> writeInitializeMessage(InitializationInput initializationInput) { return writeMessage(new InitializeMessage(initializationInput)); } MessageWriter(); }
MessageWriter { Future<Boolean> writeInitializeMessage(InitializationInput initializationInput) { return writeMessage(new InitializeMessage(initializationInput)); } MessageWriter();  }
MessageWriter { Future<Boolean> writeInitializeMessage(InitializationInput initializationInput) { return writeMessage(new InitializeMessage(initializationInput)); } MessageWriter();   }