feat(notify): 重构通信内核并补齐 stream/bulk/record/transfer 能力

- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
2026-04-15 15:24:36 +08:00
parent d14d13c393
commit 09d972c7b7
216 changed files with 51374 additions and 1715 deletions
@@ -0,0 +1,118 @@
+package starnotify
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+
+	"b612.me/notify"
+)
+
+func TestUseSignalReliabilityClient(t *testing.T) {
+	const key = "signal-reliable-client"
+	_ = DeleteClient(key)
+	defer DeleteClient(key)
+
+	NewClient(key)
+	err := UseSignalReliabilityClient(key, &notify.SignalReliabilityOptions{
+		AckTimeout:        50 * time.Millisecond,
+		SendRetry:         4,
+		ReceiveCacheLimit: 8,
+	})
+	if err != nil {
+		t.Fatalf("UseSignalReliabilityClient failed: %v", err)
+	}
+}
+
+func TestUseSignalReliabilityServer(t *testing.T) {
+	const key = "signal-reliable-server"
+	_ = DeleteServer(key)
+	defer DeleteServer(key)
+
+	NewServer(key)
+	opts := notify.DefaultSignalReliabilityOptions()
+	err := UseSignalReliabilityServer(key, &opts)
+	if err != nil {
+		t.Fatalf("UseSignalReliabilityServer failed: %v", err)
+	}
+}
+
+func TestUseSignalReliabilityMissingKey(t *testing.T) {
+	if err := UseSignalReliabilityClient("missing-client", nil); err == nil {
+		t.Fatal("UseSignalReliabilityClient should fail for missing client key")
+	}
+	if err := UseSignalReliabilityServer("missing-server", nil); err == nil {
+		t.Fatal("UseSignalReliabilityServer should fail for missing server key")
+	}
+}
+
+func TestGetSignalReliabilityStatsByKey(t *testing.T) {
+	const clientKey = "signal-reliable-stats-client"
+	const serverKey = "signal-reliable-stats-server"
+	_ = DeleteClient(clientKey)
+	_ = DeleteServer(serverKey)
+	defer DeleteClient(clientKey)
+	defer DeleteServer(serverKey)
+
+	NewClient(clientKey)
+	NewServer(serverKey)
+
+	clientStats, err := GetSignalReliabilityStatsClient(clientKey)
+	if err != nil {
+		t.Fatalf("GetSignalReliabilityStatsClient failed: %v", err)
+	}
+	if clientStats != (notify.SignalReliabilityStats{}) {
+		t.Fatalf("client stats mismatch: %+v", clientStats)
+	}
+
+	serverStats, err := GetSignalReliabilityStatsServer(serverKey)
+	if err != nil {
+		t.Fatalf("GetSignalReliabilityStatsServer failed: %v", err)
+	}
+	if serverStats != (notify.SignalReliabilityStats{}) {
+		t.Fatalf("server stats mismatch: %+v", serverStats)
+	}
+}
+
+func TestGetSignalReliabilityStatsMissingKey(t *testing.T) {
+	if _, err := GetSignalReliabilityStatsClient("missing-client"); err == nil {
+		t.Fatal("GetSignalReliabilityStatsClient should fail for missing key")
+	}
+	if _, err := GetSignalReliabilityStatsServer("missing-server"); err == nil {
+		t.Fatal("GetSignalReliabilityStatsServer should fail for missing key")
+	}
+}
+
+func TestConnectRetryWrappersContextCanceled(t *testing.T) {
+	const clientKey = "signal-reliable-retry-client"
+	const serverKey = "signal-reliable-retry-server"
+	_ = DeleteClient(clientKey)
+	_ = DeleteServer(serverKey)
+	defer DeleteClient(clientKey)
+	defer DeleteServer(serverKey)
+
+	NewClient(clientKey)
+	NewServer(serverKey)
+
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	connectErr := ConnectClientWithRetryCtx(ctx, clientKey, "tcp", "127.0.0.1:1", &notify.ConnectRetryOptions{
+		MaxAttempts: 3,
+		BaseDelay:   time.Millisecond,
+		MaxDelay:    time.Millisecond,
+	})
+	if !errors.Is(connectErr, context.Canceled) {
+		t.Fatalf("ConnectClientWithRetryCtx error = %v, want %v", connectErr, context.Canceled)
+	}
+
+	listenErr := ListenServerWithRetryCtx(ctx, serverKey, "tcp", "127.0.0.1:1", &notify.ConnectRetryOptions{
+		MaxAttempts: 3,
+		BaseDelay:   time.Millisecond,
+		MaxDelay:    time.Millisecond,
+	})
+	if !errors.Is(listenErr, context.Canceled) {
+		t.Fatalf("ListenServerWithRetryCtx error = %v, want %v", listenErr, context.Canceled)
+	}
+}