package reqctx

import "testing"

func TestParseHTTPRealIP(t *testing.T) {
	got := ParseHTTPRealIP(" 203.0.113.10 ", "198.51.100.1, 192.0.2.1", "192.0.2.1:1234")
	if got != "203.0.113.10" {
		t.Fatalf("x-real-ip %q", got)
	}
	got = ParseHTTPRealIP("", "198.51.100.1, 192.0.2.1", "192.0.2.1:1234")
	if got != "198.51.100.1" {
		t.Fatalf("xff %q", got)
	}
	got = ParseHTTPRealIP("", "", "192.0.2.1:1234")
	if got != "192.0.2.1" {
		t.Fatalf("remote %q", got)
	}
}

func TestRealIPFromContext(t *testing.T) {
	ctx := With(t.Context(), &RequestContext{RealIP: "203.0.113.10"})
	if RealIP(ctx) != "203.0.113.10" {
		t.Fatalf("got %q", RealIP(ctx))
	}
	if RealIP(t.Context()) != "" {
		t.Fatal("empty ctx")
	}
}

func TestLangFromContext(t *testing.T) {
	ctx := With(t.Context(), &RequestContext{Language: "en-US"})
	if Language(ctx) != "en" {
		t.Fatalf("got %q", Language(ctx))
	}
	if Language(t.Context()) != "zh-CN" {
		t.Fatalf("default %q", Language(t.Context()))
	}
}