package handlers import ( "context" "go-udemy-web-1/internal/models" "log" "net/http" "net/http/httptest" "net/url" "testing" ) type postData struct { key string value string } var theTests = []struct { name string url string method string params []postData expectedStatusCode int }{ // {"home", "/", "GET", []postData{}, http.StatusOK}, // {"about", "/about", "GET", []postData{}, http.StatusOK}, // {"gq", "/generals-quarters", "GET", []postData{}, http.StatusOK}, // {"ms", "/majors-suite", "GET", []postData{}, http.StatusOK}, // {"sa", "/availability", "GET", []postData{}, http.StatusOK}, // {"contact", "/contact", "GET", []postData{}, http.StatusOK}, // {"ma", "/make-reservation", "GET", []postData{}, http.StatusOK}, // {"post-search-avail", "/availability", "POST", []postData{ // {key: "start", value: "2020-01-01"}, // {key: "end", value: "2020-01-02"}, // }, http.StatusOK}, // {"post-search-avail-json", "/availability-json", "POST", []postData{ // {key: "start", value: "2020-01-01"}, // {key: "end", value: "2020-01-02"}, // }, http.StatusOK}, // {"make-reservation", "/make-reservation", "POST", []postData{ // {key: "first_name", value: "John"}, // {key: "last_name", value: "Smith"}, // {key: "email", value: "me@here.com"}, // {key: "phone", value: "12345"}, // }, http.StatusOK}, } func TestHandlers(t *testing.T) { routes := getRoutes() ts := httptest.NewTLSServer(routes) defer ts.Close() for _, e := range theTests { if e.method == "GET" { resp, err := ts.Client().Get(ts.URL + e.url) if err != nil { t.Log(err) t.Fatal(err) } if resp.StatusCode != e.expectedStatusCode { t.Errorf("for %s, expected %d but got %d", e.name, e.expectedStatusCode, resp.StatusCode) } } else { values := url.Values{} for _, x := range e.params { values.Add(x.key, x.value) } resp, err := ts.Client().PostForm(ts.URL+e.url, values) if err != nil { t.Log(err) t.Fatal(err) } if resp.StatusCode != e.expectedStatusCode { t.Errorf("for %s, expected %d but got %d", e.name, e.expectedStatusCode, resp.StatusCode) } } } } 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 }