From ba653d4bd0f1ed212e8dc09e51ca5ffc4b06b5c8 Mon Sep 17 00:00:00 2001 From: ejona Date: Fri, 7 Nov 2014 10:26:37 -0800 Subject: [PATCH] Move StubConfigTest to third_party ------------- Created by MOE: http://code.google.com/p/moe-java MOE_MIGRATED_REVID=79430433 --- .../net/stubby/stub/StubConfigTest.java | 48 +++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 integration-testing/src/test/java/com/google/net/stubby/stub/StubConfigTest.java diff --git a/integration-testing/src/test/java/com/google/net/stubby/stub/StubConfigTest.java b/integration-testing/src/test/java/com/google/net/stubby/stub/StubConfigTest.java new file mode 100644 index 0000000000..0170003a32 --- /dev/null +++ b/integration-testing/src/test/java/com/google/net/stubby/stub/StubConfigTest.java @@ -0,0 +1,48 @@ +package com.google.net.stubby.stub; + +import static org.junit.Assert.assertEquals; + +import com.google.net.stubby.Call; +import com.google.net.stubby.Channel; +import com.google.net.stubby.MethodDescriptor; +import com.google.net.stubby.testing.integration.TestServiceGrpc; + +import org.junit.Test; +import org.junit.runner.RunWith; +import org.junit.runners.JUnit4; + +import java.util.concurrent.TimeUnit; + +/** + * Tests for stub reconfiguration + */ +@RunWith(JUnit4.class) +public class StubConfigTest { + + @Test + public void testConfigureTimeout() { + // Create a default stub + TestServiceGrpc.TestServiceBlockingStub stub = + TestServiceGrpc.newBlockingStub(new FakeChannel()); + assertEquals(TimeUnit.SECONDS.toMicros(1), + stub.getServiceDescriptor().fullDuplexCall.getTimeout()); + // Reconfigure it + stub = stub.configureNewStub() + .setTimeout(2, TimeUnit.SECONDS) + .build(); + // New altered config + assertEquals(TimeUnit.SECONDS.toMicros(2), + stub.getServiceDescriptor().fullDuplexCall.getTimeout()); + // Default config unchanged + assertEquals(TimeUnit.SECONDS.toMicros(1), + TestServiceGrpc.CONFIG.fullDuplexCall.getTimeout()); + } + + + private static class FakeChannel implements Channel { + @Override + public Call newCall(MethodDescriptor method) { + return null; + } + } +}