udemy-go-web-1/internal/handlers/handlers_test.go
2024-07-02 13:16:27 +02:00

76 lines
2.0 KiB
Go

package handlers
import (
"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)
}
}
}
}