starainrt
09d972c7b7
- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package notify
|
|
|
|
import "testing"
|
|
|
|
func TestClientPublishSendFileEventObserverOnly(t *testing.T) {
|
|
client := NewClient().(*ClientCommon)
|
|
|
|
var observed []FileEvent
|
|
var handled []FileEvent
|
|
client.setFileEventObserver(func(event FileEvent) {
|
|
observed = append(observed, event)
|
|
})
|
|
client.SetFileHandler(func(event FileEvent) {
|
|
handled = append(handled, event)
|
|
})
|
|
|
|
event := FileEvent{
|
|
Kind: EnvelopeFileChunk,
|
|
Packet: FilePacket{FileID: "send-1", Size: 32},
|
|
}
|
|
client.publishSendFileEvent(event)
|
|
|
|
if got, want := len(observed), 1; got != want {
|
|
t.Fatalf("observed count mismatch: got %d want %d", got, want)
|
|
}
|
|
if got, want := len(handled), 0; got != want {
|
|
t.Fatalf("handled count mismatch: got %d want %d", got, want)
|
|
}
|
|
if got, want := observed[0].Packet.FileID, "send-1"; got != want {
|
|
t.Fatalf("observed fileID mismatch: got %q want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestClientPublishReceivedFileEventObserverAndHandler(t *testing.T) {
|
|
client := NewClient().(*ClientCommon)
|
|
|
|
var observed []FileEvent
|
|
var handled []FileEvent
|
|
client.setFileEventObserver(func(event FileEvent) {
|
|
observed = append(observed, event)
|
|
})
|
|
client.SetFileHandler(func(event FileEvent) {
|
|
handled = append(handled, event)
|
|
})
|
|
|
|
event := FileEvent{
|
|
Kind: EnvelopeFileEnd,
|
|
Packet: FilePacket{FileID: "recv-1", Size: 64},
|
|
Received: 64,
|
|
Done: true,
|
|
}
|
|
client.publishReceivedFileEvent(event)
|
|
|
|
if got, want := len(observed), 1; got != want {
|
|
t.Fatalf("observed count mismatch: got %d want %d", got, want)
|
|
}
|
|
if got, want := len(handled), 1; got != want {
|
|
t.Fatalf("handled count mismatch: got %d want %d", got, want)
|
|
}
|
|
if got, want := observed[0].Packet.FileID, "recv-1"; got != want {
|
|
t.Fatalf("observed fileID mismatch: got %q want %q", got, want)
|
|
}
|
|
if got, want := handled[0].Packet.FileID, "recv-1"; got != want {
|
|
t.Fatalf("handled fileID mismatch: got %q want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestServerPublishSendFileEventObserverOnly(t *testing.T) {
|
|
server := NewServer().(*ServerCommon)
|
|
|
|
var observed []FileEvent
|
|
var handled []FileEvent
|
|
server.setFileEventObserver(func(event FileEvent) {
|
|
observed = append(observed, event)
|
|
})
|
|
server.SetFileHandler(func(event FileEvent) {
|
|
handled = append(handled, event)
|
|
})
|
|
|
|
event := FileEvent{
|
|
Kind: EnvelopeFileMeta,
|
|
Packet: FilePacket{FileID: "server-send-1", Size: 128},
|
|
}
|
|
server.publishSendFileEvent(event)
|
|
|
|
if got, want := len(observed), 1; got != want {
|
|
t.Fatalf("observed count mismatch: got %d want %d", got, want)
|
|
}
|
|
if got, want := len(handled), 0; got != want {
|
|
t.Fatalf("handled count mismatch: got %d want %d", got, want)
|
|
}
|
|
if got, want := observed[0].Packet.FileID, "server-send-1"; got != want {
|
|
t.Fatalf("observed fileID mismatch: got %q want %q", got, want)
|
|
}
|
|
}
|