49 lines
1.5 KiB
Go
49 lines
1.5 KiB
Go
|
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...))
|
||
|
}
|