348 lines
7.6 KiB
Go
348 lines
7.6 KiB
Go
package list
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/MakeNowJust/heredoc"
|
|
"github.com/cli/cli/v2/internal/browser"
|
|
"github.com/cli/cli/v2/internal/config"
|
|
"github.com/cli/cli/v2/internal/ghrepo"
|
|
"github.com/cli/cli/v2/internal/run"
|
|
"github.com/cli/cli/v2/pkg/cmdutil"
|
|
"github.com/cli/cli/v2/pkg/httpmock"
|
|
"github.com/cli/cli/v2/pkg/iostreams"
|
|
"github.com/google/shlex"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_NewCmdList(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
args string
|
|
isTTY bool
|
|
want ListOptions
|
|
wantErr string
|
|
}{
|
|
{
|
|
name: "no arguments",
|
|
args: "",
|
|
isTTY: true,
|
|
want: ListOptions{
|
|
Limit: 30,
|
|
IncludeParents: true,
|
|
WebMode: false,
|
|
Organization: "",
|
|
},
|
|
},
|
|
{
|
|
name: "limit",
|
|
args: "--limit 1",
|
|
isTTY: true,
|
|
want: ListOptions{
|
|
Limit: 1,
|
|
IncludeParents: true,
|
|
WebMode: false,
|
|
Organization: "",
|
|
},
|
|
},
|
|
{
|
|
name: "include parents",
|
|
args: "--parents=false",
|
|
isTTY: true,
|
|
want: ListOptions{
|
|
Limit: 30,
|
|
IncludeParents: false,
|
|
WebMode: false,
|
|
Organization: "",
|
|
},
|
|
},
|
|
{
|
|
name: "org",
|
|
args: "--org \"my-org\"",
|
|
isTTY: true,
|
|
want: ListOptions{
|
|
Limit: 30,
|
|
IncludeParents: true,
|
|
WebMode: false,
|
|
Organization: "my-org",
|
|
},
|
|
},
|
|
{
|
|
name: "web mode",
|
|
args: "--web",
|
|
isTTY: true,
|
|
want: ListOptions{
|
|
Limit: 30,
|
|
IncludeParents: true,
|
|
WebMode: true,
|
|
Organization: "",
|
|
},
|
|
},
|
|
{
|
|
name: "invalid limit",
|
|
args: "--limit 0",
|
|
isTTY: true,
|
|
wantErr: "invalid limit: 0",
|
|
},
|
|
{
|
|
name: "repo and org specified",
|
|
args: "--org \"my-org\" -R \"owner/repo\"",
|
|
isTTY: true,
|
|
wantErr: "only one of --repo and --org may be specified",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ios, _, _, _ := iostreams.Test()
|
|
ios.SetStdoutTTY(tt.isTTY)
|
|
ios.SetStdinTTY(tt.isTTY)
|
|
ios.SetStderrTTY(tt.isTTY)
|
|
|
|
f := &cmdutil.Factory{
|
|
IOStreams: ios,
|
|
}
|
|
|
|
var opts *ListOptions
|
|
cmd := NewCmdList(f, func(o *ListOptions) error {
|
|
opts = o
|
|
return nil
|
|
})
|
|
cmd.PersistentFlags().StringP("repo", "R", "", "")
|
|
|
|
argv, err := shlex.Split(tt.args)
|
|
require.NoError(t, err)
|
|
cmd.SetArgs(argv)
|
|
|
|
cmd.SetIn(&bytes.Buffer{})
|
|
cmd.SetOut(io.Discard)
|
|
cmd.SetErr(io.Discard)
|
|
|
|
_, err = cmd.ExecuteC()
|
|
if tt.wantErr != "" {
|
|
require.EqualError(t, err, tt.wantErr)
|
|
return
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
assert.Equal(t, tt.want.Limit, opts.Limit)
|
|
assert.Equal(t, tt.want.WebMode, opts.WebMode)
|
|
assert.Equal(t, tt.want.Organization, opts.Organization)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_RulesetList_Web(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
stdoutTTY bool
|
|
wantStdout string
|
|
wantBrowse string
|
|
}{
|
|
{
|
|
name: "repo tty",
|
|
stdoutTTY: true,
|
|
wantStdout: "Opening github.com/OWNER/REPO in your browser.\n",
|
|
wantBrowse: "https://github.com/OWNER/REPO",
|
|
},
|
|
{
|
|
name: "org tty",
|
|
stdoutTTY: true,
|
|
wantStdout: "Opening github.com/OWNER/REPO in your browser.\n",
|
|
wantBrowse: "https://github.com/OWNER/REPO",
|
|
},
|
|
{
|
|
name: "repo non-tty",
|
|
stdoutTTY: false,
|
|
wantStdout: "",
|
|
wantBrowse: "https://github.com/OWNER/REPO",
|
|
},
|
|
{
|
|
name: "org non-tty",
|
|
stdoutTTY: false,
|
|
wantStdout: "",
|
|
wantBrowse: "https://github.com/OWNER/REPO",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
reg := &httpmock.Registry{}
|
|
reg.StubRepoInfoResponse("OWNER", "REPO", "main")
|
|
|
|
browser := &browser.Stub{}
|
|
opts := &ListOptions{
|
|
WebMode: true,
|
|
HttpClient: func() (*http.Client, error) {
|
|
return &http.Client{Transport: reg}, nil
|
|
},
|
|
BaseRepo: func() (ghrepo.Interface, error) {
|
|
return ghrepo.New("OWNER", "REPO"), nil
|
|
},
|
|
Browser: browser,
|
|
}
|
|
|
|
io, _, stdout, _ := iostreams.Test()
|
|
|
|
opts.IO = io
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
io.SetStdoutTTY(tt.stdoutTTY)
|
|
|
|
_, teardown := run.Stub()
|
|
defer teardown(t)
|
|
|
|
if err := listRun(opts); err != nil {
|
|
t.Errorf("listRun() error = %v", err)
|
|
}
|
|
assert.Equal(t, "", stdout.String())
|
|
assert.Equal(t, tt.wantStdout, stdout.String())
|
|
reg.Verify(t)
|
|
browser.Verify(t, tt.wantBrowse)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_listRun(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
isTTY bool
|
|
opts ListOptions
|
|
wantErr string
|
|
wantStdout string
|
|
wantStderr string
|
|
wantBrowse string
|
|
}{
|
|
{
|
|
name: "list repo rulesets",
|
|
isTTY: true,
|
|
wantStdout: heredoc.Doc(`
|
|
|
|
Showing 3 of 3 rulesets in OWNER/REPO
|
|
|
|
ID NAME SOURCE STATUS RULES
|
|
4 test OWNER/REPO (repo) evaluate 1
|
|
42 asdf OWNER/REPO (repo) active 2
|
|
77 foobar Org-Name (org) disabled 4
|
|
`),
|
|
wantStderr: "",
|
|
wantBrowse: "",
|
|
},
|
|
{
|
|
name: "list org rulesets",
|
|
isTTY: true,
|
|
opts: ListOptions{
|
|
Organization: "my-org",
|
|
},
|
|
wantStdout: heredoc.Doc(`
|
|
|
|
Showing 3 of 3 rulesets in my-org
|
|
|
|
ID NAME SOURCE STATUS RULES
|
|
4 test OWNER/REPO (repo) evaluate 1
|
|
42 asdf OWNER/REPO (repo) active 2
|
|
77 foobar Org-Name (org) disabled 4
|
|
`),
|
|
wantStderr: "",
|
|
wantBrowse: "",
|
|
},
|
|
{
|
|
name: "machine-readable",
|
|
isTTY: false,
|
|
wantStdout: heredoc.Doc(`
|
|
4 test OWNER/REPO (repo) evaluate 1
|
|
42 asdf OWNER/REPO (repo) active 2
|
|
77 foobar Org-Name (org) disabled 4
|
|
`),
|
|
wantStderr: "",
|
|
wantBrowse: "",
|
|
},
|
|
{
|
|
name: "repo web mode, TTY",
|
|
isTTY: true,
|
|
opts: ListOptions{
|
|
WebMode: true,
|
|
},
|
|
wantStdout: "Opening github.com/OWNER/REPO/rules in your browser.\n",
|
|
wantStderr: "",
|
|
wantBrowse: "https://github.com/OWNER/REPO/rules",
|
|
},
|
|
{
|
|
name: "org web mode, TTY",
|
|
isTTY: true,
|
|
opts: ListOptions{
|
|
WebMode: true,
|
|
Organization: "my-org",
|
|
},
|
|
wantStdout: "Opening github.com/organizations/my-org/settings/rules in your browser.\n",
|
|
wantStderr: "",
|
|
wantBrowse: "https://github.com/organizations/my-org/settings/rules",
|
|
},
|
|
{
|
|
name: "repo web mode, non-TTY",
|
|
isTTY: false,
|
|
opts: ListOptions{
|
|
WebMode: true,
|
|
},
|
|
wantStdout: "",
|
|
wantStderr: "",
|
|
wantBrowse: "https://github.com/OWNER/REPO/rules",
|
|
},
|
|
{
|
|
name: "org web mode, non-TTY",
|
|
isTTY: false,
|
|
opts: ListOptions{
|
|
WebMode: true,
|
|
Organization: "my-org",
|
|
},
|
|
wantStdout: "",
|
|
wantStderr: "",
|
|
wantBrowse: "https://github.com/organizations/my-org/settings/rules",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ios, _, stdout, stderr := iostreams.Test()
|
|
ios.SetStdoutTTY(tt.isTTY)
|
|
ios.SetStdinTTY(tt.isTTY)
|
|
ios.SetStderrTTY(tt.isTTY)
|
|
|
|
fakeHTTP := &httpmock.Registry{}
|
|
fakeHTTP.Register(httpmock.GraphQL(`query RulesetList\b`), httpmock.FileResponse("./fixtures/rulesetList.json"))
|
|
|
|
tt.opts.IO = ios
|
|
tt.opts.HttpClient = func() (*http.Client, error) {
|
|
return &http.Client{Transport: fakeHTTP}, nil
|
|
}
|
|
tt.opts.BaseRepo = func() (ghrepo.Interface, error) {
|
|
return ghrepo.FromFullName("OWNER/REPO")
|
|
}
|
|
browser := &browser.Stub{}
|
|
tt.opts.Browser = browser
|
|
tt.opts.Config = func() (config.Config, error) {
|
|
return config.NewBlankConfig(), nil
|
|
}
|
|
|
|
err := listRun(&tt.opts)
|
|
|
|
if tt.wantErr != "" {
|
|
require.EqualError(t, err, tt.wantErr)
|
|
return
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
if tt.wantBrowse != "" {
|
|
browser.Verify(t, tt.wantBrowse)
|
|
}
|
|
|
|
assert.Equal(t, tt.wantStdout, stdout.String())
|
|
assert.Equal(t, tt.wantStderr, stderr.String())
|
|
})
|
|
}
|
|
}
|