feat: improved port selection

This commit is contained in:
yusing
2025-05-04 01:32:01 +08:00
parent ccb4639f43
commit 5f780f4902
2 changed files with 30 additions and 6 deletions

View File

@@ -3,6 +3,7 @@ package route
import (
"testing"
"github.com/docker/docker/api/types/container"
"github.com/yusing/go-proxy/internal/common"
"github.com/yusing/go-proxy/internal/docker"
loadbalance "github.com/yusing/go-proxy/internal/net/gphttp/loadbalancer/types"
@@ -136,3 +137,14 @@ func TestRouteValidate(t *testing.T) {
expect.NotNil(t, r.HealthCheck)
})
}
func TestPreferredPort(t *testing.T) {
ports := map[int]container.Port{
22: {PrivatePort: 22},
1000: {PrivatePort: 1000},
3000: {PrivatePort: 80},
}
port := preferredPort(ports)
expect.Equal(t, port, 3000)
}