mirror of
https://github.com/smallnest/rpcx.git
synced 2025-10-08 09:30:13 +08:00
39 lines
718 B
Go
39 lines
718 B
Go
package serverplugin
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
metrics "github.com/rcrowley/go-metrics"
|
|
"github.com/smallnest/rpcx/server"
|
|
)
|
|
|
|
func TestConsulRegistry(t *testing.T) {
|
|
s := server.NewServer()
|
|
|
|
r := &ConsulRegisterPlugin{
|
|
ServiceAddress: "tcp@127.0.0.1:8972",
|
|
ConsulServers: []string{"127.0.0.1:8500"},
|
|
BasePath: "/rpcx_test",
|
|
Metrics: metrics.NewRegistry(),
|
|
UpdateInterval: time.Minute,
|
|
}
|
|
err := r.Start()
|
|
if err != nil {
|
|
return
|
|
}
|
|
s.Plugins.Add(r)
|
|
|
|
s.RegisterName("Arith", new(Arith), "")
|
|
go s.Serve("tcp", "127.0.0.1:8972")
|
|
defer s.Close()
|
|
|
|
if len(r.Services) != 1 {
|
|
t.Fatal("failed to register services in consul")
|
|
}
|
|
|
|
if err := r.Stop(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|