|
24 | 24 | import com.google.protobuf.GeneratedMessage; |
25 | 25 | import java.io.IOException; |
26 | 26 | import java.util.ArrayList; |
| 27 | +import java.util.Arrays; |
27 | 28 | import java.util.List; |
28 | | -import junit.framework.Assert; |
29 | 29 | import org.junit.After; |
30 | 30 | import org.junit.AfterClass; |
| 31 | +import org.junit.Assert; |
31 | 32 | import org.junit.Before; |
32 | 33 | import org.junit.BeforeClass; |
33 | 34 | import org.junit.Test; |
34 | 35 |
|
35 | 36 | @javax.annotation.Generated("by GAPIC") |
36 | 37 | public class ImageAnnotatorTest { |
| 38 | + private static MockImageAnnotator mockImageAnnotator; |
37 | 39 | private static MockServiceHelper serviceHelper; |
38 | 40 | private ImageAnnotatorApi api; |
39 | 41 |
|
40 | 42 | @BeforeClass |
41 | 43 | public static void startStaticServer() { |
42 | | - MockImageAnnotator mockService = new MockImageAnnotator(); |
43 | | - serviceHelper = new MockServiceHelper("in-process-1", mockService); |
| 44 | + mockImageAnnotator = new MockImageAnnotator(); |
| 45 | + serviceHelper = |
| 46 | + new MockServiceHelper("in-process-1", Arrays.<MockGrpcService>asList(mockImageAnnotator)); |
44 | 47 | serviceHelper.start(); |
45 | 48 | } |
46 | 49 |
|
@@ -70,16 +73,17 @@ public void batchAnnotateImagesTest() { |
70 | 73 | BatchAnnotateImagesResponse expectedResponse = BatchAnnotateImagesResponse.newBuilder().build(); |
71 | 74 | List<GeneratedMessage> expectedResponses = new ArrayList<>(); |
72 | 75 | expectedResponses.add(expectedResponse); |
73 | | - serviceHelper.getService().setResponses(expectedResponses); |
| 76 | + mockImageAnnotator.setResponses(expectedResponses); |
74 | 77 |
|
75 | 78 | List<AnnotateImageRequest> requests = new ArrayList<>(); |
| 79 | + |
76 | 80 | BatchAnnotateImagesResponse actualResponse = api.batchAnnotateImages(requests); |
77 | 81 | Assert.assertEquals(expectedResponse, actualResponse); |
78 | 82 |
|
79 | | - List<GeneratedMessage> actualRequests = serviceHelper.getService().getRequests(); |
| 83 | + List<GeneratedMessage> actualRequests = mockImageAnnotator.getRequests(); |
80 | 84 | Assert.assertEquals(1, actualRequests.size()); |
81 | 85 | BatchAnnotateImagesRequest actualRequest = (BatchAnnotateImagesRequest) actualRequests.get(0); |
82 | 86 |
|
83 | | - Assert.assertEquals(actualRequest.getRequestsList(), requests); |
| 87 | + Assert.assertEquals(requests, actualRequest.getRequestsList()); |
84 | 88 | } |
85 | 89 | } |
0 commit comments