package tenancy

import "testing"

func TestResolverNamedGroup(t *testing.T) {
	resolver, err := NewResolver("^(?P<tenant>[^:]+):", "reject")
	if err != nil {
		t.Fatalf("NewResolver failed: %v", err)
	}
	tenantID, err := resolver.Resolve("alpha:secret")
	if err != nil {
		t.Fatalf("Resolve failed: %v", err)
	}
	if tenantID != "alpha" {
		t.Fatalf("expected tenant alpha, got %q", tenantID)
	}
}

func TestResolverPositionalGroup(t *testing.T) {
	resolver, err := NewResolver("^([^:]+):", "reject")
	if err != nil {
		t.Fatalf("NewResolver failed: %v", err)
	}
	tenantID, err := resolver.Resolve("beta:secret")
	if err != nil {
		t.Fatalf("Resolve failed: %v", err)
	}
	if tenantID != "beta" {
		t.Fatalf("expected tenant beta, got %q", tenantID)
	}
}

func TestResolverFallbackUsername(t *testing.T) {
	resolver, err := NewResolver("^foo:", "username")
	if err != nil {
		t.Fatalf("NewResolver failed: %v", err)
	}
	tenantID, err := resolver.Resolve("tenant_user")
	if err != nil {
		t.Fatalf("Resolve failed: %v", err)
	}
	if tenantID != "tenant_user" {
		t.Fatalf("expected tenant tenant_user, got %q", tenantID)
	}
}

func TestResolverFallbackLiteral(t *testing.T) {
	resolver, err := NewResolver("^foo:", "literal")
	if err != nil {
		t.Fatalf("NewResolver failed: %v", err)
	}
	tenantID, err := resolver.Resolve("tenant_user")
	if err != nil {
		t.Fatalf("Resolve failed: %v", err)
	}
	if tenantID != "^foo:" {
		t.Fatalf("expected tenant ^foo:, got %q", tenantID)
	}
}

func TestResolverFallbackReject(t *testing.T) {
	resolver, err := NewResolver("^foo:", "reject")
	if err != nil {
		t.Fatalf("NewResolver failed: %v", err)
	}
	if _, err := resolver.Resolve("tenant_user"); err == nil {
		t.Fatalf("expected resolve error for non-matching username")
	}
}