From f15ed052607ae6e54e4e7f427c46ffd521091959 Mon Sep 17 00:00:00 2001
From: Carl Mastrangelo <notcarl@google.com>
Date: Fri, 2 Dec 2016 09:13:37 -0800
Subject: [PATCH] stub: remove a reference to internal

---
 .../test/java/io/grpc/benchmarks/driver/LoadWorkerTest.java   | 4 ++--
 stub/src/test/java/io/grpc/stub/ServerCallsTest.java          | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/benchmarks/src/test/java/io/grpc/benchmarks/driver/LoadWorkerTest.java b/benchmarks/src/test/java/io/grpc/benchmarks/driver/LoadWorkerTest.java
index 3459c8a96f..cfa06fe1b1 100644
--- a/benchmarks/src/test/java/io/grpc/benchmarks/driver/LoadWorkerTest.java
+++ b/benchmarks/src/test/java/io/grpc/benchmarks/driver/LoadWorkerTest.java
@@ -34,11 +34,11 @@ package io.grpc.benchmarks.driver;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
 
+import io.grpc.ManagedChannel;
 import io.grpc.benchmarks.Utils;
 import io.grpc.benchmarks.proto.Control;
 import io.grpc.benchmarks.proto.Stats;
 import io.grpc.benchmarks.proto.WorkerServiceGrpc;
-import io.grpc.internal.ManagedChannelImpl;
 import io.grpc.netty.NettyChannelBuilder;
 import io.grpc.stub.StreamObserver;
 
@@ -65,7 +65,7 @@ public class LoadWorkerTest {
       .build();
 
   private LoadWorker worker;
-  private ManagedChannelImpl channel;
+  private ManagedChannel channel;
   private WorkerServiceGrpc.WorkerServiceStub workerServiceStub;
   private LinkedBlockingQueue<Stats.ClientStats> marksQueue;
 
diff --git a/stub/src/test/java/io/grpc/stub/ServerCallsTest.java b/stub/src/test/java/io/grpc/stub/ServerCallsTest.java
index c6188dd7bd..d6c9ad6158 100644
--- a/stub/src/test/java/io/grpc/stub/ServerCallsTest.java
+++ b/stub/src/test/java/io/grpc/stub/ServerCallsTest.java
@@ -40,6 +40,7 @@ import static org.junit.Assert.fail;
 
 import io.grpc.CallOptions;
 import io.grpc.ClientCall;
+import io.grpc.ManagedChannel;
 import io.grpc.Metadata;
 import io.grpc.MethodDescriptor;
 import io.grpc.ServerCall;
@@ -49,7 +50,6 @@ import io.grpc.ServiceDescriptor;
 import io.grpc.Status;
 import io.grpc.inprocess.InProcessChannelBuilder;
 import io.grpc.inprocess.InProcessServerBuilder;
-import io.grpc.internal.ManagedChannelImpl;
 
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -332,7 +332,7 @@ public class ServerCallsTest {
         .build();
     long tag = System.nanoTime();
     InProcessServerBuilder.forName("go-with-the-flow" + tag).addService(service).build().start();
-    ManagedChannelImpl channel = InProcessChannelBuilder.forName("go-with-the-flow" + tag).build();
+    ManagedChannel channel = InProcessChannelBuilder.forName("go-with-the-flow" + tag).build();
     final ClientCall<Integer, Integer> clientCall = channel.newCall(STREAMING_METHOD,
         CallOptions.DEFAULT);
     final CountDownLatch latch = new CountDownLatch(1);