STACKITGit/cli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests: add tests for GPG key commands

Browse source
This commit is contained in:
Owen Voke 2021-06-11 15:55:31 +01:00
parent dad2d52571
commit d96a773afc
No known key found for this signature in database
GPG key ID: B6A92A86D7A46D64
2 changed files with 175 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

42
pkg/cmd/gpg-key/add/add_test.go Normal file
View file

@ -0,0 +1,42 @@
package add
import (
"net/http"
"testing"
"github.com/cli/cli/internal/config"
"github.com/cli/cli/pkg/httpmock"
"github.com/cli/cli/pkg/iostreams"
"github.com/stretchr/testify/assert"
)
func Test_runAdd(t *testing.T) {
io, stdin, stdout, stderr := iostreams.Test()
io.SetStdinTTY(false)
io.SetStdoutTTY(true)
io.SetStderrTTY(true)
stdin.WriteString("PUBKEY")
tr := httpmock.Registry{}
defer tr.Verify(t)
tr.Register(
httpmock.REST("POST", "user/gpg_keys"),
httpmock.StringResponse(`{}`))
err := runAdd(&AddOptions{
IO: io,
Config: func() (config.Config, error) {
return config.NewBlankConfig(), nil
},
HTTPClient: func() (*http.Client, error) {
return &http.Client{Transport: &tr}, nil
},
KeyFile: "-",
})
assert.NoError(t, err)
assert.Equal(t, "", stdout.String())
assert.Equal(t, "✓ GPG public key added to your account\n", stderr.String())
}

133
pkg/cmd/gpg-key/list/list_test.go Normal file
View file

@ -0,0 +1,133 @@
package list
import (
"fmt"
"net/http"
"testing"
"time"
"github.com/MakeNowJust/heredoc"
"github.com/cli/cli/internal/config"
"github.com/cli/cli/pkg/httpmock"
"github.com/cli/cli/pkg/iostreams"
)
func TestListRun(t *testing.T) {
tests := []struct {
name string
opts ListOptions
isTTY bool
wantStdout string
wantStderr string
wantErr bool
}{
{
name: "list tty",
opts: ListOptions{
HTTPClient: func() (*http.Client, error) {
expiresAt, _ := time.Parse(time.RFC3339, "2021-06-11T15:44:24+01:00")
reg := &httpmock.Registry{}
reg.Register(
httpmock.REST("GET", "user/gpg_keys"),
httpmock.StringResponse(fmt.Sprintf(`[
{
"id": 1234,
"key_id": "ABCDEF1234567890",
"public_key": "xJMEWfoofoofoo",
"expires_at": "%[1]s"
},
{
"id": 5678,
"key_id": "1234567890ABCDEF",
"public_key": "xJMEWbarbarbar",
"expires_at": "%[1]s"
}
]`, expiresAt.Format(time.RFC3339))),
)
return &http.Client{Transport: reg}, nil
},
},
isTTY: true,
wantStdout: heredoc.Doc(`
ABCDEF1234567890 2021-06-11T15:44:24+01:00 xJMEWfoofoofoo
1234567890ABCDEF 2021-06-11T15:44:24+01:00 xJMEWbarbarbar
`),
wantStderr: "",
},
{
name: "list non-tty",
opts: ListOptions{
HTTPClient: func() (*http.Client, error) {
expiresAt, _ := time.Parse(time.RFC3339, "2021-06-11T15:44:24+01:00")
reg := &httpmock.Registry{}
reg.Register(
httpmock.REST("GET", "user/gpg_keys"),
httpmock.StringResponse(fmt.Sprintf(`[
{
"id": 1234,
"key_id": "ABCDEF1234567890",
"public_key": "xJMEWfoofoofoo",
"expires_at": "%[1]s"
},
{
"id": 5678,
"key_id": "1234567890ABCDEF",
"public_key": "xJMEWbarbarbar",
"expires_at": "%[1]s"
}
]`, expiresAt.Format(time.RFC3339))),
)
return &http.Client{Transport: reg}, nil
},
},
isTTY: false,
wantStdout: heredoc.Doc(`
ABCDEF1234567890 2021-06-11T15:44:24+01:00 xJMEWfoofoofoo
1234567890ABCDEF 2021-06-11T15:44:24+01:00 xJMEWbarbarbar
`),
wantStderr: "",
},
{
name: "no keys",
opts: ListOptions{
HTTPClient: func() (*http.Client, error) {
reg := &httpmock.Registry{}
reg.Register(
httpmock.REST("GET", "user/gpg_keys"),
httpmock.StringResponse(`[]`),
)
return &http.Client{Transport: reg}, nil
},
},
wantStdout: "",
wantStderr: "No GPG keys present in GitHub account.\n",
wantErr: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
io, _, stdout, stderr := iostreams.Test()
io.SetStdoutTTY(tt.isTTY)
io.SetStdinTTY(tt.isTTY)
io.SetStderrTTY(tt.isTTY)
opts := tt.opts
opts.IO = io
opts.Config = func() (config.Config, error) { return config.NewBlankConfig(), nil }
err := listRun(&opts)
if (err != nil) != tt.wantErr {
t.Errorf("linRun() return error: %v", err)
return
}
if stdout.String() != tt.wantStdout {
t.Errorf("wants stdout %q, got %q", tt.wantStdout, stdout.String())
}
if stderr.String() != tt.wantStderr {
t.Errorf("wants stderr %q, got %q", tt.wantStderr, stderr.String())
}
})
}
}