diff --git a/internal/net/http/loadbalancer/loadbalancer_test.go b/internal/net/http/loadbalancer/loadbalancer_test.go index 349565e..25199b5 100644 --- a/internal/net/http/loadbalancer/loadbalancer_test.go +++ b/internal/net/http/loadbalancer/loadbalancer_test.go @@ -4,14 +4,13 @@ import ( "testing" "github.com/yusing/go-proxy/internal/net/http/loadbalancer/types" - loadbalance "github.com/yusing/go-proxy/internal/net/http/loadbalancer/types" . "github.com/yusing/go-proxy/internal/utils/testing" ) func TestRebalance(t *testing.T) { t.Parallel() t.Run("zero", func(t *testing.T) { - lb := New(new(loadbalance.Config)) + lb := New(new(types.Config)) for range 10 { lb.AddServer(types.TestNewServer(0)) } @@ -19,7 +18,7 @@ func TestRebalance(t *testing.T) { ExpectEqual(t, lb.sumWeight, maxWeight) }) t.Run("less", func(t *testing.T) { - lb := New(new(loadbalance.Config)) + lb := New(new(types.Config)) lb.AddServer(types.TestNewServer(float64(maxWeight) * .1)) lb.AddServer(types.TestNewServer(float64(maxWeight) * .2)) lb.AddServer(types.TestNewServer(float64(maxWeight) * .3)) @@ -30,7 +29,7 @@ func TestRebalance(t *testing.T) { ExpectEqual(t, lb.sumWeight, maxWeight) }) t.Run("more", func(t *testing.T) { - lb := New(new(loadbalance.Config)) + lb := New(new(types.Config)) lb.AddServer(types.TestNewServer(float64(maxWeight) * .1)) lb.AddServer(types.TestNewServer(float64(maxWeight) * .2)) lb.AddServer(types.TestNewServer(float64(maxWeight) * .3)) diff --git a/internal/route/types/http_config_test.go b/internal/route/types/http_config_test.go index 8fbd686..bf8ccaf 100644 --- a/internal/route/types/http_config_test.go +++ b/internal/route/types/http_config_test.go @@ -41,7 +41,7 @@ func TestHTTPConfigDeserialize(t *testing.T) { if err != nil { ExpectNoError(t, err) } - ExpectDeepEqual(t, cfg.HTTPConfig, &tt.expected) + ExpectDeepEqual(t, cfg.HTTPConfig, tt.expected) }) } }