mirror of
https://github.com/jwetzell/showbridge-go.git
synced 2026-04-26 21:05:30 +00:00
add test module with kv and db capabilities to processor tests
This commit is contained in:
@@ -2,6 +2,7 @@ package processor_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"database/sql"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/jwetzell/showbridge-go/internal/common"
|
"github.com/jwetzell/showbridge-go/internal/common"
|
||||||
@@ -58,6 +59,63 @@ func (p *TestProcessor) Process(ctx context.Context, wrappedPayload common.Wrapp
|
|||||||
return wrappedPayload, nil
|
return wrappedPayload, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type TestModule struct {
|
||||||
|
kvData map[string]any
|
||||||
|
db *sql.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TestModule) Start(ctx context.Context) error {
|
||||||
|
<-ctx.Done()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TestModule) Database() *sql.DB {
|
||||||
|
if m.db == nil {
|
||||||
|
db, _ := sql.Open("sqlite", ":memory:")
|
||||||
|
|
||||||
|
db.Exec(`
|
||||||
|
CREATE TABLE test (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
value TEXT
|
||||||
|
);
|
||||||
|
INSERT INTO test (id, value) VALUES (1, 'test-1'), (2, 'test-2');
|
||||||
|
|
||||||
|
`)
|
||||||
|
m.db = db
|
||||||
|
}
|
||||||
|
return m.db
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TestModule) Stop() {}
|
||||||
|
|
||||||
|
func (m *TestModule) Type() string {
|
||||||
|
return "module.test"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TestModule) Id() string {
|
||||||
|
return "test"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TestModule) Get(key string) (any, error) {
|
||||||
|
return key, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TestModule) Set(key string, value any) error {
|
||||||
|
if m.kvData == nil {
|
||||||
|
m.kvData = make(map[string]any)
|
||||||
|
}
|
||||||
|
m.kvData[key] = value
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetTestContext(ctx context.Context) context.Context {
|
||||||
|
testModule := &TestModule{}
|
||||||
|
ctx = context.WithValue(ctx, common.ModulesContextKey, map[string]common.Module{
|
||||||
|
"test": testModule,
|
||||||
|
})
|
||||||
|
return ctx
|
||||||
|
}
|
||||||
|
|
||||||
func TestProcessorBadRegistrationNoType(t *testing.T) {
|
func TestProcessorBadRegistrationNoType(t *testing.T) {
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r == nil {
|
if r := recover(); r == nil {
|
||||||
|
|||||||
Reference in New Issue
Block a user