103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package shared
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/MakeNowJust/heredoc"
|
|
"github.com/cli/cli/internal/run"
|
|
"github.com/cli/cli/pkg/httpmock"
|
|
"github.com/cli/cli/pkg/iostreams"
|
|
"github.com/cli/cli/pkg/prompt"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type tinyConfig map[string]string
|
|
|
|
func (c tinyConfig) Get(host, key string) (string, error) {
|
|
return c[fmt.Sprintf("%s:%s", host, key)], nil
|
|
}
|
|
|
|
func (c tinyConfig) Set(host string, key string, value string) error {
|
|
c[fmt.Sprintf("%s:%s", host, key)] = value
|
|
return nil
|
|
}
|
|
|
|
func (c tinyConfig) Write() error {
|
|
return nil
|
|
}
|
|
|
|
func TestLogin_ssh(t *testing.T) {
|
|
dir := t.TempDir()
|
|
io, _, stdout, stderr := iostreams.Test()
|
|
|
|
tr := httpmock.Registry{}
|
|
defer tr.Verify(t)
|
|
|
|
tr.Register(
|
|
httpmock.REST("GET", "api/v3/"),
|
|
httpmock.ScopesResponder("repo,read:org"))
|
|
tr.Register(
|
|
httpmock.GraphQL(`query UserCurrent\b`),
|
|
httpmock.StringResponse(`{"data":{"viewer":{ "login": "monalisa" }}}`))
|
|
tr.Register(
|
|
httpmock.REST("POST", "api/v3/user/keys"),
|
|
httpmock.StringResponse(`{}`))
|
|
|
|
ask, askRestore := prompt.InitAskStubber()
|
|
defer askRestore()
|
|
|
|
ask.StubOne("SSH") // preferred protocol
|
|
ask.StubOne(true) // generate a new key
|
|
ask.StubOne("monkey") // enter a passphrase
|
|
ask.StubOne(1) // paste a token
|
|
ask.StubOne("ATOKEN") // token
|
|
|
|
rs, runRestore := run.Stub()
|
|
defer runRestore(t)
|
|
|
|
keyFile := filepath.Join(dir, "id_ed25519")
|
|
rs.Register(`ssh-keygen`, 0, "", func(args []string) {
|
|
expected := []string{
|
|
"ssh-keygen", "-t", "ed25519",
|
|
"-C", "",
|
|
"-N", "monkey",
|
|
"-f", keyFile,
|
|
}
|
|
assert.Equal(t, expected, args)
|
|
// simulate that the public key file has been generated
|
|
_ = ioutil.WriteFile(keyFile+".pub", []byte("PUBKEY"), 0600)
|
|
})
|
|
|
|
cfg := tinyConfig{}
|
|
|
|
err := Login(&LoginOptions{
|
|
IO: io,
|
|
Config: &cfg,
|
|
HTTPClient: &http.Client{Transport: &tr},
|
|
Hostname: "example.com",
|
|
Interactive: true,
|
|
sshContext: sshContext{
|
|
configDir: dir,
|
|
keygenExe: "ssh-keygen",
|
|
},
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, "", stdout.String())
|
|
assert.Equal(t, heredoc.Docf(`
|
|
Tip: you can generate a Personal Access Token here https://example.com/settings/tokens
|
|
The minimum required scopes are 'repo', 'read:org', 'admin:public_key'.
|
|
- gh config set -h example.com git_protocol ssh
|
|
✓ Configured git protocol
|
|
✓ Uploaded the SSH key to your GitHub account: %s.pub
|
|
✓ Logged in as monalisa
|
|
`, keyFile), stderr.String())
|
|
|
|
assert.Equal(t, "monalisa", cfg["example.com:user"])
|
|
assert.Equal(t, "ATOKEN", cfg["example.com:oauth_token"])
|
|
assert.Equal(t, "ssh", cfg["example.com:git_protocol"])
|
|
}
|