| 117 | |
| 118 | doAnswer(new Answer<ConnectionClientTransport>() { |
| 119 | @Override |
| 120 | public ConnectionClientTransport answer(InvocationOnMock invocation) throws Throwable { |
| 121 | final ConnectionClientTransport mockTransport = mock(ConnectionClientTransport.class); |
| 122 | when(mockTransport.getLogId()) |
| 123 | .thenReturn(InternalLogId.allocate("mocktransport", /*details=*/ null)); |
| 124 | when(mockTransport.newStream( |
| 125 | any(MethodDescriptor.class), any(Metadata.class), any(CallOptions.class), |
| 126 | ArgumentMatchers.<ClientStreamTracer[]>any())) |
| 127 | .thenReturn(mock(ClientStream.class)); |
| 128 | // Save the listener |
| 129 | doAnswer(new Answer<Runnable>() { |
| 130 | @Override |
| 131 | public Runnable answer(InvocationOnMock invocation) throws Throwable { |
| 132 | captor.add(new MockClientTransportInfo( |
| 133 | mockTransport, (ManagedClientTransport.Listener) invocation.getArguments()[0])); |
| 134 | return startRunnable; |
| 135 | } |
| 136 | }).when(mockTransport).start(any(ManagedClientTransport.Listener.class)); |
| 137 | return mockTransport; |
| 138 | } |
| 139 | }).when(mockTransportFactory) |
| 140 | .newClientTransport( |
| 141 | any(SocketAddress.class), |