Function TestBatch

backend/plugin/parser/tsql/batch/batch_test.go:58–94

(t *testing.T)

}

func TestBatch(t *testing.T) {
	type TestCase struct {
		// input is the input string.
		// The input string can be a single SQL statement or a batch of SQL statements.
		Input       string        `yaml:"input"`
		Description string        `yaml:"description"`
		Batches     []batchResult `yaml:"batches"`
	}

	a := require.New(t)

	filePath := "testdata/test_batch.yaml"
	f, err := os.Open(filePath)
	a.NoError(err)
	defer f.Close()

	bytes, err := io.ReadAll(f)
	a.NoError(err)

	var testCases []TestCase
	a.NoError(yaml.Unmarshal(bytes, &testCases))

	const (
		record = false
	)
	for i, tc := range testCases {
		batchResults := getBatchResults(a, tc.Input)
		if record {
			testCases[i].Batches = batchResults
		} else {
			a.Equalf(tc.Batches, batchResults, "Input: %s", tc.Input)
		}
	}
	if record {
		yamltest.Record(t, filePath, testCases)
	}
}

func getBatchResults(a *require.Assertions, input string) []batchResult {
	batch := NewBatcher(input)