Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go

Source identity

Field	Recorded value
Source type	GitHub source
Original record	Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go
Date	Not specified in source
Repository	Janady13/missionmatch
Revision	720d47fae476b0d38477ae4401697d372b1ea2bc
Branch	feature/opportunity-to-outcome
Path or archive member	internal/platform/ratelimit/ratelimit_test.go
Original SHA-256	ce89d4c715bc5d8fb97847551ec1f4e1c9dc885ffb4f3efe3bb0b64ebfbe9019
Git blob SHA	3f807a80308745d80a3c1bf80becce1d1ec29a8f
Reading-copy SHA-256	ce89d4c715bc5d8fb97847551ec1f4e1c9dc885ffb4f3efe3bb0b64ebfbe9019

Original source ↗

Every recorded occurrence

These 31 records contain identical original source bytes. Their paths and revisions remain part of the archive.

Original record	Branch or archive	Revision / identifier	Original location
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	feature/opportunity-to-outcome	720d47fae476b0d38477ae4401697d372b1ea2bc	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	per-tenant-routing	9bde0f9a74b89d01980fa658e5253acda7528d0c	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	collision-deconfliction	4ee5a09972608ed89f8377c6665542d5f450f0a6	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws6-retrieval	6c56e29231ec05735927fa265c97d517499dfeaf	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	workstream-a-shared-core	e0de03ac3a1ec826bd15d0e048e3b02baa23eeff	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	m5-price-to-win	41000ad30fff4129fdaf5ec84e15ba14c27d4890	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	workstream-c-slatematch	5cd60c918301b46ecd6ed3e03e20c395da0062e2	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws1-ontology-sync	b596f449a32a1016fbd403c36505cc00a1ac9ee6	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	master	1e9249840a4302ec82e123b9c9cb6f75badea3e4	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws8-teaching-loop	c823f0fb6bb2aba0f39001eb6de96f4000d9321f	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	w3-ai-staffing-recommendations	43d56387270ef4da4cabd794d97ca6a49909bc7d	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	mag-substrate	b82c7100a492dec67dfd1eda6b3f53def859be35	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws3-clearance	4b8e7ce594c068621867f9c9deed0ec64ae6c553	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	phase2-eval-fairness	d361af254701975fe239927656ababb56374655e	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	workstream-e-dataflow-deploy	433815a042f9741411e60fa553fe72143e9b10db	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ruthless-debug-sweep	b7d742f8f032984be3817efffa3eb8b13b651c1f	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	time-to-fill-survival	b66e1e505863d4f17741a4b9ede73509e45bd6c6	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ai-intelligence-layer	5d5ddff70393a197c5f54a45280bb5dd42b24755	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	graphrag-foundation	b0ddeb9f633fd138d09309d1a17b7d13bc3a67f7	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	deploy-and-ops	bdacd9d0b4d5998d25fc74bf4007f4c3ec538c7a	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws9-bench-ingestion	f585a9ef32fa554663ce1fba9a414a54b6e49cf0	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws2-alias-relationship	e4dada4601c6eb101830f9d223a9655dfa120616	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	presentation-readiness	eb574acaa29d71ce4d0f852572e87c275c79fdb7	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	vendor-intelligence	fe9796a19c261951d3f2565206c0c1b5ac294465	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	workstream-b-screenmatch	09015496d565a172bcaa15cfe9a8e4b072b13df0	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ws7-federal-signal	6a9f6ce9cfb13345ffb47770d4960e7c2d9c17c2	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	ai-brain-live	aac32fc49c878cc5bca099459307611562d0bac1	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	workstream-d-missionmatch	3be46ba89edb737182075a2b320509b42dc916f8	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	phase2-connectors	e038d4c687aec16f8bd9bc4d907eea14d91f8704	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	feature/aioe-upgrade	2178e7298d0e5d9af83afa12402fd587f0ba6ed8	Open original
Janady13/missionmatch / internal/platform/ratelimit/ratelimit_test.go	capture-readiness	bae6af7cb2997e819174a9e9cfff5e16067a1975	Open original

Recovered source transcript

Download complete source text ↓74 lines · 2.2 KB

package ratelimit

import (
	"net/http"
	"testing"
	"time"
)

func TestTokenBucket(t *testing.T) {
	l := New(1, 2) // 1 token/sec, burst 2
	now := time.Unix(1000, 0)
	l.now = func() time.Time { return now }

	// Burst of 2 is allowed, the third in the same instant is denied. Capture both calls so the second
	// always runs (it consumes the second token) rather than being short circuited.
	first, second := l.Allow("a"), l.Allow("a")
	if !first || !second {
		t.Fatal("burst of 2 should be allowed")
	}
	if l.Allow("a") {
		t.Fatal("third request in the same instant must be denied")
	}
	// A different key has its own bucket.
	if !l.Allow("b") {
		t.Fatal("a separate key must not share key a's bucket")
	}
	// After one second, one token refills.
	now = now.Add(time.Second)
	if !l.Allow("a") {
		t.Fatal("one token should refill after a second")
	}
	if l.Allow("a") {
		t.Fatal("only one token refilled, the next must be denied")
	}
}

func TestSweepDropsIdleFullBuckets(t *testing.T) {
	l := New(1, 1)
	now := time.Unix(1000, 0)
	l.now = func() time.Time { return now }
	l.Allow("a") // creates the bucket, now empty
	now = now.Add(10 * time.Second)
	l.Allow("a") // refills to full and consumes... back below full, so not swept yet
	now = now.Add(10 * time.Second)
	// "a" has refilled to full (capacity 1) by now; sweep should drop it.
	l.sweep(5 * time.Second)
	l.mu.Lock()
	_, present := l.buckets["a"]
	l.mu.Unlock()
	if present {
		t.Fatal("an idle, full bucket should be swept")
	}
}

func TestClientIP(t *testing.T) {
	r := &http.Request{RemoteAddr: "203.0.113.7:54321", Header: http.Header{}}
	if got := ClientIP(r, ""); got != "203.0.113.7" {
		t.Fatalf("TCP peer: got %q want 203.0.113.7", got)
	}
	r.Header.Set("X-Real-IP", "198.51.100.9")
	if got := ClientIP(r, "X-Real-IP"); got != "198.51.100.9" {
		t.Fatalf("trusted header: got %q want 198.51.100.9", got)
	}
	// A spoofed header is ignored when no trusted header is configured.
	if got := ClientIP(r, ""); got != "203.0.113.7" {
		t.Fatalf("untrusted: got %q want the TCP peer", got)
	}
	// A proxy chain takes the first entry.
	r.Header.Set("X-Real-IP", "198.51.100.9, 10.0.0.1")
	if got := ClientIP(r, "X-Real-IP"); got != "198.51.100.9" {
		t.Fatalf("proxy chain: got %q want the first entry", got)
	}
}