Function TestApplyTiDBLineTracking

backend/plugin/parser/tidb/omni_test.go:328–355

TestApplyTiDBLineTracking pins the leading-newline arithmetic directly across the edges the integration tests don't all exercise: no leading whitespace (line 1), single vs multiple leading newlines, CRLF line endings, a non-zero baseLine, and non-newline leading whitespace (BYT-9381).

(t *testing.T)

// (line 1), single vs multiple leading newlines, CRLF line endings, a non-zero
// baseLine, and non-newline leading whitespace (BYT-9381).
func TestApplyTiDBLineTracking(t *testing.T) {
	nodes, err := ParseTiDB("SELECT 1", "", "")
	require.NoError(t, err)
	require.Len(t, nodes, 1)
	node := nodes[0]

	cases := []struct {
		name     string
		baseLine int
		text     string
		want     int
	}{
		{"no leading whitespace is line 1", 0, "SELECT 1", 1},
		{"single leading newline", 0, "\nSELECT 1", 2},
		{"three leading newlines (the BYT-9381 bug)", 0, "\n\n\nSELECT 1", 4},
		{"CRLF leading newlines count once each", 0, "\r\n\r\n\r\nSELECT 1", 4},
		{"non-zero baseLine plus two newlines", 5, "\n\nSELECT 1", 8},
		{"leading spaces and tabs, no newline", 0, "  \tSELECT 1", 1},
		{"mixed leading whitespace with one newline", 0, "  \n  SELECT 1", 2},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			got, err := applyTiDBLineTracking(node, tc.baseLine, tc.text)
			require.NoError(t, err)
			require.Equal(t, tc.want, got)
		})
	}
}

// TestCreateTableColumnLineTrackingBlankLines pins per-column line tracking for
// a CREATE TABLE preceded by blank lines (BYT-9381). The native parser may