Foreign key tests

core/rootvalue.go

@@ -20,6 +20,7 @@
 	"fmt"
 	"io"
 	"sort"
+	"strconv"
 	"strings"
 
 	"github.com/dolthub/dolt/go/libraries/doltcore/doltdb"
@@ -118,6 +119,35 @@
 			buf.WriteString("\nSchema ")
 			buf.WriteString(schema.Name)
 		}
+
+		fkc, err := root.GetForeignKeyCollection(ctx)
+		if err == nil && fkc.Count() > 0 {
+			buf.WriteString("\nForeign Keys:")
+			fkc.Iter(func(fk doltdb.ForeignKey) (stop bool, err error) {
+				buf.WriteString("\n")
+				buf.WriteString(fk.Name)
+				buf.WriteString(": ")
+				buf.WriteString(fk.TableName.String())
+				buf.WriteString("(")
+				for i, tag := range fk.ReferencedTableColumns {
+					if i > 0 {
+						buf.WriteString(",")
+					}
+					buf.WriteString(strconv.Itoa(int(tag)))
+				}
+				buf.WriteString(") ON ")
+				buf.WriteString(fk.ReferencedTableName.String())
+				buf.WriteString("(")
+				for i, tag := range fk.ReferencedTableColumns {
+					if i > 0 {
+						buf.WriteString(",")
+					}
+					buf.WriteString(strconv.Itoa(int(tag)))
+				}
+				buf.WriteString(")\n")
+				return false, nil
+			})
+		}
 	}
 
 	return buf.String()

testing/go/foreign_keys_test.go

@@ -0,0 +1,232 @@
+// Copyright 2024 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package _go
+
+import (
+	"testing"
+
+	"github.com/dolthub/go-mysql-server/sql"
+)
+
+func TestForeignKeys(t *testing.T) {
+	RunScripts(
+		t,
+		[]ScriptTest{
+			{
+				Name: "simple foreign key",
+				SetUpScript: []string{
+					`CREATE TABLE parent (a INT PRIMARY KEY, b int)`,
+					`CREATE TABLE child (a INT PRIMARY KEY, b INT, FOREIGN KEY (b) REFERENCES parent(a))`,
+					`INSERT INTO parent VALUES (1, 1)`,
+				},
+				Assertions: []ScriptTestAssertion{
+					{
+						Query: "INSERT INTO child VALUES (1, 1)",
+					},
+					{
+						Query: "INSERT INTO child VALUES (2, 1)",
+					},
+					{
+						Query:       "INSERT INTO child VALUES (2, 2)",
+						ExpectedErr: "Foreign key violation",
+					},
+				},
+			},
+			{
+				Name: "foreign key with dolt_add, dolt_commit",
+				SetUpScript: []string{
+					"create table test (pk int, \"value\" int, primary key(pk));",
+					"CREATE TABLE test_info (id int, info varchar(255), test_pk int, primary key(id), foreign key (test_pk) references test(pk))",
+					"INSERT INTO test VALUES (0, 0), (1, 1), (2,2)",
+					"SELECT DOLT_ADD('.')",
+				},
+				Assertions: []ScriptTestAssertion{
+					{
+						Query: "SELECT * FROM dolt_status",
+						Expected: []sql.Row{
+							{"public.test", 1, "new table"},
+							{"public.test_info", 1, "new table"},
+						},
+					},
+					{
+						Query:            "SELECT dolt_commit('-am', 'new tables')",
+						SkipResultsCheck: true,
+					},
+					{
+						Query:    "SELECT * FROM dolt_status",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "SELECT * FROM dolt_schema_diff('HEAD', 'WORKING', 'test_info')",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "INSERT INTO test_info VALUES (2, 'two', 2)",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:       "INSERT INTO test_info VALUES (3, 'three', 3)",
+						ExpectedErr: "Foreign key violation",
+					},
+					{
+						Query: "SELECT * FROM test_info",
+						Expected: []sql.Row{
+							{2, "two", 2},
+						},
+					},
+				},
+			},
+			{
+				Name: "foreign key with explicit schema",
+				SetUpScript: []string{
+					"create table parent (pk int, \"value\" int, primary key(pk));",
+					"CREATE TABLE child (id int, info varchar(255), test_pk int, primary key(id), foreign key (test_pk) references public.parent(pk))",
+					"INSERT INTO parent VALUES (0, 0), (1, 1), (2,2)",
+					"SELECT DOLT_ADD('.')",
+				},
+				Assertions: []ScriptTestAssertion{
+					{
+						Query: "SELECT * FROM dolt_status",
+						Expected: []sql.Row{
+							{"public.child", 1, "new table"},
+							{"public.parent", 1, "new table"},
+						},
+					},
+					{
+						Query:            "SELECT dolt_commit('-am', 'new tables')",
+						SkipResultsCheck: true,
+					},
+					{
+						Query:    "SELECT * FROM dolt_status",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "SELECT * FROM dolt_schema_diff('HEAD', 'WORKING', 'child')",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "INSERT INTO child VALUES (2, 'two', 2)",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:       "INSERT INTO child VALUES (3, 'three', 3)",
+						ExpectedErr: "Foreign key violation",
+					},
+					{
+						Query: "SELECT * FROM child",
+						Expected: []sql.Row{
+							{2, "two", 2},
+						},
+					},
+				},
+			},
+			{
+				Name: "foreign key in another schema with search path",
+				Skip: true, // no GMS support for schemas in foreign key defns
+				SetUpScript: []string{
+					"create schema parent",
+					"create schema child",
+					"set search_path to 'parent, child'",
+					"create table parent.parent (pk int, \"value\" int, primary key(pk));",
+					"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id), foreign key (test_pk) references parent(pk))",
+					"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
+					"SELECT DOLT_ADD('.')",
+				},
+				Assertions: []ScriptTestAssertion{
+					{
+						Query: "SELECT * FROM dolt_status",
+						Expected: []sql.Row{
+							{"child.child", 1, "new table"},
+							{"parent.parent", 1, "new table"},
+						},
+					},
+					{
+						Query:            "SELECT dolt_commit('-am', 'new tables')",
+						SkipResultsCheck: true,
+					},
+					{
+						Query:    "SELECT * FROM dolt_status",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "SELECT * FROM dolt_schema_diff('HEAD', 'WORKING', 'child')",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "INSERT INTO child.child VALUES (2, 'two', 2)",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:       "INSERT INTO child.child VALUES (3, 'three', 3)",
+						ExpectedErr: "Foreign key violation",
+					},
+					{
+						Query: "SELECT * FROM child.child",
+						Expected: []sql.Row{
+							{2, "two", 2},
+						},
+					},
+				},
+			},
+			{
+				Name: "foreign key in another schema",
+				Skip: true, // no GMS support for schemas in foreign key defns
+				SetUpScript: []string{
+					"create schema parent",
+					"create schema child",
+					"create table parent.parent (pk int, \"value\" int, primary key(pk));",
+					"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id), foreign key (test_pk) references parent.parent(pk))",
+					"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
+					"SELECT DOLT_ADD('.')",
+				},
+				Assertions: []ScriptTestAssertion{
+					{
+						Query: "SELECT * FROM dolt_status",
+						Expected: []sql.Row{
+							{"child.child", 1, "new table"},
+							{"parent.parent", 1, "new table"},
+						},
+					},
+					{
+						Query:            "SELECT dolt_commit('-am', 'new tables')",
+						SkipResultsCheck: true,
+					},
+					{
+						Query:    "SELECT * FROM dolt_status",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "SELECT * FROM dolt_schema_diff('HEAD', 'WORKING', 'child')",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:    "INSERT INTO child.child VALUES (2, 'two', 2)",
+						Expected: []sql.Row{},
+					},
+					{
+						Query:       "INSERT INTO child.child VALUES (3, 'three', 3)",
+						ExpectedErr: "Foreign key violation",
+					},
+					{
+						Query: "SELECT * FROM child.child",
+						Expected: []sql.Row{
+							{2, "two", 2},
+						},
+					},
+				},
+			},
+		},
+	)
+}