package cmd import ( "fmt" "github.com/stretchr/testify/require" "github.com/urfave/cli/v2" "heckel.io/ntfy/server" "heckel.io/ntfy/test" "regexp" "testing" ) func TestCLI_Token_AddListRemove(t *testing.T) { s, conf, port := newTestServerWithAuth(t) defer test.StopServer(t, s, port) app, stdin, _, stderr := newTestApp() stdin.WriteString("mypass\nmypass") require.Nil(t, runUserCommand(app, conf, "add", "phil")) require.Contains(t, stderr.String(), "user phil added with role user") app, _, _, stderr = newTestApp() require.Nil(t, runTokenCommand(app, conf, "add", "phil")) require.Regexp(t, `token tk_.+ created for user phil, never expires`, stderr.String()) app, _, _, stderr = newTestApp() require.Nil(t, runTokenCommand(app, conf, "list", "phil")) require.Regexp(t, `user phil\n- tk_.+, never expires, accessed from 0.0.0.0 at .+`, stderr.String()) re := regexp.MustCompile(`tk_\w+`) token := re.FindString(stderr.String()) app, _, _, stderr = newTestApp() require.Nil(t, runTokenCommand(app, conf, "remove", "phil", token)) require.Regexp(t, fmt.Sprintf("token %s for user phil removed", token), stderr.String()) app, _, _, stderr = newTestApp() require.Nil(t, runTokenCommand(app, conf, "list")) require.Equal(t, "no users with tokens\n", stderr.String()) } func runTokenCommand(app *cli.App, conf *server.Config, args ...string) error { userArgs := []string{ "ntfy", "token", "--auth-file=" + conf.AuthFile, } return app.Run(append(userArgs, args...)) }