@@ -31,6 +31,8 @@ import (
3131 "trpc.group/trpc-go/trpc-go/transport"
3232)
3333
34+ const preWarmAddress = "prewarm.example.test:8000"
35+
3436func 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
4749func TestInitializableClientPreWarmDisabled (t * testing.T ) {
@@ -87,7 +89,7 @@ type preWarmSelector struct{}
8789
8890func (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