mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-03 14:39:26 +08:00
e1bf893956
- Add `util.ResolveGitHubToken` to resolve GitHub API tokens with priority order (`GITHUB_TOKEN`, `github_token`, and `GITSTORE_GIT_TOKEN` for GitHub repositories). - Set GitHub `Authorization` headers in management version check and asset updater requests when a token is resolved. Closes: #5189
63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
package util
|
|
|
|
import "testing"
|
|
|
|
func TestResolveGitHubToken(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
githubToken string
|
|
lowerToken string
|
|
gitStoreToken string
|
|
gitStoreURL string
|
|
want string
|
|
}{
|
|
{
|
|
name: "GITHUB_TOKEN has highest priority",
|
|
githubToken: " primary-token ",
|
|
lowerToken: "lower-token",
|
|
gitStoreToken: "gitstore-token",
|
|
gitStoreURL: "https://github.com/example/repo.git",
|
|
want: "primary-token",
|
|
},
|
|
{
|
|
name: "lowercase token is second priority",
|
|
githubToken: " ",
|
|
lowerToken: " lower-token ",
|
|
gitStoreToken: "gitstore-token",
|
|
gitStoreURL: "https://github.com/example/repo.git",
|
|
want: "lower-token",
|
|
},
|
|
{
|
|
name: "Git store token is used for GitHub",
|
|
gitStoreToken: " gitstore-token ",
|
|
gitStoreURL: "HTTPS://GITHUB.COM/example/repo.git",
|
|
want: "gitstore-token",
|
|
},
|
|
{
|
|
name: "Git store token is ignored for other hosts",
|
|
gitStoreToken: "gitstore-token",
|
|
gitStoreURL: "https://gitlab.com/example/repo.git",
|
|
},
|
|
{
|
|
name: "Git store token is ignored without URL",
|
|
gitStoreToken: "gitstore-token",
|
|
},
|
|
{
|
|
name: "no token configured",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Setenv("GITHUB_TOKEN", tt.githubToken)
|
|
t.Setenv("github_token", tt.lowerToken)
|
|
t.Setenv("GITSTORE_GIT_TOKEN", tt.gitStoreToken)
|
|
t.Setenv("GITSTORE_GIT_URL", tt.gitStoreURL)
|
|
|
|
if got := ResolveGitHubToken(); got != tt.want {
|
|
t.Fatalf("ResolveGitHubToken() = %q, want %q", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|