mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
6f8e94629f
* Add global lru for routes with keys being the appname + path * minor comment fixes * remove duplicate entires from THIRD_PARTY * Make sure that we lock and unlock on get, refresh and delete on the cache
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"net/http/httputil"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/iron-io/functions/api"
|
|
"github.com/iron-io/functions/api/datastore"
|
|
"github.com/iron-io/functions/api/models"
|
|
"github.com/iron-io/functions/api/mqs"
|
|
"github.com/iron-io/functions/api/runner"
|
|
"github.com/iron-io/functions/api/runner/task"
|
|
"github.com/iron-io/functions/api/server/internal/routecache"
|
|
)
|
|
|
|
type testSpecialHandler struct{}
|
|
|
|
func (h *testSpecialHandler) Handle(c HandlerContext) error {
|
|
c.Set(api.AppName, "test")
|
|
return nil
|
|
}
|
|
|
|
func TestSpecialHandlerSet(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
tasks := make(chan task.Request)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
rnr, cancelrnr := testRunner(t)
|
|
defer cancelrnr()
|
|
|
|
go runner.StartWorkers(ctx, rnr, tasks)
|
|
|
|
s := &Server{
|
|
Runner: rnr,
|
|
Router: gin.New(),
|
|
Datastore: &datastore.Mock{
|
|
Apps: []*models.App{
|
|
{Name: "test"},
|
|
},
|
|
Routes: []*models.Route{
|
|
{Path: "/test", Image: "iron/hello", AppName: "test"},
|
|
},
|
|
},
|
|
MQ: &mqs.Mock{},
|
|
tasks: tasks,
|
|
Enqueue: DefaultEnqueue,
|
|
hotroutes: routecache.New(2),
|
|
}
|
|
|
|
router := s.Router
|
|
router.Use(prepareMiddleware(ctx))
|
|
s.bindHandlers()
|
|
s.AddSpecialHandler(&testSpecialHandler{})
|
|
|
|
_, rec := routerRequest(t, router, "GET", "/test", nil)
|
|
if rec.Code != 200 {
|
|
dump, _ := httputil.DumpResponse(rec.Result(), true)
|
|
t.Fatalf("Test SpecialHandler: expected special handler to run functions successfully. Response:\n%s", dump)
|
|
}
|
|
}
|