3a8417f6c7
IsEnterprise currently returns `true` when `IsTenancy` returns true. We prefer that this behavior is orthogonal. This commit adds failing tests to the instances where IsEnterprise should not behave the same as IsTenancy. These test cases are expected to pass with the inclusion of the coming improvements to go-gh to handle the orthogonality of IsEnterprise and IsTenancy.
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package status
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/cli/cli/v2/api"
|
|
"github.com/cli/cli/v2/pkg/httpmock"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_getCurrentUsername(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
username string
|
|
hostname string
|
|
serverUsername string
|
|
currentUsername string
|
|
}{
|
|
{
|
|
name: "dotcom",
|
|
username: "@me",
|
|
hostname: "github.com",
|
|
currentUsername: "@me",
|
|
},
|
|
{
|
|
name: "ghec data residency",
|
|
username: "@me",
|
|
hostname: "stampname.ghe.com",
|
|
currentUsername: "@me",
|
|
},
|
|
{
|
|
name: "ghes",
|
|
username: "@me",
|
|
hostname: "my.server.com",
|
|
serverUsername: "@serverUserName",
|
|
currentUsername: "@serverUserName",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
clientStub := &http.Client{}
|
|
if tt.serverUsername != "" {
|
|
reg := &httpmock.Registry{}
|
|
defer reg.Verify(t)
|
|
|
|
reg.Register(
|
|
httpmock.GraphQL(`query UserCurrent\b`),
|
|
httpmock.StringResponse(`{"data":{"viewer":{"login":"`+tt.serverUsername+`"}}}`),
|
|
)
|
|
|
|
clientStub.Transport = reg
|
|
}
|
|
|
|
apiClientStub := api.NewClientFromHTTP(clientStub)
|
|
|
|
currentUsername, err := getCurrentUsername(tt.username, tt.hostname, apiClientStub)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, tt.currentUsername, currentUsername)
|
|
})
|
|
}
|
|
}
|