aheh/MasterHttpRelayVPN
1
0
Fork 0
mirror of https://github.com/masterking32/MasterHttpRelayVPN.git synced 2026-05-18 23:54:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor session-level ping into explicit busy/aggressive/backoff state machine

Browse Source
This commit is contained in:
Amin.MasterkinG
2026-04-21 11:20:31 +03:30
parent 26e51fc031
commit a144703f5d
2 changed files with 71 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/client/sender_workers_test.go
+46
View File
@@ -391,6 +391,21 @@ func TestIdleIntervalForStreakBacksOffWithIdlePongs(t *testing.T) {
}
}
func TestNoteMeaningfulActivitySetsBusyState(t *testing.T) {
cfg := testClientConfig()
client := New(cfg, nil)
now := time.Now()
client.noteMeaningfulActivity(now)
if got := client.pingState.Load(); got != pingStateBusy {
t.Fatalf("expected busy ping state, got %d", got)
}
if client.nextPingDueUnixMS.Load() <= now.UnixMilli() {
t.Fatal("expected next ping due to be scheduled after meaningful activity")
}
}
func TestCompletePingWithPongIncrementsStreakOnlyWithoutRealTraffic(t *testing.T) {
cfg := testClientConfig()
client := New(cfg, nil)
@@ -404,6 +419,9 @@ func TestCompletePingWithPongIncrementsStreakOnlyWithoutRealTraffic(t *testing.T
if got := client.idlePongStreak.Load(); got != 1 {
t.Fatalf("expected pong streak to increment to 1, got %d", got)
}
if got := client.pingState.Load(); got != pingStateBackoffIdle {
t.Fatalf("expected backoff idle ping state, got %d", got)
}
nextDue := client.nextPingDueUnixMS.Load()
if nextDue <= now.UnixMilli() {
t.Fatal("expected next ping due to be scheduled in the future after idle pong")
@@ -418,6 +436,9 @@ func TestCompletePingWithPongIncrementsStreakOnlyWithoutRealTraffic(t *testing.T
if got := client.idlePongStreak.Load(); got != 0 {
t.Fatalf("expected pong streak reset after real traffic, got %d", got)
}
if got := client.pingState.Load(); got != pingStateBusy {
t.Fatalf("expected busy ping state after meaningful traffic, got %d", got)
}
if client.nextPingDueUnixMS.Load() <= now.UnixMilli() {
t.Fatal("expected next ping due to be rescheduled after meaningful traffic")
}
@@ -438,6 +459,9 @@ func TestCompletePingWithPongStaysAggressiveBeforeWarmThreshold(t *testing.T) {
if got := client.idlePongStreak.Load(); got != 0 {
t.Fatalf("expected pong streak to stay at 0 before warm threshold, got %d", got)
}
if got := client.pingState.Load(); got != pingStateAggressiveIdle {
t.Fatalf("expected aggressive idle ping state before warm threshold, got %d", got)
}
nextDue := client.nextPingDueUnixMS.Load()
expectedMin := now.Add(900 * time.Millisecond).UnixMilli()
@@ -447,6 +471,28 @@ func TestCompletePingWithPongStaysAggressiveBeforeWarmThreshold(t *testing.T) {
}
}
func TestFailPingReturnsToAggressiveIdle(t *testing.T) {
cfg := testClientConfig()
client := New(cfg, nil)
now := time.Now()
client.noteMeaningfulActivity(now.Add(-10 * time.Second))
client.idlePongStreak.Store(3)
client.setPingState(pingStateBackoffIdle)
client.failPing()
if got := client.idlePongStreak.Load(); got != 0 {
t.Fatalf("expected pong streak reset after ping failure, got %d", got)
}
if got := client.pingState.Load(); got != pingStateAggressiveIdle {
t.Fatalf("expected aggressive idle ping state after failure, got %d", got)
}
if client.nextPingDueUnixMS.Load() <= time.Now().UnixMilli() {
t.Fatal("expected next ping due to be rescheduled after ping failure")
}
}
func TestInboundReorderAllowsCloseReadAndCloseWriteOnSameSequence(t *testing.T) {
cfg := testClientConfig()
client := New(cfg, nil)