Add SSWS API-key Authorization scheme

Okta API tokens authenticate as "Authorization: SSWS <token>", a
scheme none of the existing API-key modes (Bearer, custom header,
Basic) can express. Add Registration.APIKeyAuthScheme, plumb it onto
APIKeyConnection.Scheme, and send it via a new schemeAuthTransport.

The three API-key presentations (BasicAuth, Header, Scheme) are
mutually exclusive; Register rejects setting more than one so a
misconfiguration fails at process start rather than silently.

Signed-off-by: Aurélien Sibiril <81782+aureliensibiril@users.noreply.github.com>
This commit is contained in:
Aurélien Sibiril
2026-06-04 14:44:51 +02:00
parent 5dd8769d19
commit 511472aca3
5 changed files with 122 additions and 5 deletions

View File

@@ -102,3 +102,34 @@ func TestAPIKeyConnection_Client_BearerDefault(t *testing.T) {
require.Truef(t, ok, "expected *oauth2Transport, got %T", client.Transport)
assert.Equal(t, "token", transport.token)
}
func TestSchemeAuthTransport_RoundTrip(t *testing.T) {
t.Parallel()
rec := &recordingRoundTripper{}
transport := &schemeAuthTransport{scheme: "SSWS", token: "00aBcDeF", underlying: rec}
req := httptest.NewRequest(http.MethodGet, "https://acme.okta.com/api/v1/users", nil)
_, err := transport.RoundTrip(req)
require.NoError(t, err)
require.NotNil(t, rec.lastRequest)
assert.Equal(t, "SSWS 00aBcDeF", rec.lastRequest.Header.Get("Authorization"))
// The original request must be left untouched (RoundTrip clones it).
assert.Empty(t, req.Header.Get("Authorization"), "original request must not be mutated")
}
func TestAPIKeyConnection_Client_Scheme(t *testing.T) {
t.Parallel()
conn := &APIKeyConnection{APIKey: "00aBcDeF", Scheme: "SSWS"}
client, err := conn.Client(context.Background())
require.NoError(t, err)
transport, ok := client.Transport.(*schemeAuthTransport)
require.Truef(t, ok, "expected *schemeAuthTransport, got %T", client.Transport)
assert.Equal(t, "SSWS", transport.scheme)
assert.Equal(t, "00aBcDeF", transport.token)
}