54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
package frp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAgentTunnelBindsLoopbackAndReleasesPort(t *testing.T) {
|
||
|
|
probe, err := net.Listen("tcp", "127.0.0.1:0")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
port := probe.Addr().(*net.TCPAddr).Port
|
||
|
|
_ = probe.Close()
|
||
|
|
|
||
|
|
tunnel := newAgentTunnel(context.Background(), 1, port, SSHInstaller{})
|
||
|
|
if err := tunnel.Start(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if _, err := net.Listen("tcp", fmt.Sprintf("127.0.0.1:%d", port)); err == nil {
|
||
|
|
t.Fatal("隧道启动后本地端口应已被占用")
|
||
|
|
}
|
||
|
|
state, _ := tunnel.State()
|
||
|
|
if state != "connected" && state != "reconnecting" {
|
||
|
|
t.Fatalf("state=%q, want connected 或 reconnecting", state)
|
||
|
|
}
|
||
|
|
tunnel.Stop()
|
||
|
|
if listener, err := net.Listen("tcp", fmt.Sprintf("127.0.0.1:%d", port)); err != nil {
|
||
|
|
t.Fatalf("隧道停止后端口未释放: %v", err)
|
||
|
|
} else {
|
||
|
|
_ = listener.Close()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNextTunnelBackoff(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
in time.Duration
|
||
|
|
want time.Duration
|
||
|
|
}{
|
||
|
|
{0, agentTunnelBackoffMin},
|
||
|
|
{time.Second, 2 * time.Second},
|
||
|
|
{40 * time.Second, agentTunnelBackoffMax},
|
||
|
|
{agentTunnelBackoffMax, agentTunnelBackoffMax},
|
||
|
|
}
|
||
|
|
for _, tc := range cases {
|
||
|
|
if got := nextTunnelBackoff(tc.in); got != tc.want {
|
||
|
|
t.Fatalf("nextTunnelBackoff(%v) = %v, 期望 %v", tc.in, got, tc.want)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|