showbridge-go/internal/processor/test/string-create_test.go

package processor_test

import (
	"testing"

	"github.com/jwetzell/showbridge-go/internal/common"
	"github.com/jwetzell/showbridge-go/internal/config"
	"github.com/jwetzell/showbridge-go/internal/processor"
	"github.com/jwetzell/showbridge-go/internal/test"
)

func TestStringCreateFromRegistry(t *testing.T) {
	registration, ok := processor.ProcessorRegistry["string.create"]
	if !ok {
		t.Fatalf("string.create processor not registered")
	}

	processorInstance, err := registration.New(config.ProcessorConfig{
		Type: "string.create",
		Params: map[string]any{
			"template": "{{.Payload}}",
		},
	})
	if err != nil {
		t.Fatalf("failed to create string.create processor: %s", err)
	}

	if processorInstance.Type() != "string.create" {
		t.Fatalf("string.create processor has wrong type: %s", processorInstance.Type())
	}

	payload := "hello"
	expected := "hello"

	got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(t.Context(), payload))
	if err != nil {
		t.Fatalf("string.create processing failed: %s", err)
	}

	if got.Payload != expected {
		t.Fatalf("string.create got %+v, expected %+v", got, expected)
	}
}

func TestGoodStringCreate(t *testing.T) {

	tests := []struct {
		name     string
		params   map[string]any
		payload  any
		expected string
	}{
		{
			name:     "string payload",
			params:   map[string]any{"template": "{{.Payload}}"},
			payload:  "hello",
			expected: "hello",
		},
		{
			name:     "number payload",
			params:   map[string]any{"template": "{{.Payload}}"},
			payload:  4,
			expected: "4",
		},
		{
			name:     "boolean payload",
			params:   map[string]any{"template": "{{.Payload}}"},
			payload:  true,
			expected: "true",
		},
		{
			name:     "struct payload - field",
			params:   map[string]any{"template": "{{.Payload.Data}}"},
			payload:  test.TestStruct{Data: "test"},
			expected: "test",
		},
		{
			name:     "struct payload - method",
			params:   map[string]any{"template": "{{.Payload.GetData}}"},
			payload:  test.TestStruct{Data: "test"},
			expected: "test",
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			registration, ok := processor.ProcessorRegistry["string.create"]
			if !ok {
				t.Fatalf("string.create processor not registered")
			}

			processorInstance, err := registration.New(config.ProcessorConfig{
				Type:   "string.create",
				Params: test.params,
			})

			if err != nil {
				t.Fatalf("string.create failed to create processor: %s", err)
			}

			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(t.Context(), test.payload))

			if err != nil {
				t.Fatalf("string.create processing failed: %s", err)
			}

			gotStrings, ok := got.Payload.(string)
			if !ok {
				t.Fatalf("string.create returned a %T payload: %+v", got, got)
			}
			if gotStrings != test.expected {
				t.Fatalf("string.create got %+v, expected %s", got, test.expected)
			}
		})
	}
}

func TestBadStringCreate(t *testing.T) {
	tests := []struct {
		name        string
		params      map[string]any
		payload     any
		errorString string
	}{
		{
			name:        "no template param",
			payload:     "hello",
			params:      map[string]any{},
			errorString: "string.create template error: not found",
		},
		{
			name:    "non string template",
			payload: "hello",
			params: map[string]any{
				"template": 1,
			},
			errorString: "string.create template error: not a string",
		},
		{
			name:    "invalid template",
			payload: "hello",
			params: map[string]any{
				"template": "{{.",
			},
			errorString: "template: template:1: illegal number syntax: \".\"",
		},
		{
			name:    "bad property in template",
			payload: "hello",
			params: map[string]any{
				"template": "{{.Invalid}}",
			},
			errorString: "template: template:1:2: executing \"template\" at <.Invalid>: can't evaluate field Invalid in type common.WrappedPayload",
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {

			registration, ok := processor.ProcessorRegistry["string.create"]
			if !ok {
				t.Fatalf("string.create processor not registered")
			}

			processorInstance, err := registration.New(config.ProcessorConfig{
				Type:   "string.create",
				Params: test.params,
			})

			if err != nil {
				if test.errorString != err.Error() {
					t.Fatalf("string.create got error '%s', expected '%s'", err.Error(), test.errorString)
				}
				return
			}

			got, err := processorInstance.Process(t.Context(), common.GetWrappedPayload(t.Context(), test.payload))

			if err == nil {
				t.Fatalf("string.create expected to fail but got payload: %+v", got)
			}

			if err.Error() != test.errorString {
				t.Fatalf("string.create got error '%s', expected '%s'", err.Error(), test.errorString)
			}
		})
	}
}