Function TestResolveCapiURL

pkg/cmd/agent-task/shared/capi_test.go:16–62

(t *testing.T)

)

func TestResolveCapiURL(t *testing.T) {
	tests := []struct {
		name    string
		resp    string
		wantURL string
		wantErr bool
	}{
		{
			name:    "returns resolved URL",
			resp:    `{"data":{"viewer":{"copilotEndpoints":{"api":"https://test-copilot-api.example.com"}}}}`,
			wantURL: "https://test-copilot-api.example.com",
		},
		{
			name:    "ghe.com tenant URL",
			resp:    `{"data":{"viewer":{"copilotEndpoints":{"api":"https://test-copilot-api.tenant.example.com"}}}}`,
			wantURL: "https://test-copilot-api.tenant.example.com",
		},
		{
			name:    "empty URL returns error",
			resp:    `{"data":{"viewer":{"copilotEndpoints":{"api":""}}}}`,
			wantErr: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			reg := &httpmock.Registry{}
			defer reg.Verify(t)

			reg.Register(
				httpmock.GraphQL(`query CopilotEndpoints\b`),
				httpmock.StringResponse(tt.resp),
			)

			httpClient := &http.Client{Transport: reg}
			url, err := resolveCapiURL(httpClient, "github.com")

			if tt.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			assert.Equal(t, tt.wantURL, url)
		})
	}
}

func TestCapiClientFuncResolvesURL(t *testing.T) {
	reg := &httpmock.Registry{}