Skip to content

tests for cross-schema foreign keys #878

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 11 commits into from
Oct 25, 2024
Merged

tests for cross-schema foreign keys #878

Changes from 8 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
a600cad
Fixed test
zachmu Oct 22, 2024
277584f
More tests
zachmu Oct 22, 2024
0e72a21
Another test, failing
zachmu Oct 22, 2024
338c5db
Merge branch 'main' into zachmu/cross-schema-fks
zachmu Oct 22, 2024
8ba6912
more tests
zachmu Oct 22, 2024
6e780ab
More testing
zachmu Oct 22, 2024
d75e591
More tests
zachmu Oct 22, 2024
5b88f92
Formatting
zachmu Oct 22, 2024
faed871
Merge branch 'main' into zachmu/cross-schema-fks
zachmu Oct 25, 2024
508860e
New deps
zachmu Oct 25, 2024
32ab888
Remove local overrides
zachmu Oct 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
237 changes: 230 additions & 7 deletions testing/go/foreign_keys_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -134,13 +134,66 @@ func TestForeignKeys(t *testing.T) {
},
{
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",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
"set search_path to 'parent, child'",
"create table parent.parent (pk int, \"value\" int, primary key(pk));",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val 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 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"},
{"fake.parent", 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 VALUES (2, 'two', 2)",
Expected: []sql.Row{},
},
{
Query: "INSERT INTO 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 with search path, parent table not on search path",
SetUpScript: []string{
"create schema parent",
"create schema child",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
"set search_path to 'child, fake'",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val 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('.')",
},
Expand All @@ -149,6 +202,7 @@ func TestForeignKeys(t *testing.T) {
Query: "SELECT * FROM dolt_status",
Expected: []sql.Row{
{"child.child", 1, "new table"},
{"fake.parent", 1, "new table"},
{"parent.parent", 1, "new table"},
},
},
Expand All @@ -165,11 +219,11 @@ func TestForeignKeys(t *testing.T) {
Expected: []sql.Row{},
},
{
Query: "INSERT INTO child.child VALUES (2, 'two', 2)",
Query: "INSERT INTO child VALUES (2, 'two', 2)",
Expected: []sql.Row{},
},
{
Query: "INSERT INTO child.child VALUES (3, 'three', 3)",
Query: "INSERT INTO child VALUES (3, 'three', 3)",
ExpectedErr: "Foreign key violation",
},
{
Expand All @@ -181,12 +235,14 @@ func TestForeignKeys(t *testing.T) {
},
},
{
Name: "foreign key in another schema",
Skip: true, // no GMS support for schemas in foreign key defns
Name: "foreign key in another schema, no search path",
SetUpScript: []string{
"create schema parent",
"create schema child",
"create table parent.parent (pk int, \"value\" int, primary key(pk));",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val 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('.')",
Expand All @@ -196,6 +252,7 @@ func TestForeignKeys(t *testing.T) {
Query: "SELECT * FROM dolt_status",
Expected: []sql.Row{
{"child.child", 1, "new table"},
{"fake.parent", 1, "new table"},
{"parent.parent", 1, "new table"},
},
},
Expand Down Expand Up @@ -227,6 +284,172 @@ func TestForeignKeys(t *testing.T) {
},
},
},
{
Name: "add foreign key in another schema on search path",
SetUpScript: []string{
"create schema parent",
"create schema child",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
"set search_path to 'child, parent'",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val int, primary key(pk));`,
"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id))",
"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
"SELECT DOLT_COMMIT('-Am', 'new tables')",
},
Assertions: []ScriptTestAssertion{
{
Query: "INSERT INTO child.child VALUES (2, 'two', 2)",
Expected: []sql.Row{},
},
{
Query: "ALTER TABLE child ADD FOREIGN KEY (test_pk) REFERENCES parent(pk)",
SkipResultsCheck: true,
},
{
Query: "INSERT INTO child VALUES (3, 'three', 3)",
ExpectedErr: "Foreign key violation",
},
{
Query: "SELECT * FROM child",
Expected: []sql.Row{
{2, "two", 2},
},
},
},
},
{
Name: "add foreign key in another schema, parent table not on search path",
SetUpScript: []string{
"create schema parent",
"create schema child",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
"set search_path to 'child, fake'",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val int, primary key(pk));`,
"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id))",
"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
"SELECT DOLT_COMMIT('-Am', 'new tables')",
},
Assertions: []ScriptTestAssertion{
{
Query: "INSERT INTO child.child VALUES (2, 'two', 2)",
Expected: []sql.Row{},
},
{
Query: "ALTER TABLE child ADD FOREIGN KEY (test_pk) REFERENCES parent.parent(pk)",
SkipResultsCheck: true,
},
{
Query: "INSERT INTO child VALUES (3, 'three', 3)",
ExpectedErr: "Foreign key violation",
},
{
Query: "SELECT * FROM child",
Expected: []sql.Row{
{2, "two", 2},
},
},
},
},
{
Name: "add foreign key in another schema, no search path",
SetUpScript: []string{
"create schema parent",
"create schema child",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val int, primary key(pk));`,
"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id))",
"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
"SELECT DOLT_COMMIT('-Am', 'new tables')",
},
Assertions: []ScriptTestAssertion{
{
Query: "INSERT INTO child.child VALUES (2, 'two', 2)",
Expected: []sql.Row{},
},
{
Query: "ALTER TABLE child.child ADD FOREIGN KEY (test_pk) REFERENCES parent.parent(pk)",
SkipResultsCheck: true,
},
{
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: "drop foreign key in another schema, on search path",
SetUpScript: []string{
"create schema parent",
"create schema child",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
"set search_path to 'child, parent'",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val int, primary key(pk));`,
"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id))",
"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
"SELECT DOLT_COMMIT('-Am', 'new tables')",
"INSERT INTO child.child VALUES (2, 'two', 2)",
"ALTER TABLE child.child ADD FOREIGN KEY (test_pk) REFERENCES parent(pk)",
},
Assertions: []ScriptTestAssertion{
{
Query: "INSERT INTO child.child VALUES (3, 'three', 3)",
ExpectedErr: "Foreign key violation",
},
{
Query: "alter table child DROP constraint child_ibfk_1",
SkipResultsCheck: true,
},
{
Query: "INSERT INTO child.child VALUES (3, 'three', 3)",
Expected: []sql.Row{},
},
},
},
{
Name: "drop foreign key in another schema, no search path",
Skip: true, // not getting the explicit schema name passed to the node
SetUpScript: []string{
"create schema parent",
"create schema child",
"create schema fake",
"call dolt_commit('-Am', 'create schemas')",
`create table parent.parent (pk int, val int, primary key(pk));`,
`create table fake.parent (pk int, val int, primary key(pk));`,
"CREATE TABLE child.child (id int, info varchar(255), test_pk int, primary key(id))",
"INSERT INTO parent.parent VALUES (0, 0), (1, 1), (2,2)",
"SELECT DOLT_COMMIT('-Am', 'new tables')",
"INSERT INTO child.child VALUES (2, 'two', 2)",
"ALTER TABLE child.child ADD FOREIGN KEY (test_pk) REFERENCES parent.parent(pk)",
},
Assertions: []ScriptTestAssertion{
{
Query: "INSERT INTO child.child VALUES (3, 'three', 3)",
ExpectedErr: "Foreign key violation",
},
{
Query: "alter table child.child DROP constraint child_ibfk_1",
SkipResultsCheck: true,
},
{
Query: "INSERT INTO child.child VALUES (3, 'three', 3)",
Expected: []sql.Row{},
},
},
},
},
)
}
Loading