package main import ( "bytes" "encoding/json" "fmt" "net/http/httptest" "net/url" "testing" "github.com/spf13/afero" ) func (hs *HTTPServer) TestRequest(path string, args map[string]string) *httptest.ResponseRecorder { vals := url.Values{} vals.Set("id", TestTeamID) for k, v := range args { vals.Set(k, v) } recorder := httptest.NewRecorder() request := httptest.NewRequest( "GET", fmt.Sprintf("%s?%s", path, vals.Encode()), bytes.NewReader([]byte{}), ) hs.ServeHTTP(recorder, request) return recorder } func TestHttpd(t *testing.T) { server := NewTestServer() hs := NewHTTPServer("/", server.MothServer) if r := hs.TestRequest("/", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } if r := hs.TestRequest("/index.html", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } if r := hs.TestRequest("/rolodex.html", nil); r.Result().StatusCode != 404 { t.Error(r.Result()) } if r := hs.TestRequest("/state", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"Config":{"Devel":false},"Enabled":true,"TeamNames":{},"PointsLog":[],"Puzzles":{}}` { t.Error("Unexpected state", r.Body.String()) } if r := hs.TestRequest("/register", map[string]string{"id": "bad team id", "name": "GoTeam"}); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"status":"fail","data":{"short":"not registered","description":"team ID not found in list of valid team IDs"}}` { t.Error("Register bad team ID failed") } if r := hs.TestRequest("/register", map[string]string{"name": "GoTeam"}); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"status":"success","data":{"short":"registered","description":"team ID registered"}}` { t.Error("Register failed") } if r := hs.TestRequest("/register", map[string]string{"name": "GoTeam"}); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"status":"success","data":{"short":"already registered","description":"team ID has already been registered"}}` { t.Error("Register failed", r.Body.String()) } server.refresh() if r := hs.TestRequest("/state", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"Config":{"Devel":false},"Enabled":true,"TeamNames":{"self":"GoTeam"},"PointsLog":[],"Puzzles":{"pategory":[1]}}` { t.Error("Unexpected state", r.Body.String()) } if r := hs.TestRequest("/content/pategory", nil); r.Result().StatusCode != 404 { t.Error(r.Result()) } if r := hs.TestRequest("/content/pategory/1/not-here", nil); r.Result().StatusCode != 404 { t.Error(r.Result()) } if r := hs.TestRequest("/content/pategory/2/moo.txt", nil); r.Result().StatusCode != 404 { t.Error(r.Result()) } if r := hs.TestRequest("/content/pategory/1/", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } if r := hs.TestRequest("/content/pategory/1/moo.txt", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `moo` { t.Error("Unexpected body", r.Body.String()) } if r := hs.TestRequest("/answer", map[string]string{"cat": "pategory", "points": "1", "answer": "moo"}); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"status":"fail","data":{"short":"not accepted","description":"incorrect answer"}}` { t.Error("Unexpected body", r.Body.String()) } if r := hs.TestRequest("/answer", map[string]string{"cat": "pategory", "points": "1", "answer": "answer123"}); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"status":"success","data":{"short":"accepted","description":"1 points awarded in pategory"}}` { t.Error("Unexpected body", r.Body.String()) } server.refresh() if r := hs.TestRequest("/content/pategory/2/puzzle.json", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } state := StateExport{} if r := hs.TestRequest("/state", nil); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if err := json.Unmarshal(r.Body.Bytes(), &state); err != nil { t.Error(err) } else if len(state.PointsLog) != 1 { t.Errorf("Points log wrong length. Wanted 1, got %v (length %d)", state.PointsLog, len(state.PointsLog)) } else if len(state.Puzzles["pategory"]) != 2 { t.Error("Didn't unlock next puzzle") } if r := hs.TestRequest("/answer", map[string]string{"cat": "pategory", "points": "1", "answer": "answer123"}); r.Result().StatusCode != 200 { t.Error(r.Result()) } else if r.Body.String() != `{"status":"fail","data":{"short":"not accepted","description":"points already awarded to this team in this category"}}` { t.Error("Unexpected body", r.Body.String()) } } func TestDevelMemHttpd(t *testing.T) { srv := NewTestServer() { hs := NewHTTPServer("/", srv.MothServer) if r := hs.TestRequest("/mothballer/pategory.md", nil); r.Result().StatusCode != 404 { t.Error("Should have gotten a 404 for mothballer in prod mode") } } { srv.Config.Devel = true hs := NewHTTPServer("/", srv.MothServer) if r := hs.TestRequest("/mothballer/pategory.md", nil); r.Result().StatusCode != 500 { t.Log(r.Body.String()) t.Log(r.Result()) t.Error("Should have given us an internal server error, since category is a mothball") } } } func TestDevelFsHttps(t *testing.T) { fs := afero.NewBasePathFs(afero.NewOsFs(), "testdata") transpilerProvider := NewTranspilerProvider(fs) srv := NewMothServer(Configuration{Devel: true}, NewTestTheme(), NewTestState(), transpilerProvider) hs := NewHTTPServer("/", srv) if r := hs.TestRequest("/mothballer/cat0.mb", nil); r.Result().StatusCode != 200 { t.Log(r.Body.String()) t.Log(r.Result()) t.Error("Didn't get a Mothball") } }