Function TestTokenRefreshOnExpiry

internal/auth/auth_test.go:88–131

(t *testing.T)

}

func TestTokenRefreshOnExpiry(t *testing.T) {
	refreshCalls := 0
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/oauth/tokens" {
			t.Fatalf("unexpected path: %s", r.URL.Path)
		}
		refreshCalls++
		resp := OAuthToken{
			AccessToken:  "refreshed-token",
			RefreshToken: "new-refresh",
			ExpiresIn:    3600,
		}
		w.Header().Set("Content-Type", "application/json")
		json.NewEncoder(w).Encode(resp)
	}))
	defer server.Close()

	t.Setenv("HEY_TOKEN", "")
	mgr := testManager(t, server)

	// Save credentials that are expired (ExpiresAt in the past)
	expired := &Credentials{
		AccessToken:   "old-token",
		RefreshToken:  "refresh-tok",
		ExpiresAt:     time.Now().Unix() - 600,
		OAuthType:     "oauth",
		TokenEndpoint: fmt.Sprintf("%s/oauth/tokens", server.URL),
	}
	if err := mgr.GetStore().Save(mgr.CredentialKey(), expired); err != nil {
		t.Fatalf("Save: %v", err)
	}

	token, err := mgr.AccessToken(context.Background())
	if err != nil {
		t.Fatalf("AccessToken: %v", err)
	}

	if token != "refreshed-token" {
		t.Errorf("token = %q, want %q", token, "refreshed-token")
	}
	if refreshCalls != 1 {
		t.Errorf("refresh calls = %d, want 1", refreshCalls)
	}
}