func TestIndex(t *testing.T) { server := httptest.NewServer(router.NewRouter()) defer server.Close() res, err := http.Get(server.URL) if err != nil { t.Error(err) } defer res.Body.Close() body, err := ioutil.ReadAll(res.Body) if err != nil { t.Error(err) } if res.StatusCode != 200 { t.Error("invalid status code") } if !strings.Contains(string(body), "Hi there!") { t.Error("invalid body") } }
func TestLogin(t *testing.T) { server := httptest.NewServer(router.NewRouter()) defer server.Close() res, err := http.PostForm(server.URL+"/login", url.Values{}) if err != nil { t.Error(err) } defer res.Body.Close() body, err := ioutil.ReadAll(res.Body) if err != nil { t.Error(err) } if res.StatusCode != 200 { t.Error("invalid status code") } if !strings.Contains(string(body), "Hi there!") { t.Error("invalid body") } }
func main() { r := router.NewRouter() log.Fatal(http.ListenAndServe(":8080", r)) }