Function TestTeeDropOnMemoryGuard

pkg/agent/proxy/relay/relay_test.go:220–251

(t *testing.T)

}

func TestTeeDropOnMemoryGuard(t *testing.T) {
	t.Parallel()
	var pressure atomic.Bool
	pressure.Store(true)
	drops := newDropSink()

	h := newHarness(t, Config{
		MemoryGuardCheck:     pressure.Load,
		OnMarkMockIncomplete: drops.record,
	})

	// Forward traffic under memory pressure. Real traffic still flows.
	go h.writeClient([]byte("visible"))
	got := h.readDest(len("visible"))
	if string(got) != "visible" {
		t.Fatalf("dest got %q, want %q", got, "visible")
	}

	// But the FakeConn does not receive the chunk.
	_ = h.r.ClientStream().SetReadDeadline(time.Now().Add(100 * time.Millisecond))
	_, err := h.r.ClientStream().ReadChunk()
	if err == nil {
		t.Fatalf("ClientStream should have timed out, got chunk")
	}
	if !errors.Is(err, fakeconn.ErrDeadlineExceeded) && err.(net.Error).Timeout() != true {
		t.Fatalf("expected deadline error, got %v", err)
	}

	if drops.count(DropMemoryPressure) == 0 {
		t.Fatalf("expected memory_pressure drop reason, got %v", drops.snapshot())
	}
}

func TestTeeDropOnPerConnCap(t *testing.T) {
	t.Parallel()