func TestCreatesARequestWithTheCorrectHostAndUrl(t *testing.T) {
spec := gspec.New(t)
context := core.NewContext(gspec.Request().Url("/test.json").Req)
req := newRequest(context, &core.Config{Upstream: "s3.viki.com"})
spec.Expect(req.Host).ToEqual("s3.viki.com")
spec.Expect(req.URL.Path).ToEqual("/test.json")
}
func TestCreatesARequestWithTheCorrectRange(t *testing.T) {
spec := gspec.New(t)
context := core.NewContext(gspec.Request().Url("somefile.mp4").Req)
context.Chunk = 3
req := newRequest(context, &core.Config{RangedExtensions: map[string]bool{".mp4": true}})
spec.Expect(req.Header.Get("range")).ToEqual("bytes=6291456-8388607")
}
func TestIgnoresTheRangeForNonRangeTypes(t *testing.T) {
spec := gspec.New(t)
context := core.NewContext(gspec.Request().Url("somefile.mp4").Req)
context.Chunk = 3
req := newRequest(context, new(core.Config))
spec.Expect(req.Header.Get("range")).ToEqual("")
}
func TestExecutesAListAction(t *testing.T) {
f := func(context interface{}) Response { return Json(`{"spice":"mustflow"}`).Response }
req := gspec.Request().Url("/v4/sessions.json").Req
res := httptest.NewRecorder()
router := newRouter(Configure().Route(R("LIST", "v4", "sessions", f)))
router.ServeHTTP(res, req)
assertResponse(t, res, 200, `{"spice":"mustflow"}`)
}
func TestExecutesAnActionRegardlessOfCasing(t *testing.T) {
f := func(context interface{}) Response { return Json(`{"name":"duncan"}`).Response }
req := gspec.Request().Url("/v2/GHOLAS/123g.json").Req
res := httptest.NewRecorder()
router := newRouter(Configure().Route(R("GET", "v2", "gholas", f)))
router.ServeHTTP(res, req)
assertResponse(t, res, 200, `{"name":"duncan"}`)
}
func TestLogsStatisticsToRedis(t *testing.T) {
conn := pool.Get()
defer cleanup(conn)
spec := gspec.New(t)
Run(core.NewContext(gspec.Request().Url("/something/funny.txt").Req), nil, core.NoopMiddleware)
time.Sleep(time.Second * 1)
spec.Expect(redis.Int(conn.Do("zscore", "hits", "/something/funny.txt"))).ToEqual(1)
}
func TestHandlesANilResponse(t *testing.T) {
f := func(context interface{}) Response { return nil }
c := Configure().Route(R("GET", "v1", "worms", f))
req := gspec.Request().Url("/v1/worms/22w.json").Method("GET").Req
res := httptest.NewRecorder()
newRouter(c).ServeHTTP(res, req)
assertResponse(t, res, 500, `{"error":"internal server error","code":500}`)
}
func TestHandlesBodiesLargerThanAllowed(t *testing.T) {
f := func(context *TestContext) Response { return Json("").Response }
c := Configure().Route(R("GET", "v1", "worms", f).BodyFactory(testBodyFactory)).ContextFactory(testContextFactory).Dispatcher(testDispatcher).BodyPool(3, 1)
req := gspec.Request().Url("/v1/worms/22w.json").Method("GET").BodyString(`{"hello":"World"}`).Req
res := httptest.NewRecorder()
newRouter(c).ServeHTTP(res, req)
assertResponse(t, res, 413, `{"error":"body too large","code":413}`)
}
func TestExecutesAPutAction(t *testing.T) {
f := func(context interface{}) Response { return Json(`{"name":"shaihulud"}`).Response }
req := gspec.Request().Url("/v1/worms/22w.json").Method("PUT").Req
res := httptest.NewRecorder()
router := newRouter(Configure().Route(R("PUT", "v1", "worms", f)))
router.ServeHTTP(res, req)
assertResponse(t, res, 200, `{"name":"shaihulud"}`)
}
func TestNoIpPresent(t *testing.T) {
f := func(context *TestContext) Response {
gspec.New(t).Expect(context.Ip).ToEqual("127.0.0.1")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json").Method("GET").Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestLoadCountryFromIpWhenNoXGeoIpHeader(t *testing.T) {
f := func(context *TestContext) Response {
gspec.New(t).Expect(context.Country).ToEqual("cn")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Header("x-forwarded-for", "218.108.232.190").Url("/v1/worms/22w.json").Method("GET")
newRouter(c).ServeHTTP(httptest.NewRecorder(), req.Req)
}
func TestLoadIpFromXForwardedHeader(t *testing.T) {
f := func(context *TestContext) Response {
gspec.New(t).Expect(context.Ip).ToEqual("12.12.12.12")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Header("x-forwarded-for", "12.12.12.12,13.13.13.13").Url("/v1/worms/22w.json").Method("GET")
newRouter(c).ServeHTTP(httptest.NewRecorder(), req.Req)
}
func TestParsesQueryStirngWithEmptyPairAtTheStart(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
spec.Expect(context.Query["app"]).ToEqual("100004a")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json?&app=100004a").Method("GET").Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestHandlesMultipleQuestionMarksInQueryString(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
spec.Expect(context.Query["app"]).ToEqual("100005a")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json?app=100002a?app=100005a").Method("GET").Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestParsesAQueryStringWithAMissingValue2(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
spec.Expect(context.Query["b"]).ToEqual("")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json?b=").Method("GET").Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestStoresRawBody(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
spec.Expect(string(context.RawBody)).ToEqual(`{"hello":"World"}`)
return Json("").Response
}
c := Configure().LoadRawBody().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json").Method("GET").BodyString(`{"hello":"World"}`).Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestDefaulsTheBodyValueWhenNoBodyIsPresent(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
input := context.Body.(*TestBody)
spec.Expect(input.Hello).ToEqual("")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f).BodyFactory(testBodyFactory)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json").Method("GET").Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestDeletesTheFilesDirectory(t *testing.T) {
spec := gspec.New(t)
dir := setupTempFolder(spec)
defer cleanupTempFolder()
res := httptest.NewRecorder()
Run(core.NewContext(gspec.Request().Method("purge").Req), res, nil)
spec.Expect(res.Code).ToEqual(200)
_, err := os.Stat(dir)
spec.Expect(err).ToNotBeNil()
}
func TestParsesQueryString(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
spec.Expect(context.Query["app"]).ToEqual("6003")
spec.Expect(context.Query["t"]).ToEqual("1 2")
return Json("").Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json?APP=6003&t=1%202&").Req
newRouter(c).ServeHTTP(httptest.NewRecorder(), req)
}
func TestOnlyAllowsWhitelistedIpsToPurge(t *testing.T) {
spec := gspec.New(t)
dir := setupTempFolder(spec)
defer cleanupTempFolder()
res := httptest.NewRecorder()
Run(core.NewContext(gspec.Request().Method("purge").RemoteAddr("23.33.24.55:4343").Req), res, nil)
spec.Expect(res.Code).ToEqual(401)
spec.Expect(res.Body.Len()).ToEqual(0)
_, err := os.Stat(dir)
spec.Expect(err).ToBeNil()
}
func TestCallsTheNextMiddleware(t *testing.T) {
spec := gspec.New(t)
context := core.NewContext(gspec.Request().Req)
res := httptest.NewRecorder()
var called bool
next := func(c *core.Context, r http.ResponseWriter) {
spec.Expect(c).ToEqual(context)
spec.Expect(r).ToEqual(res)
called = true
}
Run(context, res, next)
spec.Expect(called).ToEqual(true)
}
func TestCreatingAndDispatchingThroughCustomTypes(t *testing.T) {
spec := gspec.New(t)
f := func(context *TestContext) Response {
spec.Expect(context.Name).ToEqual("leto")
spec.Expect(context.Params.Version).ToEqual("v1")
spec.Expect(context.Params.Resource).ToEqual("worms")
spec.Expect(context.Params.Id).ToEqual("22w")
return Json(`{"name":"bigjohn"}`).Status(203).Response
}
c := Configure().Route(R("GET", "v1", "worms", f)).ContextFactory(testContextFactory).Dispatcher(testDispatcher)
req := gspec.Request().Url("/v1/worms/22w.json").Method("GET").Req
res := httptest.NewRecorder()
newRouter(c).ServeHTTP(res, req)
assertResponse(t, res, 203, `{"name":"bigjohn"}`)
}
func TestRoutesToANestedResource(t *testing.T) {
spec := gspec.New(t)
f := func(context interface{}) Response {
spec.Expect(context.(*BaseContext).Params.ParentResource).ToEqual("gholas")
spec.Expect(context.(*BaseContext).Params.ParentId).ToEqual("123g")
spec.Expect(context.(*BaseContext).Params.Resource).ToEqual("history")
spec.Expect(context.(*BaseContext).Params.Id).ToEqual("")
return Json(`{"name":"history"}`).Response
}
req := gspec.Request().Url("/v2/gholas/123g/history.json").Req
res := httptest.NewRecorder()
router := newRouter(Configure().Route(R("LIST", "v2", "gholas/history", f)))
router.ServeHTTP(res, req)
assertResponse(t, res, 200, `{"name":"history"}`)
}
func TestNotFoundOnShortUrls(t *testing.T) {
req := gspec.Request().Url("/x").Req
res := httptest.NewRecorder()
newRouter(Configure()).ServeHTTP(res, req)
assertResponse(t, res, 404, `{"error":"not found","code":404}`)
}
func TestSetsTheRightChuckWhenLargeRange(t *testing.T) {
spec := gspec.New(t)
c := NewContext(gspec.Request().Header("range", "bytes=3000000-9000000").Req)
spec.Expect(c.Chunk).ToEqual(1)
}
func TestTheContextsDirectories(t *testing.T) {
spec := gspec.New(t)
c := NewContext(gspec.Request().Url("/test.json").Req)
spec.Expect(c.Dir).ToEqual("/tmp/01/0196/0196f4b7a30827487e3272e9499749e9/")
spec.Expect(c.TempDir).ToEqual("/tmp/tmp/")
}
func TestTheContextsFiles(t *testing.T) {
spec := gspec.New(t)
c := NewContext(gspec.Request().Url("/over9000.json").Req)
spec.Expect(c.HeaderFile).ToEqual("/tmp/d2/d21f/d21fd0eba7a9f34038292d38c8ff4837/d21fd0eba7a9f34038292d38c8ff4837_0.hdr")
spec.Expect(c.DataFile).ToEqual("/tmp/d2/d21f/d21fd0eba7a9f34038292d38c8ff4837/d21fd0eba7a9f34038292d38c8ff4837_0.dat")
}
func TestNotFoundOnUnknownControllers(t *testing.T) {
req := gspec.Request().Url("/v4/cats.json").Req
res := httptest.NewRecorder()
newRouter(Configure()).ServeHTTP(res, req)
assertResponse(t, res, 404, `{"error":"not found","code":404}`)
}
func TestTheContextsKey(t *testing.T) {
spec := gspec.New(t)
c := NewContext(gspec.Request().Header("range", "bytes=30000000-90000000").Url("/test.json?page=1").Req)
spec.Expect(c.Key).ToEqual("0196f4b7a30827487e3272e9499749e9_14")
}
func TestDefaultsToTheFirstChunk(t *testing.T) {
spec := gspec.New(t)
c := NewContext(gspec.Request().Req)
spec.Expect(c.Chunk).ToEqual(0)
}
