package utils import ( "errors" "fmt" "net/http" "net/http/httptest" "testing" "github.com/gorilla/mux" ) func TestSupportedVersion(t *testing.T) { req, err := http.NewRequest("GET", fmt.Sprintf("/v%s/libpod/testing/versions", APIVersion[LibpodTree][CurrentAPIVersion]), nil) if err != nil { t.Fatal(err) } req = mux.SetURLVars(req, map[string]string{"version": APIVersion[LibpodTree][CurrentAPIVersion].String()}) rr := httptest.NewRecorder() handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { _, err := SupportedVersionWithDefaults(r) switch { case errors.Is(err, ErrVersionNotGiven): // for compat endpoints version optional w.WriteHeader(http.StatusInternalServerError) fmt.Fprint(w, err.Error()) case errors.Is(err, ErrVersionNotSupported): // version given but not supported w.WriteHeader(http.StatusBadRequest) fmt.Fprint(w, err.Error()) case err != nil: w.WriteHeader(http.StatusInternalServerError) fmt.Fprint(w, err.Error()) default: // all good w.WriteHeader(http.StatusOK) fmt.Fprint(w, "OK") } }) handler.ServeHTTP(rr, req) if status := rr.Code; status != http.StatusOK { t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK) } // Check the response body is what we expect. expected := `OK` if rr.Body.String() != expected { t.Errorf("handler returned unexpected body: got %q want %q", rr.Body.String(), expected) } } func TestUnsupportedVersion(t *testing.T) { version := "999.999.999" req, err := http.NewRequest("GET", fmt.Sprintf("/v%s/libpod/testing/versions", version), nil) if err != nil { t.Fatal(err) } req = mux.SetURLVars(req, map[string]string{"version": version}) rr := httptest.NewRecorder() handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { _, err := SupportedVersionWithDefaults(r) switch { case errors.Is(err, ErrVersionNotGiven): // for compat endpoints version optional w.WriteHeader(http.StatusInternalServerError) fmt.Fprint(w, err.Error()) case errors.Is(err, ErrVersionNotSupported): // version given but not supported w.WriteHeader(http.StatusBadRequest) fmt.Fprint(w, err.Error()) case err != nil: w.WriteHeader(http.StatusInternalServerError) fmt.Fprint(w, err.Error()) default: // all good w.WriteHeader(http.StatusOK) fmt.Fprint(w, "OK") } }) handler.ServeHTTP(rr, req) if status := rr.Code; status != http.StatusBadRequest { t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusBadRequest) } // Check the response body is what we expect. expected := ErrVersionNotSupported.Error() if rr.Body.String() != expected { t.Errorf("handler returned unexpected body: got %q want %q", rr.Body.String(), expected) } } func TestEqualVersion(t *testing.T) { version := "1.30.0" req, err := http.NewRequest("GET", fmt.Sprintf("/v%s/libpod/testing/versions", version), nil) if err != nil { t.Fatal(err) } req = mux.SetURLVars(req, map[string]string{"version": version}) rr := httptest.NewRecorder() handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { _, err := SupportedVersion(r, "=="+version) switch { case errors.Is(err, ErrVersionNotGiven): // for compat endpoints version optional w.WriteHeader(http.StatusInternalServerError) fmt.Fprint(w, err.Error()) case errors.Is(err, ErrVersionNotSupported): // version given but not supported w.WriteHeader(http.StatusBadRequest) fmt.Fprint(w, err.Error()) case err != nil: w.WriteHeader(http.StatusInternalServerError) fmt.Fprint(w, err.Error()) default: // all good w.WriteHeader(http.StatusOK) fmt.Fprint(w, "OK") } }) handler.ServeHTTP(rr, req) if status := rr.Code; status != http.StatusOK { t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK) } // Check the response body is what we expect. expected := http.StatusText(http.StatusOK) if rr.Body.String() != expected { t.Errorf("handler returned unexpected body: got %q want %q", rr.Body.String(), expected) } }