add ut

pkg/frontend/session_test.go

@@ -384,6 +384,16 @@ func TestSession_TxnCompilerContext(t *testing.T) {
 		convey.So(object, convey.ShouldNotBeNil)
 		convey.So(tableRef, convey.ShouldNotBeNil)
 
+		ref := &plan.ObjectRef{
+			SchemaName: "schema",
+			PubInfo: &plan.PubInfo{
+				TenantId: 0,
+			},
+		}
+		object, tableRef = tcc.ResolveIndexTableByRef(ref, "indexTable", &plan2.Snapshot{TS: ts})
+		convey.So(object, convey.ShouldNotBeNil)
+		convey.So(tableRef, convey.ShouldNotBeNil)
+
 		pkd := tcc.GetPrimaryKeyDef("abc", "t1", &plan2.Snapshot{TS: ts})
 		convey.So(len(pkd), convey.ShouldBeZeroValue)
 

pkg/sql/plan/utils.go

@@ -2721,10 +2721,10 @@ func offsetToString(offset int) string {
 // return !strings.HasPrefix(tableDef.Name, catalog.IndexTableNamePrefix)
 // }
 
-// DbNameOfObjRef return subscription name of ObjectRef if exists,
+// DbNameOfObjRef return subscription name of ObjectRef if exists, to avoid the mismatching of account id and db name
 func DbNameOfObjRef(objRef *ObjectRef) string {
 	if objRef.SubscriptionName == "" {
 		return objRef.SchemaName
 	}
 	return objRef.SubscriptionName
-}
+}

pkg/sql/plan/utils_test.go

@@ -101,3 +101,39 @@ func TestHandleOptimizerHints(t *testing.T) {
 	handleOptimizerHints("skipDedup=1", builder)
 	require.Equal(t, 1, builder.optimizerHints.skipDedup)
 }
+
+func TestDbNameOfObjRef(t *testing.T) {
+	type args struct {
+		objRef *ObjectRef
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "case 1",
+			args: args{
+				objRef: &ObjectRef{
+					SchemaName: "db",
+				},
+			},
+			want: "db",
+		},
+		{
+			name: "case 2",
+			args: args{
+				objRef: &ObjectRef{
+					SchemaName:       "whatever",
+					SubscriptionName: "sub",
+				},
+			},
+			want: "sub",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			assert.Equalf(t, tt.want, DbNameOfObjRef(tt.args.objRef), "DbNameOfObjRef(%v)", tt.args.objRef)
+		})
+	}
+}