79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package discover_test
|
||
|
||
import (
|
||
"fmt"
|
||
"net"
|
||
"net/http"
|
||
"testing"
|
||
"time"
|
||
|
||
"apigo.cc/go/discover"
|
||
)
|
||
|
||
func TestMultipleDiscoverer(t *testing.T) {
|
||
// 启动两个模拟服务
|
||
l1, _ := net.Listen("tcp", "127.0.0.1:18011")
|
||
mux1 := http.NewServeMux()
|
||
mux1.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { _, _ = w.Write([]byte("OK1")) })
|
||
server1 := &http.Server{Handler: mux1}
|
||
go func() { _ = server1.Serve(l1) }()
|
||
defer server1.Close()
|
||
|
||
l2, _ := net.Listen("tcp", "127.0.0.1:18012")
|
||
mux2 := http.NewServeMux()
|
||
mux2.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { _, _ = w.Write([]byte("OK2")) })
|
||
server2 := &http.Server{Handler: mux2}
|
||
go func() { _ = server2.Serve(l2) }()
|
||
defer server2.Close()
|
||
|
||
registry := "redis://127.0.0.1:6379/15"
|
||
|
||
// 实例 1
|
||
d1 := discover.NewDiscoverer()
|
||
c1conf := d1.GetConfig()
|
||
c1conf.App = "app1"
|
||
c1conf.Registry = registry
|
||
d1.SetConfig(c1conf)
|
||
if !d1.Start("127.0.0.1:18011") {
|
||
t.Skip("redis not available")
|
||
}
|
||
defer d1.Stop()
|
||
|
||
// 实例 2
|
||
d2 := discover.NewDiscoverer()
|
||
c2conf := d2.GetConfig()
|
||
c2conf.App = "app2"
|
||
c2conf.Registry = registry
|
||
d2.SetConfig(c2conf)
|
||
if !d2.Start("127.0.0.1:18012") {
|
||
t.Skip("redis not available")
|
||
}
|
||
defer d2.Stop()
|
||
|
||
// 实例 1 发现并调用自己
|
||
d1.AddExternalApp("app1", "1")
|
||
time.Sleep(200 * time.Millisecond) // 等待同步
|
||
c1 := d1.NewCaller(nil, nil)
|
||
res1 := c1.Get("app1", "/")
|
||
if res1.Error != nil || res1.String() != "OK1" {
|
||
t.Errorf("d1 call app1 failed: %v, %s", res1.Error, res1.String())
|
||
}
|
||
|
||
// 实例 2 发现并调用 实例 1
|
||
d2.AddExternalApp("app1", "1")
|
||
time.Sleep(200 * time.Millisecond) // 等待同步
|
||
c2 := d2.NewCaller(nil, nil)
|
||
res2 := c2.Get("app1", "/")
|
||
if res2.Error != nil || res2.String() != "OK1" {
|
||
t.Errorf("d2 call app1 failed: %v, %s", res2.Error, res2.String())
|
||
}
|
||
|
||
// 验证独立性：d1 不应该能直接调用 app2 (除非手动 AddExternalApp)
|
||
res3 := c1.Get("app2", "/")
|
||
if res3.Error == nil {
|
||
t.Error("d1 should not find app2 without AddExternalApp")
|
||
}
|
||
|
||
fmt.Println("Multiple Discoverer instances verified")
|
||
}
|