Function TestSnapshot_BasicCounts

internal/nodehub/metrics_test.go:10–56

(t *testing.T)

)

func TestSnapshot_BasicCounts(t *testing.T) {
	hub := New(Options{PeerExtractor: mdPeerExtractor})
	env := newTestEnv(t, hub)

	// Offline call → ErrNodeOffline + 计数
	if _, err := hub.Call(context.Background(), "ghost", "x", nil); !errors.Is(err, ErrNodeOffline) {
		t.Fatalf("expected ErrNodeOffline, got %v", err)
	}
	snap := hub.Snapshot()
	if snap.CallsTotal != 1 || snap.CallsErrTotal != 1 || snap.CallsOfflineTotal != 1 {
		t.Fatalf("unexpected counters after offline call: %+v", snap)
	}
	if snap.OnlineCount != 0 || len(snap.OnlineNodeIDs) != 0 {
		t.Fatalf("expected zero online: %+v", snap)
	}

	// 上线一个节点
	cc := env.dial(t)
	mock := startNodeMock(t, cc, "n1")
	waitOnline(t, hub, "n1")

	mock.handle("ping", func(string, []byte) (bool, []byte, string) { return true, []byte("{}"), "" })

	if _, err := hub.Call(context.Background(), "n1", "ping", nil); err != nil {
		t.Fatalf("call: %v", err)
	}

	snap = hub.Snapshot()
	if snap.OnlineCount != 1 {
		t.Fatalf("OnlineCount=%d, want 1", snap.OnlineCount)
	}
	if len(snap.OnlineNodeIDs) != 1 || snap.OnlineNodeIDs[0] != "n1" {
		t.Fatalf("OnlineNodeIDs=%v", snap.OnlineNodeIDs)
	}
	if snap.CallsTotal != 2 {
		t.Fatalf("CallsTotal=%d, want 2", snap.CallsTotal)
	}
	if snap.CallsErrTotal != 1 {
		t.Fatalf("CallsErrTotal=%d, want 1", snap.CallsErrTotal)
	}
	if snap.CallLatencyAvgNs == 0 {
		t.Fatalf("expected non-zero latency avg")
	}
	if ts, ok := snap.LastSeen["n1"]; !ok || ts == 0 {
		t.Fatalf("LastSeen missing for n1: %+v", snap.LastSeen)
	}
}

func TestSnapshot_PushUsageAck(t *testing.T) {
	rp := &recordingPush{failOn: map[uint64]bool{2: true}}