1fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Protocol Buffers - Google's data interchange format 2fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Copyright 2008 Google Inc. All rights reserved. 3afb4b72037e3f13db208590fc782c4bc8e27f862Jeff Davidson// https://developers.google.com/protocol-buffers/ 4fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 5fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Redistribution and use in source and binary forms, with or without 6fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modification, are permitted provided that the following conditions are 7fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// met: 8fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 9fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Redistributions of source code must retain the above copyright 10fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// notice, this list of conditions and the following disclaimer. 11fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Redistributions in binary form must reproduce the above 12fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// copyright notice, this list of conditions and the following disclaimer 13fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// in the documentation and/or other materials provided with the 14fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// distribution. 15fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Neither the name of Google Inc. nor the names of its 16fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contributors may be used to endorse or promote products derived from 17fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// this software without specific prior written permission. 18fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 19fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 31fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillepackage com.google.protobuf; 32fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 33d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Savilleimport com.google.protobuf.Descriptors.FileDescriptor; 34fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport com.google.protobuf.Descriptors.MethodDescriptor; 35d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Savilleimport google.protobuf.no_generic_services_test.UnittestNoGenericServices; 36fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.MessageWithNoOuter; 37fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.ServiceWithNoOuter; 38fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.UnittestProto.TestAllTypes; 39fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.UnittestProto.TestService; 40fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.UnittestProto.FooRequest; 41fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.UnittestProto.FooResponse; 42fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.UnittestProto.BarRequest; 43fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport protobuf_unittest.UnittestProto.BarResponse; 44fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 45fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport org.easymock.classextension.EasyMock; 46fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport org.easymock.classextension.IMocksControl; 47fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport org.easymock.IArgumentMatcher; 48fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 49d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Savilleimport java.util.HashSet; 50d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Savilleimport java.util.Set; 51d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 52fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleimport junit.framework.TestCase; 53fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 54fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville/** 55fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville * Tests services and stubs. 56fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville * 57fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville * @author kenton@google.com Kenton Varda 58fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville */ 59fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillepublic class ServiceTest extends TestCase { 60fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private IMocksControl control; 61fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private RpcController mockController; 62fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 63fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private final Descriptors.MethodDescriptor fooDescriptor = 64fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService.getDescriptor().getMethods().get(0); 65fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private final Descriptors.MethodDescriptor barDescriptor = 66fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService.getDescriptor().getMethods().get(1); 67fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 68fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville @Override 69fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville protected void setUp() throws Exception { 70fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville super.setUp(); 71fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control = EasyMock.createStrictControl(); 72fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockController = control.createMock(RpcController.class); 73fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 74fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 75fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // ================================================================= 76fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 77fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** Tests Service.callMethod(). */ 78fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void testCallMethod() throws Exception { 79fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FooRequest fooRequest = FooRequest.newBuilder().build(); 80fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BarRequest barRequest = BarRequest.newBuilder().build(); 81fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockCallback<Message> fooCallback = new MockCallback<Message>(); 82fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockCallback<Message> barCallback = new MockCallback<Message>(); 83fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService mockService = control.createMock(TestService.class); 84fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 85fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockService.foo(EasyMock.same(mockController), EasyMock.same(fooRequest), 86fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville this.<FooResponse>wrapsCallback(fooCallback)); 87fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockService.bar(EasyMock.same(mockController), EasyMock.same(barRequest), 88fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville this.<BarResponse>wrapsCallback(barCallback)); 89fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.replay(); 90fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 91fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockService.callMethod(fooDescriptor, mockController, 92fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville fooRequest, fooCallback); 93fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockService.callMethod(barDescriptor, mockController, 94fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville barRequest, barCallback); 95fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.verify(); 96fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 97fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 98fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** Tests Service.get{Request,Response}Prototype(). */ 99fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void testGetPrototype() throws Exception { 100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService mockService = control.createMock(TestService.class); 101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertSame(mockService.getRequestPrototype(fooDescriptor), 103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FooRequest.getDefaultInstance()); 104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertSame(mockService.getResponsePrototype(fooDescriptor), 105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FooResponse.getDefaultInstance()); 106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertSame(mockService.getRequestPrototype(barDescriptor), 107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BarRequest.getDefaultInstance()); 108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertSame(mockService.getResponsePrototype(barDescriptor), 109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BarResponse.getDefaultInstance()); 110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** Tests generated stubs. */ 113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void testStub() throws Exception { 114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FooRequest fooRequest = FooRequest.newBuilder().build(); 115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BarRequest barRequest = BarRequest.newBuilder().build(); 116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockCallback<FooResponse> fooCallback = new MockCallback<FooResponse>(); 117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockCallback<BarResponse> barCallback = new MockCallback<BarResponse>(); 118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcChannel mockChannel = control.createMock(RpcChannel.class); 119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService stub = TestService.newStub(mockChannel); 120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockChannel.callMethod( 122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(fooDescriptor), 123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(mockController), 124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(fooRequest), 125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(FooResponse.getDefaultInstance()), 126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville this.<Message>wrapsCallback(fooCallback)); 127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville mockChannel.callMethod( 128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(barDescriptor), 129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(mockController), 130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(barRequest), 131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(BarResponse.getDefaultInstance()), 132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville this.<Message>wrapsCallback(barCallback)); 133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.replay(); 134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville stub.foo(mockController, fooRequest, fooCallback); 136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville stub.bar(mockController, barRequest, barCallback); 137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.verify(); 138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** Tests generated blocking stubs. */ 141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void testBlockingStub() throws Exception { 142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FooRequest fooRequest = FooRequest.newBuilder().build(); 143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BarRequest barRequest = BarRequest.newBuilder().build(); 144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BlockingRpcChannel mockChannel = 145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.createMock(BlockingRpcChannel.class); 146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService.BlockingInterface stub = 147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestService.newBlockingStub(mockChannel); 148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FooResponse fooResponse = FooResponse.newBuilder().build(); 150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BarResponse barResponse = BarResponse.newBuilder().build(); 151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.expect(mockChannel.callBlockingMethod( 153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(fooDescriptor), 154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(mockController), 155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(fooRequest), 156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(FooResponse.getDefaultInstance()))).andReturn(fooResponse); 157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.expect(mockChannel.callBlockingMethod( 158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(barDescriptor), 159fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(mockController), 160fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(barRequest), 161fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(BarResponse.getDefaultInstance()))).andReturn(barResponse); 162fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.replay(); 163fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 164fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertSame(fooResponse, stub.foo(mockController, fooRequest)); 165fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertSame(barResponse, stub.bar(mockController, barRequest)); 166fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.verify(); 167fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 168fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 169fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void testNewReflectiveService() { 170fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ServiceWithNoOuter.Interface impl = 171fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.createMock(ServiceWithNoOuter.Interface.class); 172fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcController controller = control.createMock(RpcController.class); 173fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville Service service = ServiceWithNoOuter.newReflectiveService(impl); 174fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 175fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MethodDescriptor fooMethod = 176fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ServiceWithNoOuter.getDescriptor().findMethodByName("Foo"); 177fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MessageWithNoOuter request = MessageWithNoOuter.getDefaultInstance(); 178fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcCallback<Message> callback = new RpcCallback<Message>() { 179fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void run(Message parameter) { 180fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // No reason this should be run. 181fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville fail(); 182fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 183fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville }; 184fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcCallback<TestAllTypes> specializedCallback = 185fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcUtil.specializeCallback(callback); 186fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 187fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville impl.foo(EasyMock.same(controller), EasyMock.same(request), 188fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.same(specializedCallback)); 189fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.expectLastCall(); 190fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 191fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.replay(); 192fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 193fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville service.callMethod(fooMethod, controller, request, callback); 194fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 195fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.verify(); 196fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 197fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 198fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void testNewReflectiveBlockingService() throws ServiceException { 199fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ServiceWithNoOuter.BlockingInterface impl = 200fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.createMock(ServiceWithNoOuter.BlockingInterface.class); 201fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcController controller = control.createMock(RpcController.class); 202fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville BlockingService service = 203fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ServiceWithNoOuter.newReflectiveBlockingService(impl); 204fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 205fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MethodDescriptor fooMethod = 206fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ServiceWithNoOuter.getDescriptor().findMethodByName("Foo"); 207fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MessageWithNoOuter request = MessageWithNoOuter.getDefaultInstance(); 208fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 209fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestAllTypes expectedResponse = TestAllTypes.getDefaultInstance(); 210fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.expect(impl.foo(EasyMock.same(controller), EasyMock.same(request))) 211fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville .andReturn(expectedResponse); 212fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 213fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.replay(); 214fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 215fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville Message response = 216fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville service.callBlockingMethod(fooMethod, controller, request); 217fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville assertEquals(expectedResponse, response); 218fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 219fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville control.verify(); 220fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 221fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 222d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville public void testNoGenericServices() throws Exception { 223d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // Non-services should be usable. 224d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville UnittestNoGenericServices.TestMessage message = 225d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville UnittestNoGenericServices.TestMessage.newBuilder() 226d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville .setA(123) 227d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville .setExtension(UnittestNoGenericServices.testExtension, 456) 228d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville .build(); 229d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertEquals(123, message.getA()); 230d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertEquals(1, UnittestNoGenericServices.TestEnum.FOO.getNumber()); 231d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 232d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // Build a list of the class names nested in UnittestNoGenericServices. 233d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville String outerName = "google.protobuf.no_generic_services_test." + 234d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville "UnittestNoGenericServices"; 235d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville Class<?> outerClass = Class.forName(outerName); 236d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 237d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville Set<String> innerClassNames = new HashSet<String>(); 238d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville for (Class<?> innerClass : outerClass.getClasses()) { 239d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville String fullName = innerClass.getName(); 240d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // Figure out the unqualified name of the inner class. 241d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // Note: Surprisingly, the full name of an inner class will be separated 242d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // from the outer class name by a '$' rather than a '.'. This is not 243d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // mentioned in the documentation for java.lang.Class. I don't want to 244d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // make assumptions, so I'm just going to accept any character as the 245d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // separator. 246d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertTrue(fullName.startsWith(outerName)); 247a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 248a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (!Service.class.isAssignableFrom(innerClass) && 249a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson !Message.class.isAssignableFrom(innerClass) && 250a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson !ProtocolMessageEnum.class.isAssignableFrom(innerClass)) { 251a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // Ignore any classes not generated by the base code generator. 252a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson continue; 253a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 254a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 255d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville innerClassNames.add(fullName.substring(outerName.length() + 1)); 256d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville } 257d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 258d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // No service class should have been generated. 259d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertTrue(innerClassNames.contains("TestMessage")); 260d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertTrue(innerClassNames.contains("TestEnum")); 261d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertFalse(innerClassNames.contains("TestService")); 262d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 263d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville // But descriptors are there. 264d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville FileDescriptor file = UnittestNoGenericServices.getDescriptor(); 265d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertEquals(1, file.getServices().size()); 266d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertEquals("TestService", file.getServices().get(0).getName()); 267d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertEquals(1, file.getServices().get(0).getMethods().size()); 268d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville assertEquals("Foo", 269d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville file.getServices().get(0).getMethods().get(0).getName()); 270d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville } 271d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 272fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // ================================================================= 273fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 274fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** 275fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville * wrapsCallback() is an EasyMock argument predicate. wrapsCallback(c) 276fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville * matches a callback if calling that callback causes c to be called. 277fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville * In other words, c wraps the given callback. 278fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville */ 279fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private <Type extends Message> RpcCallback<Type> wrapsCallback( 280fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockCallback<?> callback) { 281fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EasyMock.reportMatcher(new WrapsCallback(callback)); 282fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return null; 283fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 284fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 285fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** The parameter to wrapsCallback() must be a MockCallback. */ 286fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private static class MockCallback<Type extends Message> 287fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville implements RpcCallback<Type> { 288fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private boolean called = false; 289fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 290fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public boolean isCalled() { return called; } 291fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 292fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void reset() { called = false; } 293fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void run(Type message) { called = true; } 294fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 295fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 296fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville /** Implementation of the wrapsCallback() argument matcher. */ 297fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private static class WrapsCallback implements IArgumentMatcher { 298fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private MockCallback<?> callback; 299fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 300fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public WrapsCallback(MockCallback<?> callback) { 301fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville this.callback = callback; 302fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 303fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 304fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville @SuppressWarnings("unchecked") 305fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public boolean matches(Object actual) { 306fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville if (!(actual instanceof RpcCallback)) { 307fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return false; 308fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 309fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville RpcCallback actualCallback = (RpcCallback)actual; 310fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 311fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville callback.reset(); 312fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville actualCallback.run(null); 313fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return callback.isCalled(); 314fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 315fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 316fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public void appendTo(StringBuffer buffer) { 317fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville buffer.append("wrapsCallback(mockCallback)"); 318fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 319fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 320fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} 321