feat: backend service

backend/internal/geo/geo_test.go

@@ -0,0 +1,82 @@
+package geo
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"go.uber.org/zap"
+)
+
+func TestLanguageForCountry(t *testing.T) {
+	cases := map[string]string{
+		"DE": "de",
+		"de": "de", // case-insensitive input
+		"RU": "ru",
+		"BR": "pt",
+		"":   "",
+		"ZZ": "",
+	}
+	for input, want := range cases {
+		if got := languageForCountry(input); got != want {
+			t.Errorf("languageForCountry(%q) = %q, want %q", input, got, want)
+		}
+	}
+}
+
+func TestLookupCountryNilSafety(t *testing.T) {
+	var s *Service
+	if got := s.LookupCountry("8.8.8.8"); got != "" {
+		t.Errorf("nil Service LookupCountry = %q, want empty", got)
+	}
+}
+
+func TestLanguageForIPNilSafety(t *testing.T) {
+	var s *Service
+	if got := s.LanguageForIP("8.8.8.8"); got != "" {
+		t.Errorf("nil Service LanguageForIP = %q, want empty", got)
+	}
+}
+
+func TestSetLoggerNilSafety(t *testing.T) {
+	var s *Service
+	s.SetLogger(zap.NewNop())
+	s.SetLogger(nil)
+
+	live := &Service{}
+	live.SetLogger(nil) // does not panic; falls back to nop logger.
+}
+
+func TestDrainNilSafety(t *testing.T) {
+	var s *Service
+	s.Drain(context.Background())
+}
+
+func TestDrainReturnsWhenContextDone(t *testing.T) {
+	live := &Service{}
+	live.bgCtx, live.bgCancel = context.WithCancel(context.Background())
+
+	ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
+	defer cancel()
+
+	start := time.Now()
+	live.Drain(ctx)
+	if elapsed := time.Since(start); elapsed > 5*time.Second {
+		t.Fatalf("Drain blocked too long: %s", elapsed)
+	}
+}
+
+func TestCloseIdempotent(t *testing.T) {
+	live := &Service{}
+	live.bgCtx, live.bgCancel = context.WithCancel(context.Background())
+	if err := live.Close(); err != nil {
+		t.Fatalf("first Close: %v", err)
+	}
+	if err := live.Close(); err != nil {
+		t.Fatalf("second Close: %v", err)
+	}
+	var nilSvc *Service
+	if err := nilSvc.Close(); err != nil {
+		t.Fatalf("nil Service Close: %v", err)
+	}
+}