Function TestExtractTableRefs

backend/plugin/parser/mysql/query_span_test.go:130–188

(t *testing.T)

}

func TestExtractTableRefs(t *testing.T) {
	tests := []struct {
		statement string
		expected  []base.SchemaResource
	}{
		{
			statement: "SELECT * FROM t1 WHERE c1 = 1;",
			expected: []base.SchemaResource{
				{
					Database: "db",
					Table:    "t1",
				},
			},
		},
		{
			statement: "SELECT * FROM db1.t1 JOIN db2.t2 ON t1.c1 = t2.c1;",
			expected: []base.SchemaResource{
				{
					Database: "db1",
					Table:    "t1",
				},
				{
					Database: "db2",
					Table:    "t2",
				},
			},
		},
		{
			statement: "SELECT a > (select max(a) from t1) FROM t2;",
			expected: []base.SchemaResource{
				{
					Database: "db",
					Table:    "t1",
				},
				{
					Database: "db",
					Table:    "t2",
				},
			},
		},
	}

	for _, test := range tests {
		parseResult, err := ParseMySQLOmni(test.statement)
		require.NoError(t, err, "failed to parse statement: %s", test.statement)
		require.Len(t, parseResult.Items, 1, "expected one parse result for statement: %s", test.statement)

		sourceColumns := collectOmniAccessTables(parseResult.Items[0], "db", false)
		var resources []base.SchemaResource
		for resource := range sourceColumns {
			resources = append(resources, base.SchemaResource{
				Database: resource.Database,
				Schema:   resource.Schema,
				Table:    resource.Table,
			})
		}
		require.ElementsMatch(t, test.expected, resources, test.statement)
	}