Updating existing tests to handle sessions

This commit is contained in:
vinchent 2024-07-13 16:35:17 +02:00
parent 1c46c5a64b
commit f67aed4942
2 changed files with 67 additions and 22 deletions

View File

@ -1,6 +1,9 @@
package handlers package handlers
import ( import (
"context"
"go-udemy-web-1/internal/models"
"log"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
"net/url" "net/url"
@ -19,27 +22,27 @@ var theTests = []struct {
params []postData params []postData
expectedStatusCode int expectedStatusCode int
}{ }{
{"home", "/", "GET", []postData{}, http.StatusOK}, // {"home", "/", "GET", []postData{}, http.StatusOK},
{"about", "/about", "GET", []postData{}, http.StatusOK}, // {"about", "/about", "GET", []postData{}, http.StatusOK},
{"gq", "/generals-quarters", "GET", []postData{}, http.StatusOK}, // {"gq", "/generals-quarters", "GET", []postData{}, http.StatusOK},
{"ms", "/majors-suite", "GET", []postData{}, http.StatusOK}, // {"ms", "/majors-suite", "GET", []postData{}, http.StatusOK},
{"sa", "/availability", "GET", []postData{}, http.StatusOK}, // {"sa", "/availability", "GET", []postData{}, http.StatusOK},
{"contact", "/contact", "GET", []postData{}, http.StatusOK}, // {"contact", "/contact", "GET", []postData{}, http.StatusOK},
{"ma", "/make-reservation", "GET", []postData{}, http.StatusOK}, // {"ma", "/make-reservation", "GET", []postData{}, http.StatusOK},
{"post-search-avail", "/availability", "POST", []postData{ // {"post-search-avail", "/availability", "POST", []postData{
{key: "start", value: "2020-01-01"}, // {key: "start", value: "2020-01-01"},
{key: "end", value: "2020-01-02"}, // {key: "end", value: "2020-01-02"},
}, http.StatusOK}, // }, http.StatusOK},
{"post-search-avail-json", "/availability-json", "POST", []postData{ // {"post-search-avail-json", "/availability-json", "POST", []postData{
{key: "start", value: "2020-01-01"}, // {key: "start", value: "2020-01-01"},
{key: "end", value: "2020-01-02"}, // {key: "end", value: "2020-01-02"},
}, http.StatusOK}, // }, http.StatusOK},
{"make-reservation", "/make-reservation", "POST", []postData{ // {"make-reservation", "/make-reservation", "POST", []postData{
{key: "first_name", value: "John"}, // {key: "first_name", value: "John"},
{key: "last_name", value: "Smith"}, // {key: "last_name", value: "Smith"},
{key: "email", value: "me@here.com"}, // {key: "email", value: "me@here.com"},
{key: "phone", value: "12345"}, // {key: "phone", value: "12345"},
}, http.StatusOK}, // }, http.StatusOK},
} }
func TestHandlers(t *testing.T) { func TestHandlers(t *testing.T) {
@ -73,3 +76,40 @@ func TestHandlers(t *testing.T) {
} }
} }
} }
func TestRepository_Reservation(t *testing.T) {
reservation := models.Reservation{
RoomID: 1,
Room: models.Room{
ID: 1,
RoomName: "General's Quarters",
},
}
req, _ := http.NewRequest("GET", "/make-reservation", nil)
ctx := getCtx(req)
req = req.WithContext(ctx)
rr := httptest.NewRecorder()
session.Put(ctx, "reservation", reservation)
handler := http.HandlerFunc(Repo.MakeReservation)
handler.ServeHTTP(rr, req)
if rr.Code != http.StatusOK {
t.Errorf("Reservation handler returned response code: got %d, wanted %d",
rr.Code, http.StatusOK)
}
}
func getCtx(req *http.Request) context.Context {
ctx, err := session.Load(req.Context(), req.Header.Get("X-Session"))
if err != nil {
log.Println(err)
}
return ctx
}

View File

@ -11,6 +11,7 @@ import (
"net/http" "net/http"
"os" "os"
"path/filepath" "path/filepath"
"testing"
"time" "time"
"github.com/alexedwards/scs/v2" "github.com/alexedwards/scs/v2"
@ -26,7 +27,7 @@ var (
session *scs.SessionManager session *scs.SessionManager
) )
func getRoutes() http.Handler { func TestMain(m *testing.M) {
gob.Register(models.Reservation{}) gob.Register(models.Reservation{})
// change this to true when in production // change this to true when in production
app.InProduction = false app.InProduction = false
@ -57,6 +58,10 @@ func getRoutes() http.Handler {
render.NewRenderer(&app) render.NewRenderer(&app)
os.Exit(m.Run())
}
func getRoutes() http.Handler {
mux := chi.NewMux() mux := chi.NewMux()
mux.Use(middleware.Recoverer) mux.Use(middleware.Recoverer)