add constructor for test module type

This commit is contained in:
Joel Wetzell
2026-05-19 21:55:43 -05:00
parent be1e01cc3f
commit 5db23627d6
3 changed files with 12 additions and 5 deletions
+3 -3
View File
@@ -19,7 +19,7 @@ func TestModuleBadRegistrationNoType(t *testing.T) {
module.RegisterModule(module.ModuleRegistration{
Type: "",
New: func(config config.ModuleConfig) (common.Module, error) {
return &test.TestModule{}, nil
return test.NewTestModule("test"), nil
},
})
}
@@ -47,14 +47,14 @@ func TestModuleBadRegistrationExistingType(t *testing.T) {
module.RegisterModule(module.ModuleRegistration{
Type: "module.test",
New: func(config config.ModuleConfig) (common.Module, error) {
return &test.TestModule{}, nil
return test.NewTestModule("test"), nil
},
})
module.RegisterModule(module.ModuleRegistration{
Type: "module.test",
New: func(config config.ModuleConfig) (common.Module, error) {
return &test.TestModule{}, nil
return test.NewTestModule("test"), nil
},
})
}
@@ -98,7 +98,7 @@ func TestBadModuleOutput(t *testing.T) {
name: "no module param",
params: map[string]any{},
payload: "test",
modules: map[string]common.Module{"test": &test.TestModule{}},
modules: map[string]common.Module{"test": test.NewTestModule("test")},
errorString: "module.output module error: not found",
},
{
@@ -107,7 +107,7 @@ func TestBadModuleOutput(t *testing.T) {
"module": 123,
},
payload: "test",
modules: map[string]common.Module{"test": &test.TestModule{}},
modules: map[string]common.Module{"test": test.NewTestModule("test")},
errorString: "module.output module error: not a string",
},
{
+7
View File
@@ -8,7 +8,14 @@ import (
_ "modernc.org/sqlite"
)
func NewTestModule(id string) *TestModule {
return &TestModule{
id: id,
}
}
type TestModule struct {
id string
}
func (m *TestModule) Start(ctx context.Context, router common.RouteIO) error {