Files
github--github-mcp-server/pkg/utils/api_test.go
T
2026-07-13 12:37:57 +08:00

76 lines
1.8 KiB
Go

package utils //nolint:revive //TODO: figure out a better name for this package
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestParseAPIHost(t *testing.T) {
tests := []struct {
name string
input string
wantRestURL string
wantErr bool
}{
{
name: "empty string defaults to dotcom",
input: "",
wantRestURL: "https://api.github.com/",
},
{
name: "github.com hostname",
input: "https://github.com",
wantRestURL: "https://api.github.com/",
},
{
name: "subdomain of github.com",
input: "https://foo.github.com",
wantRestURL: "https://api.github.com/",
},
{
name: "hostname ending in github.com but not a subdomain",
input: "https://mycompanygithub.com",
wantRestURL: "https://mycompanygithub.com/api/v3/",
},
{
name: "hostname ending in notgithub.com",
input: "https://notgithub.com",
wantRestURL: "https://notgithub.com/api/v3/",
},
{
name: "ghe.com hostname",
input: "https://ghe.com",
wantRestURL: "https://api.ghe.com/",
},
{
name: "subdomain of ghe.com",
input: "https://mycompany.ghe.com",
wantRestURL: "https://api.mycompany.ghe.com/",
},
{
name: "hostname ending in ghe.com but not a subdomain",
input: "https://myghe.com",
wantRestURL: "https://myghe.com/api/v3/",
},
{
name: "missing scheme",
input: "github.com",
wantErr: true,
},
}
for _, tc := range tests {
t.Run(tc.name, func(t *testing.T) {
host, err := parseAPIHost(tc.input)
if tc.wantErr {
assert.Error(t, err)
return
}
require.NoError(t, err)
assert.Equal(t, tc.wantRestURL, host.restURL.String())
})
}
}