83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package openai
|
||
|
||
import (
|
||
"net/url"
|
||
"sync"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestSessionStore_Stop_Idempotent(t *testing.T) {
|
||
store := NewSessionStore()
|
||
|
||
store.Stop()
|
||
store.Stop()
|
||
|
||
select {
|
||
case <-store.stopCh:
|
||
// ok
|
||
case <-time.After(time.Second):
|
||
t.Fatal("stopCh 未关闭")
|
||
}
|
||
}
|
||
|
||
func TestSessionStore_Stop_Concurrent(t *testing.T) {
|
||
store := NewSessionStore()
|
||
|
||
var wg sync.WaitGroup
|
||
for range 50 {
|
||
wg.Add(1)
|
||
go func() {
|
||
defer wg.Done()
|
||
store.Stop()
|
||
}()
|
||
}
|
||
|
||
wg.Wait()
|
||
|
||
select {
|
||
case <-store.stopCh:
|
||
// ok
|
||
case <-time.After(time.Second):
|
||
t.Fatal("stopCh 未关闭")
|
||
}
|
||
}
|
||
|
||
func TestBuildAuthorizationURLForPlatform_OpenAI(t *testing.T) {
|
||
authURL := BuildAuthorizationURLForPlatform("state-1", "challenge-1", DefaultRedirectURI, OAuthPlatformOpenAI)
|
||
parsed, err := url.Parse(authURL)
|
||
if err != nil {
|
||
t.Fatalf("Parse URL failed: %v", err)
|
||
}
|
||
q := parsed.Query()
|
||
if got := q.Get("client_id"); got != ClientID {
|
||
t.Fatalf("client_id mismatch: got=%q want=%q", got, ClientID)
|
||
}
|
||
if got := q.Get("codex_cli_simplified_flow"); got != "true" {
|
||
t.Fatalf("codex flow mismatch: got=%q want=true", got)
|
||
}
|
||
if got := q.Get("id_token_add_organizations"); got != "true" {
|
||
t.Fatalf("id_token_add_organizations mismatch: got=%q want=true", got)
|
||
}
|
||
}
|
||
|
||
// TestBuildAuthorizationURLForPlatform_Sora 验证 Sora 平台复用 Codex CLI 的 client_id，
|
||
// 但不启用 codex_cli_simplified_flow。
|
||
func TestBuildAuthorizationURLForPlatform_Sora(t *testing.T) {
|
||
authURL := BuildAuthorizationURLForPlatform("state-2", "challenge-2", DefaultRedirectURI, OAuthPlatformSora)
|
||
parsed, err := url.Parse(authURL)
|
||
if err != nil {
|
||
t.Fatalf("Parse URL failed: %v", err)
|
||
}
|
||
q := parsed.Query()
|
||
if got := q.Get("client_id"); got != ClientID {
|
||
t.Fatalf("client_id mismatch: got=%q want=%q (Sora should reuse Codex CLI client_id)", got, ClientID)
|
||
}
|
||
if got := q.Get("codex_cli_simplified_flow"); got != "" {
|
||
t.Fatalf("codex flow should be empty for sora, got=%q", got)
|
||
}
|
||
if got := q.Get("id_token_add_organizations"); got != "true" {
|
||
t.Fatalf("id_token_add_organizations mismatch: got=%q want=true", got)
|
||
}
|
||
}
|