notifications_test.go

package main

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/shurcooL/notifications/fs"
	"golang.org/x/net/webdav"
)

// Test that visiting /notificationsv1 without being logged in
// redirects to /login.
func TestNotificationsV1RedirectsLogin(t *testing.T) {
	mux := http.NewServeMux()

	users, _, err := newUsersService(webdav.NewMemFS())
	if err != nil {
		t.Fatal(err)
	}
	initNotifications(mux, fs.NewService(webdav.NewMemFS(), users), nil, users, nil)

	req := httptest.NewRequest(http.MethodGet, "/notificationsv1", nil)
	rr := httptest.NewRecorder()
	mux.ServeHTTP(rr, req)
	resp := rr.Result()
	if got, want := resp.StatusCode, http.StatusSeeOther; got != want {
		t.Errorf("got status code %d %s, want %d %s", got, http.StatusText(got), want, http.StatusText(want))
	}
	if got, want := resp.Header.Get("Location"), "/login?return=%2Fnotificationsv1"; got != want {
		t.Errorf("got Location header %q, want %q", got, want)
	}
}