repogen/internal/spec/models_test.go

package spec_test

import (
	"testing"

	"git.kmsign.com/royalcat/repogen/internal/spec"
)

type OperationTestCase struct {
	Operation    spec.Operation
	ExpectedName string
}

func TestOperationName(t *testing.T) {
	testTable := []OperationTestCase{
		{
			Operation:    spec.InsertOperation{},
			ExpectedName: "Insert",
		},
		{
			Operation:    spec.FindOperation{},
			ExpectedName: "Find",
		},
		{
			Operation:    spec.UpdateOperation{},
			ExpectedName: "Update",
		},
		{
			Operation:    spec.DeleteOperation{},
			ExpectedName: "Delete",
		},
		{
			Operation:    spec.CountOperation{},
			ExpectedName: "Count",
		},
	}

	for _, testCase := range testTable {
		t.Run(testCase.ExpectedName, func(t *testing.T) {
			if testCase.Operation.Name() != testCase.ExpectedName {
				t.Errorf("Expected = %+v\nReceived = %+v", testCase.ExpectedName, testCase.Operation.Name())
			}
		})
	}
}