Skip to content

Commit 134da05

Browse files
committed
client: avoid private IP in prewarm tests
1 parent 338e673 commit 134da05

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

client/pre_warm_test.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ import (
3131
"trpc.group/trpc-go/trpc-go/transport"
3232
)
3333

34+
const preWarmAddress = "prewarm.example.test:8000"
35+
3436
func TestInitializableClientPreWarm(t *testing.T) {
3537
cli, ok := client.New().(client.InitializableClient)
3638
require.True(t, ok)
@@ -41,7 +43,7 @@ func TestInitializableClientPreWarm(t *testing.T) {
4143
client.WithPool(pool),
4244
client.WithPreWarm(transport.PreWarmOptions{ConnsPerNode: 2}),
4345
))
44-
require.Equal(t, []string{"10.0.0.1:8000", "10.0.0.1:8000"}, pool.addresses)
46+
require.Equal(t, []string{preWarmAddress, preWarmAddress}, pool.addresses)
4547
}
4648

4749
func TestInitializableClientPreWarmDisabled(t *testing.T) {
@@ -87,7 +89,7 @@ type preWarmSelector struct{}
8789

8890
func (preWarmSelector) Select(serviceName string, _ ...selector.Option) (*registry.Node, error) {
8991
if serviceName == "backend" {
90-
return &registry.Node{Network: "tcp", Address: "10.0.0.1:8000"}, nil
92+
return &registry.Node{Network: "tcp", Address: preWarmAddress}, nil
9193
}
9294
return nil, errors.New("unknown service")
9395
}

0 commit comments

Comments
 (0)