// NewTestClient will return a new heroku.Client that's configured to interact
// with a instance of the empire HTTP server.
func NewTestClient(t testing.TB) (*heroku.Client, *empiretest.Server) {
e := empiretest.NewEmpire(t)
s := empiretest.NewTestServer(t, e, server.Options{
Authenticator: auth.Anyone(&empire.User{Name: "fake"}),
})
c := &heroku.Client{
Username: "",
Password: "",
}
c.URL = s.URL
return c, s
}
func TestLoginTwoFactor(t *testing.T) {
s := empiretest.NewTestServer(t, nil, server.Options{
Authenticator: auth.StaticAuthenticator("twofactor", "bar", "code", &empire.User{Name: "fake"}),
})
defer s.Close()
input := "twofactor\nbar\ncode\n"
cmd := NewCmd(s.URL, "login")
cmd.Stdin = strings.NewReader(input)
out, err := cmd.CombinedOutput()
if err != nil {
t.Fatal(err)
}
if got, want := string(out), "Enter email: Enter two-factor auth code: Logged in.\n"; got != want {
t.Fatalf("%q", got)
}
}
func TestLoginUnauthorized(t *testing.T) {
s := empiretest.NewTestServer(t, nil, server.Options{
Authenticator: auth.StaticAuthenticator("fake", "bar", "", &empire.User{Name: "fake"}),
})
defer s.Close()
input := "foo\nbar\n"
cmd := NewCmd(s.URL, "login")
cmd.Stdin = strings.NewReader(input)
out, err := cmd.CombinedOutput()
if err == nil {
t.Fatal("Expected an error")
}
if got, want := string(out), "Enter email: error: Request not authenticated, API token is missing, invalid or expired Log in with `emp login`.\n"; got != want {
t.Fatalf("%q", got)
}
}
func TestLogin(t *testing.T) {
s := empiretest.NewTestServer(t, nil, server.Options{
Authenticator: auth.StaticAuthenticator("fake", "bar", "", &empire.User{Name: "fake"}),
})
cli := newCLIWithServer(t, s)
defer cli.Close()
input := "fake\nbar\n"
cmd := cli.Command("login")
cmd.Stdin = strings.NewReader(input)
out, err := cmd.CombinedOutput()
if err != nil {
t.Fatal(err)
}
if got, want := string(out), "Enter email: Logged in.\n"; got != want {
t.Fatalf("%q", got)
}
}