73 lines
2.7 KiB
Go
73 lines
2.7 KiB
Go
package mtprotoedge
|
||
|
||
import (
|
||
"sync/atomic"
|
||
"testing"
|
||
"time"
|
||
|
||
"go.uber.org/zap/zaptest"
|
||
|
||
"github.com/gotd/td/clock"
|
||
"github.com/gotd/td/proto"
|
||
"github.com/gotd/td/tg"
|
||
|
||
"telesrv/internal/rpc"
|
||
)
|
||
|
||
type countingMetrics struct {
|
||
connOpened atomic.Int64
|
||
connClosed atomic.Int64
|
||
handshakes atomic.Int64
|
||
rpcs atomic.Int64
|
||
inbound atomic.Int64
|
||
outbound atomic.Int64
|
||
}
|
||
|
||
func (m *countingMetrics) ConnOpened() { m.connOpened.Add(1) }
|
||
func (m *countingMetrics) ConnClosed() { m.connClosed.Add(1) }
|
||
func (m *countingMetrics) HandshakeDone(time.Duration) { m.handshakes.Add(1) }
|
||
func (m *countingMetrics) RPCHandled(string, time.Duration, error) { m.rpcs.Add(1) }
|
||
func (m *countingMetrics) InboundRPCQueued(string, int, int) {}
|
||
func (m *countingMetrics) InboundRPCStarted(string, time.Duration) { m.inbound.Add(1) }
|
||
func (m *countingMetrics) InboundRPCDropped(string, string) {}
|
||
func (m *countingMetrics) OutboundSend(uint32, time.Duration, int, error) {
|
||
m.outbound.Add(1)
|
||
}
|
||
func (m *countingMetrics) OutboundResend(int, error) {}
|
||
func (m *countingMetrics) OutboundDropped(string) {}
|
||
func (m *countingMetrics) OutboundQueueWait(int, int) {}
|
||
|
||
// TestMetricsHooks 验证 M5：连接、握手、RPC 的 metrics 钩子被正确调用。
|
||
func TestMetricsHooks(t *testing.T) {
|
||
const dc = 2
|
||
m := &countingMetrics{}
|
||
router := rpc.New(rpc.Config{DC: dc, IP: "127.0.0.1", Port: 2398}, rpc.Deps{}, zaptest.NewLogger(t), clock.System)
|
||
addr, pub, _ := startTestServer(t, Options{DC: dc, RPC: router, Metrics: m})
|
||
conn, auth, cipher := dialHandshake(t, addr, dc, pub)
|
||
|
||
clientMsgID := proto.NewMessageIDGen(time.Now)
|
||
sendEncrypted(t, conn, cipher, auth, clientMsgID.New(proto.MessageFromClient), &tg.HelpGetConfigRequest{})
|
||
collectReplies(t, conn, cipher, auth.AuthKey, proto.ResultTypeID)
|
||
|
||
if got := m.connOpened.Load(); got < 1 {
|
||
t.Errorf("ConnOpened called %d times, want >= 1", got)
|
||
}
|
||
if got := m.handshakes.Load(); got != 1 {
|
||
t.Errorf("HandshakeDone called %d times, want 1", got)
|
||
}
|
||
if got := m.rpcs.Load(); got != 1 {
|
||
t.Errorf("RPCHandled called %d times, want 1", got)
|
||
}
|
||
if got := m.inbound.Load(); got != 1 {
|
||
t.Errorf("InboundRPCStarted called %d times, want 1", got)
|
||
}
|
||
// new_session_created / ack 走 fire-and-forget（异步），可能在 client 收到 rpc_result 后
|
||
// 才被 outbound actor 处理；轮询等其最终发送完成。M5 验证发送计数，不约束同步时序。
|
||
deadline := time.Now().Add(2 * time.Second)
|
||
for m.outbound.Load() < 3 && time.Now().Before(deadline) {
|
||
time.Sleep(5 * time.Millisecond)
|
||
}
|
||
if got := m.outbound.Load(); got < 3 {
|
||
t.Errorf("OutboundSend called %d times, want >= 3", got)
|
||
}
|
||
}
|