1 files changed, 68 insertions, 0 deletions
diff --git a/src/test/java/com/orbekk/protobuf/ProtobufFunctionalTest.java b/src/test/java/com/orbekk/protobuf/ProtobufFunctionalTest.java
new file mode 100644
index 0000000..d60a016
--- /dev/null
+++ b/src/test/java/com/orbekk/protobuf/ProtobufFunctionalTest.java
@@ -0,0 +1,68 @@
+package com.orbekk.protobuf;
+
+import java.util.concurrent.CountDownLatch;
+
+import org.junit.Before;
+import static org.hamcrest.Matchers.*;
+import static org.junit.Assert.*;
+
+import com.google.protobuf.RpcCallback;
+import com.google.protobuf.RpcController;
+import com.orbekk.protobuf.Test.Type1;
+import com.orbekk.protobuf.Test.Type2;
+
+public class ProtobufFunctionalTest {
+    CountDownLatch returnC = new CountDownLatch(1);
+    SimpleProtobufServer server = SimpleProtobufServer.create(0);
+    int serverport = server.getPort();
+    RpcChannel channel = RpcChannel.create("localhost", serverport);
+    TestService directService = new TestService();
+    Test.Service service = Test.Service.newStub(channel);
+    
+    @Before public void setUp() {
+        server.registerService(directService);
+    }
+    
+    public class TestService extends Test.Service {
+        @Override
+        public void testA(RpcController controller, Type1 request,
+                RpcCallback<Type2> done) {
+            Type2 response = Type2.newBuilder()
+                    .setMessage("TestA")
+                    .build();
+            done.run(response);
+        }
+
+        @Override
+        public void testB(RpcController controller, Type1 request,
+                RpcCallback<Type2> done) {
+            controller.setFailed("error");
+        }
+
+        @Override
+        public void testC(RpcController controller, Type1 request,
+                RpcCallback<Type2> done) {
+            try {
+                returnC.await();
+            } catch (InterruptedException e) {
+            }
+            Type2 response = Type2.newBuilder()
+                    .setResult(request.getA() + request.getB())
+                    .setMessage("TestC result")
+                    .build();
+            done.run(response);
+        }
+    }
+    
+    @org.junit.Test public void respondsNormally() {
+        Test.Type1 request = Test.Type1.newBuilder().build();
+        int count = 10;
+        final CountDownLatch stop = new CountDownLatch(count);
+        service.testA(null, request, new RpcCallback<Type2>() {
+            @Override public void run(Type2 result) {
+                assertThat(result.getMessage(), is("TestA"));
+                stop.countDown();
+            }
+        });
+    }
+}