/** Resets the state of the mock service. */ public void reset() { for (MockGrpcService mockService : mockServices) { mockService.reset(); } }
@Test public void testReset() { MockServiceHelper serviceHelper = new MockServiceHelper("fake-address", grpcService); serviceHelper.reset(); verify(grpcService, times(1)).getServiceDefinition(); verify(grpcService, times(1)).reset(); }
/** Sets up mocks. */ @Before public void setUp() { MockitoAnnotations.initMocks(this); when(grpcService.getServiceDefinition()) .thenReturn(ServerServiceDefinition.builder("fake-service").build()); when(grpcService2.getServiceDefinition()) .thenReturn(ServerServiceDefinition.builder("fake-service2").build()); }
public MockServiceHelper(String addressString, List<MockGrpcService> mockServices) { this.addressString = addressString; this.mockServices = Lists.newArrayList(mockServices); InProcessServerBuilder builder = InProcessServerBuilder.forName(addressString); for (MockGrpcService mockService : mockServices) { builder.addService(mockService.getServiceDefinition()); } this.server = builder.build(); }