56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"kra/pkg/module"
|
|
platformtask "kra/pkg/task"
|
|
)
|
|
|
|
func TestTaskRegistryRegistersStaticModuleMethods(t *testing.T) {
|
|
method := platformtask.Method{
|
|
Name: "test.static",
|
|
Run: func(context.Context, json.RawMessage) error { return nil },
|
|
}
|
|
catalog := module.Catalog{Definitions: []module.Definition{{Tasks: []platformtask.Method{method}}}}
|
|
registry := TaskRegistry(catalog)
|
|
if _, ok := registry.Lookup(method.Name); !ok {
|
|
t.Fatalf("method %q was not registered", method.Name)
|
|
}
|
|
}
|
|
|
|
type testRouteRegistrar struct{ called bool }
|
|
|
|
func (registrar *testRouteRegistrar) RegisterRoutes(*gin.RouterGroup, *gin.RouterGroup, *gin.Engine) {
|
|
registrar.called = true
|
|
}
|
|
|
|
type testTaskContributor struct{ name string }
|
|
|
|
func (contributor testTaskContributor) RegisterTasks(registry *platformtask.Registry) {
|
|
registry.Register(platformtask.Method{
|
|
Name: contributor.name,
|
|
Run: func(context.Context, json.RawMessage) error { return nil },
|
|
})
|
|
}
|
|
|
|
func TestRuntimeAppliesAllContributions(t *testing.T) {
|
|
registry := platformtask.NewRegistry()
|
|
route := &testRouteRegistrar{}
|
|
runtime := Build(Composition{
|
|
Routes: []module.RouteRegistrar{route},
|
|
Tasks: []platformtask.Contributor{testTaskContributor{name: "test.runtime"}},
|
|
}, registry)
|
|
|
|
if _, ok := registry.Lookup("test.runtime"); !ok {
|
|
t.Fatal("runtime task contributor was not applied")
|
|
}
|
|
runtime.RegisterRoutes(nil, nil, nil)
|
|
if !route.called {
|
|
t.Fatal("runtime route contributor was not called")
|
|
}
|
|
}
|