Luis Pater
e4c957078c
- Implemented xAI OAuth2 integration with PKCE (Proof Key for Code Exchange) support. - Added logic for token exchange, refresh, and persistent storage in JSON format. - Created `xai` package with helpers for OAuth discovery, API token handling, and URL building. - Introduced `XAIExecutor` for integrating xAI credentials into runtime HTTP requests. - Added unit tests to validate OAuth flow, token persistence, and endpoint validation.
38 lines
1.2 KiB
Go
38 lines
1.2 KiB
Go
package auth
|
|
|
|
import "testing"
|
|
|
|
func TestXAIAuthenticatorProviderAndRefreshLead(t *testing.T) {
|
|
authenticator := NewXAIAuthenticator()
|
|
if authenticator.Provider() != "xai" {
|
|
t.Fatalf("Provider() = %q, want xai", authenticator.Provider())
|
|
}
|
|
lead := authenticator.RefreshLead()
|
|
if lead == nil || *lead <= 0 {
|
|
t.Fatalf("RefreshLead() = %v, want positive duration", lead)
|
|
}
|
|
}
|
|
|
|
func TestParseXAIManualCallbackTokenAcceptsRawCode(t *testing.T) {
|
|
result, ok, err := parseXAIManualCallbackToken(" V0auoESADonzF4bY_Ag2whBFnVeqzHJm6nW2uW012rqCCW5cstFV58qvDFBvnPBXXe0rZSKOcs3PwwfACKp1qg ", "state-1")
|
|
if err != nil {
|
|
t.Fatalf("parseXAIManualCallbackToken() error = %v", err)
|
|
}
|
|
if !ok {
|
|
t.Fatal("parseXAIManualCallbackToken() ok = false, want true")
|
|
}
|
|
if result.Code != "V0auoESADonzF4bY_Ag2whBFnVeqzHJm6nW2uW012rqCCW5cstFV58qvDFBvnPBXXe0rZSKOcs3PwwfACKp1qg" {
|
|
t.Fatalf("Code = %q", result.Code)
|
|
}
|
|
if result.State != "state-1" {
|
|
t.Fatalf("State = %q, want state-1", result.State)
|
|
}
|
|
}
|
|
|
|
func TestParseXAIManualCallbackTokenRejectsCallbackURL(t *testing.T) {
|
|
_, _, err := parseXAIManualCallbackToken("http://127.0.0.1:56121/callback?state=state-1&code=token-1", "state-1")
|
|
if err == nil {
|
|
t.Fatal("parseXAIManualCallbackToken() error = nil, want error")
|
|
}
|
|
}
|