Apply suggestions from code review

Co-authored-by: J. Yi <[email protected]>
Jigsaw-Code · Jan 2, 2024 · e6cb5c3 · e6cb5c3
1 parent f8c7bf8
commit e6cb5c3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/transport/packet_test.go b/transport/packet_test.go
@@ -84,7 +84,8 @@ func TestFuncPacketEndpoint(t *testing.T) {
 func TestFuncPacketDialer(t *testing.T) {
 	expectedConn := &fakeConn{}
 	expectedErr := errors.New("fake error")
-	dialer := FuncPacketDialer(func(ctx context.Context, add string) (net.Conn, error) {
+	dialer := FuncPacketDialer(func(ctx context.Context, addr string) (net.Conn, error) {
+		require.Equal(t, "unused", addr)
 		return expectedConn, expectedErr
 	})
 	conn, err := dialer.Dial(context.Background(), "unused")

diff --git a/transport/stream_test.go b/transport/stream_test.go
@@ -45,7 +45,8 @@ func TestFuncStreamEndpoint(t *testing.T) {
 func TestFuncStreamDialer(t *testing.T) {
 	expectedConn := &fakeConn{}
 	expectedErr := errors.New("fake error")
-	dialer := FuncStreamDialer(func(ctx context.Context, add string) (StreamConn, error) {
+	dialer := FuncStreamDialer(func(ctx context.Context, addr string) (StreamConn, error) {
+		require.Equal(t, "unused", addr)
 		return expectedConn, expectedErr
 	})
 	conn, err := dialer.Dial(context.Background(), "unused")