Function TestUsagePushAck

internal/nodehub/hub_test.go:377–418

(t *testing.T)

func (r *recordingPush) OnTracerouteHop(string, string, []byte) {}

func TestUsagePushAck(t *testing.T) {
	rp := &recordingPush{failOn: map[uint64]bool{7: true}}
	hub := New(Options{PeerExtractor: mdPeerExtractor, PushHandler: rp})
	env := newTestEnv(t, hub)
	cc := env.dial(t)
	mock := startNodeMock(t, cc, "n")
	waitOnline(t, hub, "n")

	mock.pushUsage(1, []byte(`{"a":1}`))
	mock.pushUsage(7, []byte(`{"a":7}`)) // handler 失败 → 不应 ack
	mock.pushUsage(2, []byte(`{"a":2}`))

	// 期望收到 seq=1 和 seq=2 的 ack（顺序不强求）
	got := map[uint64]bool{}
	deadline := time.After(2 * time.Second)
	for len(got) < 2 {
		select {
		case body := <-mock.acks:
			var v struct{ Seq uint64 }
			if err := json.Unmarshal(body, &v); err != nil {
				t.Fatalf("ack body parse: %v / %s", err, body)
			}
			got[v.Seq] = true
		case <-deadline:
			t.Fatalf("timed out waiting for acks; got=%v", got)
		}
	}
	if !got[1] || !got[2] {
		t.Fatalf("missing acks; got=%v", got)
	}
	// seq=7 不应收到 ack；再等一小段确保没多发
	select {
	case body := <-mock.acks:
		var v struct{ Seq uint64 }
		_ = json.Unmarshal(body, &v)
		if v.Seq == 7 {
			t.Fatalf("unexpected ack for failing seq 7")
		}
	case <-time.After(150 * time.Millisecond):
		// good
	}
}

// 防止 io 未使用警告（部分平台）
var _ = io.EOF