add tests for db and kv processors

internal/processor/test/db-query_test.go

@@ -0,0 +1,216 @@
+package processor_test
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/jwetzell/showbridge-go/internal/common"
+	"github.com/jwetzell/showbridge-go/internal/config"
+	"github.com/jwetzell/showbridge-go/internal/processor"
+	_ "modernc.org/sqlite"
+)
+
+func TestDbQueryFromRegistry(t *testing.T) {
+	registration, ok := processor.ProcessorRegistry["db.query"]
+	if !ok {
+		t.Fatalf("db.query processor not registered")
+	}
+
+	processorInstance, err := registration.New(config.ProcessorConfig{
+		Type: "db.query",
+		Params: map[string]any{
+			"module": "test",
+			"query":  "SELECT sqlite_version();",
+		},
+	})
+	if err != nil {
+		t.Fatalf("failed to create db.query processor: %s", err)
+	}
+
+	if processorInstance.Type() != "db.query" {
+		t.Fatalf("db.query processor has wrong type: %s", processorInstance.Type())
+	}
+
+	payload := "hello"
+	expected := map[string]any{"sqlite_version()": "3.51.3"}
+
+	got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), payload))
+	if err != nil {
+		t.Fatalf("db.query processing failed: %s", err)
+	}
+
+	if !reflect.DeepEqual(got.Payload, expected) {
+		t.Fatalf("db.query got %+v, expected %+v", got.Payload, expected)
+	}
+}
+
+func TestGoodDbQuery(t *testing.T) {
+
+	tests := []struct {
+		name     string
+		params   map[string]any
+		payload  any
+		expected any
+	}{
+		{
+			name: "basic query",
+			params: map[string]any{
+				"module": "test",
+				"query":  "select value from test where id = 1;",
+			},
+			payload:  "",
+			expected: map[string]any{"value": "test-1"},
+		},
+		{
+			name: "template query",
+			params: map[string]any{
+				"module": "test",
+				"query":  "select value from test where id = {{.Payload}};",
+			},
+			payload:  "1",
+			expected: map[string]any{"value": "test-1"},
+		},
+		{
+			name: "multiple rows",
+			params: map[string]any{
+				"module": "test",
+				"query":  "select * from test;",
+			},
+			payload: "",
+			expected: []map[string]any{
+				{"id": int64(1), "value": "test-1"},
+				{"id": int64(2), "value": "test-2"},
+			},
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			registration, ok := processor.ProcessorRegistry["db.query"]
+			if !ok {
+				t.Fatalf("db.query processor not registered")
+			}
+
+			processorInstance, err := registration.New(config.ProcessorConfig{
+				Type:   "db.query",
+				Params: test.params,
+			})
+
+			if err != nil {
+				t.Fatalf("db.query failed to create processor: %s", err)
+			}
+
+			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), test.payload))
+
+			if err != nil {
+				t.Fatalf("db.query processing failed: %s", err)
+			}
+
+			if !reflect.DeepEqual(got.Payload, test.expected) {
+				t.Fatalf("db.query got payload: %+v, expected %+v", got.Payload, test.expected)
+			}
+		})
+	}
+}
+
+func TestBadDbQuery(t *testing.T) {
+	tests := []struct {
+		name        string
+		params      map[string]any
+		payload     any
+		errorString string
+	}{
+		{
+			name:    "no module param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"query": "SELECT sqlite_version();",
+			},
+			errorString: "db.query module error: not found",
+		},
+		{
+			name:    "non string module",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": 1,
+				"query":  "SELECT sqlite_version();",
+			},
+			errorString: "db.query module error: not a string",
+		},
+		{
+			name:    "no query param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+			},
+			errorString: "db.query query error: not found",
+		},
+		{
+			name:    "non string query",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"query":  1,
+			},
+			errorString: "db.query query error: not a string",
+		},
+		{
+			name:    "query template syntax error",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"query":  "select * from {{",
+			},
+			errorString: "template: query:1: unclosed action",
+		},
+		{
+			name:    "query template error",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"query":  "select * from {{.Data}}",
+			},
+			errorString: "template: query:1:16: executing \"query\" at <.Data>: can't evaluate field Data in type common.WrappedPayload",
+		},
+		{
+			name:    "query error",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"query":  "select * from asdf;",
+			},
+			errorString: "db.query error executing query: SQL logic error: no such table: asdf (1)",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+
+			registration, ok := processor.ProcessorRegistry["db.query"]
+			if !ok {
+				t.Fatalf("db.query processor not registered")
+			}
+
+			processorInstance, err := registration.New(config.ProcessorConfig{
+				Type:   "db.query",
+				Params: test.params,
+			})
+
+			if err != nil {
+				if test.errorString != err.Error() {
+					t.Fatalf("db.query got error '%s', expected '%s'", err.Error(), test.errorString)
+				}
+				return
+			}
+
+			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), test.payload))
+
+			if err == nil {
+				t.Fatalf("db.query expected to fail but got payload: %+v", got)
+			}
+
+			if err.Error() != test.errorString {
+				t.Fatalf("db.query got error '%s', expected '%s'", err.Error(), test.errorString)
+			}
+		})
+	}
+}

internal/processor/test/kv-get_test.go

@@ -0,0 +1,178 @@
+package processor_test
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/jwetzell/showbridge-go/internal/common"
+	"github.com/jwetzell/showbridge-go/internal/config"
+	"github.com/jwetzell/showbridge-go/internal/processor"
+)
+
+func TestKvGetFromRegistry(t *testing.T) {
+	registration, ok := processor.ProcessorRegistry["kv.get"]
+	if !ok {
+		t.Fatalf("kv.get processor not registered")
+	}
+
+	processorInstance, err := registration.New(config.ProcessorConfig{
+		Type: "kv.get",
+		Params: map[string]any{
+			"module": "test",
+			"key":    "test",
+		},
+	})
+	if err != nil {
+		t.Fatalf("failed to create kv.get processor: %s", err)
+	}
+
+	if processorInstance.Type() != "kv.get" {
+		t.Fatalf("kv.get processor has wrong type: %s", processorInstance.Type())
+	}
+
+	payload := "hello"
+	expected := "test"
+
+	got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), payload))
+	if err != nil {
+		t.Fatalf("kv.get processing failed: %s", err)
+	}
+
+	if got.Payload != expected {
+		t.Fatalf("kv.get got %+v, expected %+v", got.Payload, expected)
+	}
+}
+
+func TestGoodKvGet(t *testing.T) {
+
+	tests := []struct {
+		name     string
+		params   map[string]any
+		payload  any
+		expected any
+	}{
+		{
+			name: "basic value",
+			params: map[string]any{
+				"module": "test",
+				"key":    "test",
+				"value":  "hello",
+			},
+			payload:  "hello",
+			expected: "test",
+		},
+		{
+			name: "template value",
+			params: map[string]any{
+				"module": "test",
+				"key":    "test",
+				"value":  "{{.Payload}}",
+			},
+			payload:  "hello",
+			expected: "test",
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			registration, ok := processor.ProcessorRegistry["kv.get"]
+			if !ok {
+				t.Fatalf("kv.get processor not registered")
+			}
+
+			processorInstance, err := registration.New(config.ProcessorConfig{
+				Type:   "kv.get",
+				Params: test.params,
+			})
+
+			if err != nil {
+				t.Fatalf("kv.get failed to create processor: %s", err)
+			}
+
+			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), test.payload))
+
+			if err != nil {
+				t.Fatalf("kv.get processing failed: %s", err)
+			}
+
+			if !reflect.DeepEqual(got.Payload, test.expected) {
+				t.Fatalf("kv.get got payload: %+v, expected %+v", got.Payload, test.expected)
+			}
+		})
+	}
+}
+
+func TestBadKvGet(t *testing.T) {
+	tests := []struct {
+		name        string
+		params      map[string]any
+		payload     any
+		errorString string
+	}{
+		{
+			name:    "no module param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"key": "test",
+			},
+			errorString: "kv.get module error: not found",
+		},
+		{
+			name:    "non string module",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": 1,
+				"key":    "test",
+			},
+			errorString: "kv.get module error: not a string",
+		},
+		{
+			name:    "no key param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+			},
+			errorString: "kv.get key error: not found",
+		},
+		{
+			name:    "non string key",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"key":    1,
+			},
+			errorString: "kv.get key error: not a string",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+
+			registration, ok := processor.ProcessorRegistry["kv.get"]
+			if !ok {
+				t.Fatalf("kv.get processor not registered")
+			}
+
+			processorInstance, err := registration.New(config.ProcessorConfig{
+				Type:   "kv.get",
+				Params: test.params,
+			})
+
+			if err != nil {
+				if test.errorString != err.Error() {
+					t.Fatalf("kv.get got error '%s', expected '%s'", err.Error(), test.errorString)
+				}
+				return
+			}
+
+			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), test.payload))
+
+			if err == nil {
+				t.Fatalf("kv.get expected to fail but got payload: %+v", got)
+			}
+
+			if err.Error() != test.errorString {
+				t.Fatalf("kv.get got error '%s', expected '%s'", err.Error(), test.errorString)
+			}
+		})
+	}
+}

internal/processor/test/kv-set_test.go

@@ -0,0 +1,192 @@
+package processor_test
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/jwetzell/showbridge-go/internal/common"
+	"github.com/jwetzell/showbridge-go/internal/config"
+	"github.com/jwetzell/showbridge-go/internal/processor"
+)
+
+func TestKvSetFromRegistry(t *testing.T) {
+	registration, ok := processor.ProcessorRegistry["kv.set"]
+	if !ok {
+		t.Fatalf("kv.set processor not registered")
+	}
+
+	processorInstance, err := registration.New(config.ProcessorConfig{
+		Type: "kv.set",
+		Params: map[string]any{
+			"module": "test",
+			"key":    "test",
+			"value":  "hello",
+		},
+	})
+	if err != nil {
+		t.Fatalf("failed to create kv.set processor: %s", err)
+	}
+
+	if processorInstance.Type() != "kv.set" {
+		t.Fatalf("kv.set processor has wrong type: %s", processorInstance.Type())
+	}
+
+	payload := ""
+	expected := ""
+
+	got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), payload))
+	if err != nil {
+		t.Fatalf("kv.set processing failed: %s", err)
+	}
+
+	if got.Payload != expected {
+		t.Fatalf("kv.set got %+v, expected %+v", got.Payload, expected)
+	}
+}
+
+func TestGoodKvSet(t *testing.T) {
+
+	tests := []struct {
+		name     string
+		params   map[string]any
+		payload  any
+		expected any
+	}{
+		{
+			name: "basic key/value",
+			params: map[string]any{
+				"module": "test",
+				"key":    "test",
+				"value":  "hello",
+			},
+			payload:  "",
+			expected: "",
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			registration, ok := processor.ProcessorRegistry["kv.set"]
+			if !ok {
+				t.Fatalf("kv.set processor not registered")
+			}
+
+			processorInstance, err := registration.New(config.ProcessorConfig{
+				Type:   "kv.set",
+				Params: test.params,
+			})
+
+			if err != nil {
+				t.Fatalf("kv.set failed to create processor: %s", err)
+			}
+
+			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), test.payload))
+
+			if err != nil {
+				t.Fatalf("kv.set processing failed: %s", err)
+			}
+
+			if !reflect.DeepEqual(got.Payload, test.expected) {
+				t.Fatalf("kv.set got payload: %+v, expected %+v", got.Payload, test.expected)
+			}
+		})
+	}
+}
+
+func TestBadKvSet(t *testing.T) {
+	tests := []struct {
+		name        string
+		params      map[string]any
+		payload     any
+		errorString string
+	}{
+		{
+			name:    "no module param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"key":   "test",
+				"value": "test",
+			},
+			errorString: "kv.set module error: not found",
+		},
+		{
+			name:    "non string module",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": 1,
+				"key":    "test",
+				"value":  "test",
+			},
+			errorString: "kv.set module error: not a string",
+		},
+		{
+			name:    "no key param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"value":  "test",
+			},
+			errorString: "kv.set key error: not found",
+		},
+		{
+			name:    "non string key",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"key":    1,
+				"value":  "test",
+			},
+			errorString: "kv.set key error: not a string",
+		},
+		{
+			name:    "no value param",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"key":    "test",
+			},
+			errorString: "kv.set value error: not found",
+		},
+		{
+			name:    "non string value",
+			payload: TestStruct{Data: "hello"},
+			params: map[string]any{
+				"module": "test",
+				"key":    "test",
+				"value":  1,
+			},
+			errorString: "kv.set value error: not a string",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+
+			registration, ok := processor.ProcessorRegistry["kv.set"]
+			if !ok {
+				t.Fatalf("kv.set processor not registered")
+			}
+
+			processorInstance, err := registration.New(config.ProcessorConfig{
+				Type:   "kv.set",
+				Params: test.params,
+			})
+
+			if err != nil {
+				if test.errorString != err.Error() {
+					t.Fatalf("kv.set got error '%s', expected '%s'", err.Error(), test.errorString)
+				}
+				return
+			}
+
+			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(GetTestContext(t.Context()), test.payload))
+
+			if err == nil {
+				t.Fatalf("kv.set expected to fail but got payload: %+v", got)
+			}
+
+			if err.Error() != test.errorString {
+				t.Fatalf("kv.set got error '%s', expected '%s'", err.Error(), test.errorString)
+			}
+		})
+	}
+}