xai_test.go RawHistoryFolder

1package auth
2
3import (
4	"context"
5	"encoding/json"
6	"net/http"
7	"net/http/httptest"
8	"net/url"
9	"testing"
10)
11
12func TestRequestXAIDeviceAuthorizationBuildsPrefilledURL(t *testing.T) {
13	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
14		if err := r.ParseForm(); err != nil {
15			t.Fatal(err)
16		}
17		if r.Form.Get("client_id") != xaiClientID || r.Form.Get("scope") != xaiScope {
18			t.Fatalf("unexpected form: %v", r.Form)
19		}
20		_ = json.NewEncoder(w).Encode(map[string]any{
21			"device_code": "device", "user_code": "AB CD",
22			"verification_uri": "https://accounts.example/activate",
23			"expires_in":       600,
24		})
25	}))
26	defer server.Close()
27
28	old := xaiDeviceCodeURL
29	xaiDeviceCodeURL = server.URL
30	t.Cleanup(func() { xaiDeviceCodeURL = old })
31
32	device, err := RequestXAIDeviceAuthorization(context.Background())
33	if err != nil {
34		t.Fatal(err)
35	}
36	u, err := url.Parse(device.VerificationURIComplete)
37	if err != nil {
38		t.Fatal(err)
39	}
40	if got := u.Query().Get("user_code"); got != "AB CD" {
41		t.Fatalf("user_code = %q", got)
42	}
43}
44
45func TestRefreshXAITokenPreservesRefreshToken(t *testing.T) {
46	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
47		if err := r.ParseForm(); err != nil {
48			t.Fatal(err)
49		}
50		if r.Form.Get("grant_type") != "refresh_token" || r.Form.Get("refresh_token") != "old-refresh" {
51			t.Fatalf("unexpected form: %v", r.Form)
52		}
53		_ = json.NewEncoder(w).Encode(map[string]any{
54			"access_token": "new-access", "expires_in": 3600,
55		})
56	}))
57	defer server.Close()
58
59	old := xaiTokenURL
60	xaiTokenURL = server.URL
61	t.Cleanup(func() { xaiTokenURL = old })
62
63	token, err := RefreshXAIToken(context.Background(), "old-refresh")
64	if err != nil {
65		t.Fatal(err)
66	}
67	if token.AccessToken != "new-access" || token.RefreshToken != "old-refresh" {
68		t.Fatalf("token = %+v", token)
69	}
70}