Use passthrough resolver when bufnet is used

open-telemetry · May 29, 2024 · 255855e · 255855e
1 parent 1e8c7ca
commit 255855e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/instrumentation/google.golang.org/grpc/otelgrpc/benchmark_test.go b/instrumentation/google.golang.org/grpc/otelgrpc/benchmark_test.go
@@ -39,7 +39,7 @@ func benchmark(b *testing.B, cOpt []grpc.DialOption, sOpt []grpc.ServerOption) {
 	ctx := context.Background()
 	dial := func(context.Context, string) (net.Conn, error) { return l.Dial() }
 	conn, err := grpc.NewClient(
-		"bufnet",
+		"passthrough:bufnet",
 		append([]grpc.DialOption{
 			grpc.WithContextDialer(dial),
 			grpc.WithTransportCredentials(insecure.NewCredentials()),

diff --git a/instrumentation/google.golang.org/grpc/otelgrpc/test/grpc_test.go b/instrumentation/google.golang.org/grpc/otelgrpc/test/grpc_test.go
@@ -50,13 +50,16 @@ func newGrpcTest(t testing.TB, listener net.Listener, cOpt []grpc.DialOption, sO
 
 	cOpt = append(cOpt, grpc.WithTransportCredentials(insecure.NewCredentials()))
 
+	dialAddr := listener.Addr().String()
+
 	if l, ok := listener.(interface{ Dial() (net.Conn, error) }); ok {
 		dial := func(context.Context, string) (net.Conn, error) { return l.Dial() }
 		cOpt = append(cOpt, grpc.WithContextDialer(dial))
+		dialAddr = "passthrough:" + dialAddr
 	}
 
 	conn, err := grpc.NewClient(
-		listener.Addr().String(),
+		dialAddr,
 		cOpt...,
 	)
 	require.NoError(t, err)