probo/pkg/accessreview/drivers/name_resolver_test.go

// Copyright (c) 2026 Probo Inc <hello@probo.com>.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

package drivers

import (
	"context"
	"errors"
	"net/http"
	"net/http/httptest"
	"net/url"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

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

	terminal := []int{
		http.StatusBadRequest,
		http.StatusUnauthorized,
		http.StatusForbidden,
		http.StatusNotFound,
	}
	for _, code := range terminal {
		err := nameStatusError("thing", code)
		require.Error(t, err)
		assert.ErrorIs(t, err, ErrTerminalNameResolution, "status %d must be terminal", code)
	}

	retryable := []int{
		http.StatusTooManyRequests,
		http.StatusInternalServerError,
		http.StatusBadGateway,
		http.StatusServiceUnavailable,
		http.StatusGatewayTimeout,
	}
	for _, code := range retryable {
		err := nameStatusError("thing", code)
		require.Error(t, err)
		assert.False(t, errors.Is(err, ErrTerminalNameResolution), "status %d must be retryable", code)
	}
}

// TestNameResolversTerminalOnClientError guards the fix that makes the
// grafana, metabase and tailscale name resolvers surface a permanent 4xx as
// ErrTerminalNameResolution. Without it a revoked-credential source hot-loops
// the source-name worker, which never marks such a source synced.
func TestNameResolversTerminalOnClientError(t *testing.T) {
	t.Parallel()

	srv := httptest.NewServer(
		http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
			w.WriteHeader(http.StatusUnauthorized)
			_, _ = w.Write([]byte(`{"message":"unauthorized"}`))
		}),
	)
	t.Cleanup(srv.Close)

	cases := []struct {
		name     string
		resolver NameResolver
	}{
		{name: "grafana", resolver: NewGrafanaNameResolver(srv.Client(), srv.URL)},
		{name: "metabase", resolver: NewMetabaseNameResolver(srv.Client(), srv.URL)},
		{name: "tailscale", resolver: NewTailscaleNameResolver(&http.Client{Transport: &hostRewriter{target: srv.URL}})},
	}

	for _, tc := range cases {
		t.Run(
			tc.name,
			func(t *testing.T) {
				t.Parallel()

				_, err := tc.resolver.ResolveInstanceName(context.Background())
				require.Error(t, err)
				assert.ErrorIs(t, err, ErrTerminalNameResolution)
			},
		)
	}
}

// hostRewriter redirects requests to the configured target host so that
// resolvers with hardcoded production URLs (api.notion.com, etc.) can be
// pointed at an httptest server.
type hostRewriter struct {
	target string
}

func (h *hostRewriter) RoundTrip(r *http.Request) (*http.Response, error) {
	u, err := url.Parse(h.target)
	if err != nil {
		return nil, err
	}

	r2 := r.Clone(r.Context())
	r2.URL.Scheme = u.Scheme
	r2.URL.Host = u.Host

	return http.DefaultTransport.RoundTrip(r2)
}

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

	cases := []struct {
		name    string
		status  int
		body    string
		want    string
		wantErr bool
	}{
		{
			name:   "bot with workspace_name",
			status: http.StatusOK,
			body:   `{"type":"bot","bot":{"workspace_name":"Acme Inc"}}`,
			want:   "Acme Inc",
		},
		{
			name:   "user token (no bot field)",
			status: http.StatusOK,
			body:   `{"type":"person"}`,
			want:   "",
		},
		{
			name:    "server error",
			status:  http.StatusInternalServerError,
			body:    `{"message":"boom"}`,
			wantErr: true,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/v1/users/me", r.URL.Path)
				assert.Equal(t, notionAPIVersion, r.Header.Get("Notion-Version"))
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewNotionNameResolver(client).ResolveInstanceName(context.Background())
			if tc.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("empty slug returns nothing without HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for an empty slug")
			return nil, nil
		})}

		got, err := NewSentryNameResolver(client, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, got)
	})

	cases := []struct {
		name    string
		status  int
		body    string
		want    string
		wantErr bool
	}{
		{
			name:   "200 returns name",
			status: http.StatusOK,
			body:   `{"slug":"acme","name":"Acme Inc"}`,
			want:   "Acme Inc",
		},
		{
			name:   "404 is terminal (no error, no name)",
			status: http.StatusNotFound,
			body:   `{"detail":"The requested resource does not exist"}`,
			want:   "",
		},
		{
			name:    "401 is retryable",
			status:  http.StatusUnauthorized,
			body:    `{"detail":"Authentication credentials were not provided."}`,
			wantErr: true,
		},
		{
			name:    "403 is retryable",
			status:  http.StatusForbidden,
			body:    `{"detail":"You do not have permission to perform this action."}`,
			wantErr: true,
		},
		{
			name:    "500 is retryable",
			status:  http.StatusInternalServerError,
			body:    `{"detail":"Internal Server Error"}`,
			wantErr: true,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/api/0/organizations/acme/", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewSentryNameResolver(client, "acme").ResolveInstanceName(context.Background())
			if tc.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("empty organization id returns nothing without HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for an empty organization id")
			return nil, nil
		})}

		got, err := NewQoveryNameResolver(client, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, got)
	})

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "200 returns name",
			status: http.StatusOK,
			body:   `{"id":"26ac87db-ae79-4be4-bd33-7f839f0e1647","name":"Acme Inc"}`,
			want:   "Acme Inc",
		},
		{
			name:   "401 is terminal (no error, no name)",
			status: http.StatusUnauthorized,
			body:   `{"error":"unauthorized"}`,
			want:   "",
		},
		{
			name:   "404 is terminal (no error, no name)",
			status: http.StatusNotFound,
			body:   `{"error":"not found"}`,
			want:   "",
		},
		{
			name:   "500 is terminal (no error, no name)",
			status: http.StatusInternalServerError,
			body:   `{"error":"boom"}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/organization/26ac87db-ae79-4be4-bd33-7f839f0e1647", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewQoveryNameResolver(client, "26ac87db-ae79-4be4-bd33-7f839f0e1647").ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("empty owner id returns nothing without HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for an empty owner id")
			return nil, nil
		})}

		got, err := NewRenderNameResolver(client, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, got)
	})

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "200 returns name",
			status: http.StatusOK,
			body:   `{"id":"tea-test","name":"Acme Workspace","email":"ops@example.com","type":"team"}`,
			want:   "Acme Workspace",
		},
		{
			name:   "401 is terminal (no error, no name)",
			status: http.StatusUnauthorized,
			body:   `{"message":"unauthorized"}`,
			want:   "",
		},
		{
			name:   "404 is terminal (no error, no name)",
			status: http.StatusNotFound,
			body:   `{"message":"not found"}`,
			want:   "",
		},
		{
			name:   "500 is terminal (no error, no name)",
			status: http.StatusInternalServerError,
			body:   `{"message":"boom"}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/v1/owners/tea-test", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewRenderNameResolver(client, "tea-test").ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("empty organization id returns nothing without HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for an empty organization id")
			return nil, nil
		})}

		got, err := NewNeonNameResolver(client, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, got)
	})

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "200 returns name",
			status: http.StatusOK,
			body:   `{"id":"org-cool-breeze-12345678","name":"Acme Inc","handle":"acme-inc-org-cool-breeze-12345678","plan":"launch"}`,
			want:   "Acme Inc",
		},
		{
			name:   "401 is terminal (no error, no name)",
			status: http.StatusUnauthorized,
			body:   `{"error":"unauthorized"}`,
			want:   "",
		},
		{
			name:   "404 is terminal (no error, no name)",
			status: http.StatusNotFound,
			body:   `{"error":"not found"}`,
			want:   "",
		},
		{
			name:   "500 is terminal (no error, no name)",
			status: http.StatusInternalServerError,
			body:   `{"error":"boom"}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/api/v2/organizations/org-cool-breeze-12345678", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewNeonNameResolver(client, "org-cool-breeze-12345678").ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	cases := []struct {
		name    string
		status  int
		body    string
		want    string
		wantErr bool
	}{
		{
			name:   "custom domain tailnet",
			status: http.StatusOK,
			body:   `{"users":[{"loginName":"jane@acme.example.com"},{"loginName":"bob@acme.example.com"}]}`,
			want:   "acme.example.com",
		},
		{
			name:   "most common domain wins",
			status: http.StatusOK,
			body:   `{"users":[{"loginName":"a@one.com"},{"loginName":"b@two.com"},{"loginName":"c@two.com"}]}`,
			want:   "two.com",
		},
		{
			name:   "no usable login names",
			status: http.StatusOK,
			body:   `{"users":[{"loginName":""},{"loginName":"tagged-device"}]}`,
			want:   "",
		},
		{
			name:    "server error",
			status:  http.StatusInternalServerError,
			body:    `{"message":"boom"}`,
			wantErr: true,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/api/v2/tailnet/-/users", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewTailscaleNameResolver(client).ResolveInstanceName(context.Background())
			if tc.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("personal-account slug returns a name without an HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for a personal account")
			return nil, nil
		})}

		got, err := NewHerokuNameResolver(client, herokuPersonalAccountSlug).ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Equal(t, "Personal account", got)
	})

	t.Run("team slug resolves the team name", func(t *testing.T) {
		t.Parallel()

		srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			assert.Equal(t, http.MethodGet, r.Method)
			assert.Equal(t, "/teams/acme", r.URL.Path)
			w.Header().Set("Content-Type", "application/json")
			w.WriteHeader(http.StatusOK)
			_, _ = w.Write([]byte(`{"name":"Acme Inc"}`))
		}))
		defer srv.Close()

		client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

		got, err := NewHerokuNameResolver(client, "acme").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Equal(t, "Acme Inc", got)
	})
}

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

	cases := []struct {
		name      string
		org       string
		status    int
		body      string
		want      string
		wantErr   bool
		wantNoReq bool
	}{
		{
			name:      "empty org returns empty name without HTTP call",
			org:       "",
			wantNoReq: true,
			want:      "",
		},
		{
			name:   "200 with name",
			org:    "acme",
			status: http.StatusOK,
			body:   `{"name":"Acme Inc"}`,
			want:   "Acme Inc",
		},
		{
			name:   "200 with empty name falls back to org slug",
			org:    "acme",
			status: http.StatusOK,
			body:   `{"name":""}`,
			want:   "acme",
		},
		{
			name:    "404 errors",
			org:     "missing",
			status:  http.StatusNotFound,
			body:    `{"message":"Not Found"}`,
			wantErr: true,
		},
		{
			name:    "500 errors",
			org:     "acme",
			status:  http.StatusInternalServerError,
			body:    `{"message":"boom"}`,
			wantErr: true,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			var called bool

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				called = true

				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/orgs/"+tc.org, r.URL.Path)
				assert.Equal(t, "application/vnd.github+json", r.Header.Get("Accept"))
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewGitHubNameResolver(client, tc.org).ResolveInstanceName(context.Background())
			if tc.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			assert.Equal(t, tc.want, got)

			if tc.wantNoReq {
				assert.False(t, called, "expected no HTTP call when org is empty")
			} else {
				assert.True(t, called, "expected HTTP call when org is non-empty")
			}
		})
	}
}

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

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "single workspace names the source",
			status: http.StatusOK,
			body:   `{"data":{"me":{"name":"Jane Doe","workspaces":[{"id":"w1","name":"Acme Workspace"}]}}}`,
			want:   "Acme Workspace",
		},
		{
			name:   "multiple workspaces fall back to account holder",
			status: http.StatusOK,
			body:   `{"data":{"me":{"name":"Jane Doe","workspaces":[{"id":"w1","name":"Acme"},{"id":"w2","name":"Beta"}]}}}`,
			want:   "Jane Doe",
		},
		{
			name:   "no workspaces fall back to account holder",
			status: http.StatusOK,
			body:   `{"data":{"me":{"name":"Jane Doe","workspaces":[]}}}`,
			want:   "Jane Doe",
		},
		{
			name:   "graphql error body is terminal (no name)",
			status: http.StatusOK,
			body:   `{"data":{"me":null},"errors":[{"message":"unauthorized"}]}`,
			want:   "",
		},
		{
			name:   "null me is terminal (no name)",
			status: http.StatusOK,
			body:   `{"data":{"me":null}}`,
			want:   "",
		},
		{
			name:   "non-2xx is terminal (no name)",
			status: http.StatusInternalServerError,
			body:   `{"message":"boom"}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodPost, r.Method)
				assert.Equal(t, "/graphql/v2", r.URL.Path)
				assert.Equal(t, "application/json", r.Header.Get("Content-Type"))
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewRailwayNameResolver(client).ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("empty website id returns nothing without HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for an empty website id")
			return nil, nil
		})}

		got, err := NewCrispNameResolver(client, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, got)
	})

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "200 returns website name",
			status: http.StatusOK,
			body:   `{"data":{"name":"Acme Support"}}`,
			want:   "Acme Support",
		},
		{
			name:   "401 is terminal (no name)",
			status: http.StatusUnauthorized,
			body:   `{"error":true,"reason":"not_allowed"}`,
			want:   "",
		},
		{
			name:   "404 is terminal (no name)",
			status: http.StatusNotFound,
			body:   `{"error":true,"reason":"website_not_found"}`,
			want:   "",
		},
		{
			name:   "500 is terminal (no name)",
			status: http.StatusInternalServerError,
			body:   `{"error":true,"reason":"boom"}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/v1/website/1a2b3c4d", r.URL.Path)
				assert.Equal(t, crispTierValue, r.Header.Get(crispTierHeader))
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewCrispNameResolver(client, "1a2b3c4d").ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

// roundTripperFunc adapts a function into an http.RoundTripper, useful for
// asserting that a resolver short-circuits before making any HTTP call.
type roundTripperFunc func(*http.Request) (*http.Response, error)

func (f roundTripperFunc) RoundTrip(r *http.Request) (*http.Response, error) {
	return f(r)
}

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

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "200 returns business name",
			status: http.StatusOK,
			body:   `{"merchant":{"business_name":"Acme Coffee"}}`,
			want:   "Acme Coffee",
		},
		{
			name:   "401 is terminal (no name)",
			status: http.StatusUnauthorized,
			body:   `{"errors":[{"code":"UNAUTHORIZED"}]}`,
			want:   "",
		},
		{
			name:   "403 is terminal (no name)",
			status: http.StatusForbidden,
			body:   `{"errors":[{"code":"FORBIDDEN"}]}`,
			want:   "",
		},
		{
			name:   "500 is terminal (no name)",
			status: http.StatusInternalServerError,
			body:   `{"errors":[{"code":"INTERNAL_SERVER_ERROR"}]}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/v2/merchants/me", r.URL.Path)
				assert.Equal(t, squareAPIVersion, r.Header.Get("Square-Version"))
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewSquareNameResolver(client).ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("empty account id returns nothing without HTTP call", func(t *testing.T) {
		t.Parallel()

		client := &http.Client{Transport: roundTripperFunc(func(*http.Request) (*http.Response, error) {
			t.Fatalf("resolver should not make an HTTP call for an empty account id")
			return nil, nil
		})}

		got, err := NewGoogleAnalyticsNameResolver(client, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, got)
	})

	cases := []struct {
		name   string
		status int
		body   string
		want   string
	}{
		{
			name:   "200 returns display name",
			status: http.StatusOK,
			body:   `{"displayName":"Acme Analytics"}`,
			want:   "Acme Analytics",
		},
		{
			name:   "401 is terminal (no name)",
			status: http.StatusUnauthorized,
			body:   `{"error":{"code":401}}`,
			want:   "",
		},
		{
			name:   "403 is terminal (no name)",
			status: http.StatusForbidden,
			body:   `{"error":{"code":403}}`,
			want:   "",
		},
		{
			name:   "404 is terminal (no name)",
			status: http.StatusNotFound,
			body:   `{"error":{"code":404}}`,
			want:   "",
		},
		{
			name:   "500 is terminal (no name)",
			status: http.StatusInternalServerError,
			body:   `{"error":{"code":500}}`,
			want:   "",
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/v1alpha/accounts/123456", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewGoogleAnalyticsNameResolver(client, "123456").ResolveInstanceName(context.Background())
			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

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

	t.Run("reads the workspace name from the API root", func(t *testing.T) {
		t.Parallel()

		var gotPath string

		srv := httptest.NewServer(
			http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				gotPath = r.URL.Path
				_, _ = w.Write([]byte(`{"data":{"workspace":{"id":"9aQ1Lj62S4bomZKLF4DPqW","name":"Acme Prod","slug":"acme-prod"}}}`))
			}),
		)
		t.Cleanup(srv.Close)

		name, err := NewSegmentNameResolver(srv.Client(), srv.URL).ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Equal(t, "Acme Prod", name)
		// Get Workspace is the API root, not /workspace or /workspaces/{id}.
		assert.Equal(t, "/", gotPath)
	})

	// A revoked token must not make the source-name worker retry forever.
	t.Run("a client error is terminal", func(t *testing.T) {
		t.Parallel()

		srv := httptest.NewServer(
			http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
				w.WriteHeader(http.StatusUnauthorized)
			}),
		)
		t.Cleanup(srv.Close)

		_, err := NewSegmentNameResolver(srv.Client(), srv.URL).ResolveInstanceName(context.Background())
		require.Error(t, err)
		assert.ErrorIs(t, err, ErrTerminalNameResolution)
	})

	t.Run("a server error stays retryable", func(t *testing.T) {
		t.Parallel()

		srv := httptest.NewServer(
			http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
				w.WriteHeader(http.StatusBadGateway)
			}),
		)
		t.Cleanup(srv.Close)

		_, err := NewSegmentNameResolver(srv.Client(), srv.URL).ResolveInstanceName(context.Background())
		require.Error(t, err)
		assert.False(t, errors.Is(err, ErrTerminalNameResolution))
	})

	t.Run("an unset base URL resolves to nothing", func(t *testing.T) {
		t.Parallel()

		name, err := NewSegmentNameResolver(http.DefaultClient, "").ResolveInstanceName(context.Background())
		require.NoError(t, err)
		assert.Empty(t, name)
	})
}

// The Segment Public API declares permissions on the shared UserV1 schema but
// today only populates it on the single-user read, so the driver falls back to
// GET /users/{id}. This pins the other branch: when the list does carry
// permissions, no per-user request is made. The httptest server fails the test
// if one is.
func TestSegmentDriverUsesInlinePermissions(t *testing.T) {
	t.Parallel()

	var perUserCalls int

	srv := httptest.NewServer(
		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			switch r.URL.Path {
			case "/users":
				_, _ = w.Write([]byte(`{"data":{"users":[{"id":"u1","name":"Ada","email":"ada@example.com","permissions":[{"roleName":"Workspace Owner"}]},{"id":"u2","name":"Bob","email":"bob@example.com","permissions":[]}],"pagination":{}}}`))
			case "/invites":
				_, _ = w.Write([]byte(`{"data":{"invites":[],"pagination":{}}}`))
			default:
				perUserCalls++

				w.WriteHeader(http.StatusInternalServerError)
			}
		}),
	)
	t.Cleanup(srv.Close)

	records, err := NewSegmentDriver(srv.Client(), srv.URL).ListAccounts(context.Background())
	require.NoError(t, err)
	require.Len(t, records, 2)

	assert.Zero(t, perUserCalls, "inline permissions must not trigger the per-user fetch")

	assert.Equal(t, "ada@example.com", records[0].Email)
	assert.True(t, records[0].IsAdmin)
	assert.Equal(t, []string{"Workspace Owner"}, records[0].Roles)

	// An empty (but present) permissions array is authoritative: the user
	// genuinely has no roles, so it must not be mistaken for "not populated".
	assert.Equal(t, "bob@example.com", records[1].Email)
	assert.False(t, records[1].IsAdmin)
	assert.Empty(t, records[1].Roles)
}

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

	cases := []struct {
		name       string
		status     int
		body       string
		want       string
		wantErr    bool
		isTerminal bool
	}{
		{
			name:   "main account username",
			status: http.StatusOK,
			body:   `{"account":{"credits":50000,"username":"aureliens"}}`,
			want:   "aureliens",
		},
		{
			name:   "no username in payload",
			status: http.StatusOK,
			body:   `{"account":{"credits":0}}`,
			want:   "",
		},
		{
			name:       "revoked token is terminal",
			status:     http.StatusUnauthorized,
			body:       `{"error":{"error_code":"AUTHENTICATION_FAILED"}}`,
			wantErr:    true,
			isTerminal: true,
		},
		{
			name:    "server error stays retryable",
			status:  http.StatusInternalServerError,
			body:    `{"error":{"error_code":"BOOM"}}`,
			wantErr: true,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				assert.Equal(t, http.MethodGet, r.Method)
				assert.Equal(t, "/1.3/account", r.URL.Path)
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(tc.status)
				_, _ = w.Write([]byte(tc.body))
			}))
			defer srv.Close()

			client := &http.Client{Transport: &hostRewriter{target: srv.URL}}

			got, err := NewUpCloudNameResolver(client).ResolveInstanceName(context.Background())
			if tc.wantErr {
				require.Error(t, err)
				assert.Equal(t, tc.isTerminal, errors.Is(err, ErrTerminalNameResolution))

				return
			}

			require.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}