func TestSpaceInvadersCache(t *testing.T) { t.Parallel() r := new(route.Router) r.HandleFunc("/spaceinvaders/:key", SpaceInvaders) var etag string test := tgTesting.GenerateHandlerFunc(t, SpaceInvaders) if recorder := test("/spaceinvaders/somekey", "GET", nil, r); recorder != nil { if recorder.Code != http.StatusOK { t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusOK) } etag = recorder.Header().Get("Etag") } // test caching if req, err := http.NewRequest("GET", "/spaceinvaders/somekey", nil); err != nil { t.Errorf("%v", err) } else { req.Header.Set("If-None-Match", etag) recorder := httptest.NewRecorder() r.ServeHTTP(recorder, req) if recorder.Code != http.StatusNotModified { t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusNotModified) } } }
func TestSquares(t *testing.T) { t.Parallel() r := new(route.Router) r.HandleFunc("/squares/:key", Square) test := tgTesting.GenerateHandlerFunc(t, Square) for _, p := range tgTesting.GoodParams { recorder := test("/squares/test", "GET", p, r) if recorder.Code != http.StatusOK { t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusOK) } } for _, p := range tgTesting.BadParams { recorder := test("/squares/test", "GET", p, r) if recorder.Code != http.StatusOK { t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusOK) } } // test caching: recorder := test("/squares/cache", "GET", nil, r) if recorder.Code != http.StatusOK { t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusOK) } if req, err := http.NewRequest("GET", "/squares/cache", nil); err != nil { t.Errorf("%v", err) } else { req.Header.Set("Content-Type", "application/json") req.Header.Set("If-None-Match", recorder.Header().Get("Etag")) recorder2 := httptest.NewRecorder() r.ServeHTTP(recorder2, req) if recorder2.Code != http.StatusNotModified { t.Errorf("returned %v. Expected %v.", recorder2.Code, http.StatusNotModified) } } }