added global test

Signed-off-by: Harshit Gangal <[email protected]>
vitessio · Dec 2, 2024 · a518157 · a518157
1 parent 7864e3a
commit a518157
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/.../planbuilder/testdata/vschemas/schema.sql → .../planbuilder/testdata/schemas/sschema.sql b/.../planbuilder/testdata/vschemas/schema.sql → .../planbuilder/testdata/schemas/sschema.sql
diff --git a/go/vt/vtgate/planbuilder/testdata/schemas/uschema.sql b/go/vt/vtgate/planbuilder/testdata/schemas/uschema.sql
diff --git a/go/vt/vtgate/vindexes/vschema_test.go b/go/vt/vtgate/vindexes/vschema_test.go
@@ -21,6 +21,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"os"
 	"reflect"
 	"strings"
 	"testing"
@@ -3551,6 +3552,20 @@ func TestFindTableWithSequences(t *testing.T) {
 	}
 }
 
+func TestGlobalTables(t *testing.T) {
+	input, err := os.ReadFile("../planbuilder/testdata/vschemas/schema.json")
+	require.NoError(t, err)
+
+	var vs vschemapb.SrvVSchema
+	err = json2.UnmarshalPB(input, &vs)
+	require.NoError(t, err)
+
+	got := BuildVSchema(&vs, sqlparser.NewTestParser())
+	tbl, err := got.findGlobalTable("user", false)
+	require.NoError(t, err)
+	assert.NotNil(t, tbl)
+}
+
 func vindexNames(vindexes []*ColumnVindex) (result []string) {
 	for _, vindex := range vindexes {
 		result = append(result, vindex.Name)