ntfy-server/cmd/webpush_test.go

package cmd

import (
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/require"
	"github.com/urfave/cli/v2"
	"heckel.io/ntfy/v2/server"
)

func TestCLI_WebPush_GenerateKeys(t *testing.T) {
	app, _, stdout, _ := newTestApp()
	require.Nil(t, runWebPushCommand(app, server.NewConfig(), "keys"))
	require.Contains(t, stdout.String(), "Web Push keys generated.")
}

func TestCLI_WebPush_WriteKeysToFile(t *testing.T) {
	tempDir := t.TempDir()
	t.Chdir(tempDir)
	app, _, stdout, _ := newTestApp()
	require.Nil(t, runWebPushCommand(app, server.NewConfig(), "keys", "--output-file=key-file.yaml"))
	require.Contains(t, stdout.String(), "Web Push keys written to key-file.yaml")
	require.FileExists(t, filepath.Join(tempDir, "key-file.yaml"))
}

func runWebPushCommand(app *cli.App, conf *server.Config, args ...string) error {
	webPushArgs := []string{
		"ntfy",
		"--log-level=ERROR",
		"webpush",
	}
	return app.Run(append(webPushArgs, args...))
}