Function TestServerNotifications

cli/command_server_notifications_test.go:21–104

(t *testing.T)

)

func TestServerNotifications(t *testing.T) {
	mux := http.NewServeMux()

	notificationsReceived := make(chan string, 100)

	mux.HandleFunc("/notification-webhook", func(w http.ResponseWriter, r *http.Request) {
		var b bytes.Buffer

		io.Copy(&b, r.Body)

		notificationsReceived <- b.String()
	})

	server := httptest.NewServer(mux)
	defer server.Close()

	env := testenv.NewCLITest(t, testenv.RepoFormatNotImportant, testenv.NewInProcRunner(t))

	dir0 := testutil.TempDirectory(t)
	dir1 := testutil.TempDirectory(t)
	dir2 := testutil.TempDirectory(t)

	env.RunAndExpectSuccess(t, "repo", "create", "filesystem", "--path", env.RepoDir, "--override-username=another-user", "--override-hostname=another-host")

	env.RunAndExpectSuccess(t, "snap", "create", dir0)

	env.RunAndExpectSuccess(t, "repo", "connect", "filesystem", "--path", env.RepoDir, "--override-username=test-user", "--override-hostname=test-host")
	env.RunAndExpectSuccess(t, "snap", "create", dir1)
	env.RunAndExpectSuccess(t, "snap", "create", dir2)

	// setup webhook notification
	env.RunAndExpectSuccess(t, "notification", "profile", "configure", "webhook", "--profile-name=mywebhook", "--endpoint="+server.URL+"/notification-webhook", "--method=POST", "--format=html")

	var sp testutil.ServerParameters

	jsonNotificationsReceived := make(chan string, 100)

	wait, kill := env.RunAndProcessStderrAsync(t, sp.ProcessOutput, func(line string) {
		const prefix = "NOTIFICATION: "

		if strings.HasPrefix(line, prefix) {
			t.Logf("JSON notification received: %v", line)

			jsonNotificationsReceived <- line[len(prefix):]
		}
	}, "server", "start",
		"--address=localhost:0",
		"--insecure",
		"--random-server-control-password",
		"--kopiaui-notifications",
		"--shutdown-grace-period", "100ms",
	)

	defer func() {
		kill()
		wait()
	}()