wehub-resource-sync
240 lines
8.3 KiB
Go
240 lines
8.3 KiB
Go
// 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"
|
|
)
|
|
|
|
// TestGettingStartedGuide tests that the steps in the Doltgres Getting Started Guide work correctly.
|
|
// https://github.com/dolthub/doltgresql?tab=readme-ov-file#getting-started
|
|
func TestGettingStartedGuide(t *testing.T) {
|
|
RunScripts(t, []ScriptTest{
|
|
{
|
|
Name: "Doltgres Getting Started Guide",
|
|
SetUpScript: []string{
|
|
"create table employees (\n id int8,\n last_name text,\n first_name text,\n primary key(id));",
|
|
"create table teams (\n id int8,\n team_name text,\n primary key(id));",
|
|
"create table employees_teams(\n team_id int8,\n employee_id int8,\n primary key(team_id, employee_id),\n foreign key (team_id) references teams(id),\n foreign key (employee_id) references employees(id));",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
// Make a Dolt Commit
|
|
{
|
|
Query: "select * from dolt.status;",
|
|
Expected: []sql.Row{
|
|
{"public.employees", "f", "new table"},
|
|
{"public.employees_teams", "f", "new table"},
|
|
{"public.teams", "f", "new table"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select dolt_add('teams', 'employees', 'employees_teams');",
|
|
Expected: []sql.Row{{"{0}"}},
|
|
},
|
|
{
|
|
Query: "select * from dolt.status;",
|
|
Expected: []sql.Row{
|
|
{"public.employees", "t", "new table"},
|
|
{"public.employees_teams", "t", "new table"},
|
|
{"public.teams", "t", "new table"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select length(dolt_commit('-m', 'Created initial schema')::text);",
|
|
Expected: []sql.Row{{34}},
|
|
},
|
|
{
|
|
// TODO: employees_teams is still marked as modified even though we staged and committed it. The diff
|
|
// in the working set shows a schema change of adding the FK references. For now, we reset to
|
|
// remove this artifact and prevent it from showing up in later test assertions, but this can
|
|
// be removed once the issue is fixed.
|
|
// https://github.com/dolthub/doltgresql/issues/734
|
|
Query: "select dolt_reset('--hard');",
|
|
Expected: []sql.Row{{"{0}"}},
|
|
},
|
|
{
|
|
Query: "select * from dolt.status;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select count(*) from dolt.log;",
|
|
Expected: []sql.Row{{3}},
|
|
},
|
|
|
|
// Insert Some Data
|
|
{
|
|
Query: "insert into employees values (0, 'Sehn', 'Tim'), (1, 'Hendriks', 'Brian'), (2, 'Son','Aaron'), (3, 'Fitzgerald', 'Brian');",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "insert into teams values (0, 'Engineering'), (1, 'Sales');",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "insert into employees_teams(employee_id, team_id) values (0,0), (1,0), (2,0), (0,1), (3,1);",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: `select first_name, last_name, team_name from employees
|
|
join employees_teams on (employees.id=employees_teams.employee_id)
|
|
join teams on (teams.id=employees_teams.team_id)
|
|
where team_name='Engineering';`,
|
|
Expected: []sql.Row{
|
|
{"Tim", "Sehn", "Engineering"},
|
|
{"Brian", "Hendriks", "Engineering"},
|
|
{"Aaron", "Son", "Engineering"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select * from employees_teams where employee_id='0' and team_id='1';",
|
|
Expected: []sql.Row{{1, 0}},
|
|
},
|
|
|
|
// Examine the Diff
|
|
{
|
|
Query: "select * from dolt.status order by table_name;",
|
|
Expected: []sql.Row{
|
|
{"public.employees", "f", "modified"},
|
|
{"public.employees_teams", "f", "modified"},
|
|
{"public.teams", "f", "modified"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select to_last_name, to_first_name, to_id, to_commit, from_last_name, from_first_name, from_id, diff_type from dolt_diff_employees;",
|
|
Expected: []sql.Row{
|
|
{"Sehn", "Tim", 0, "WORKING", nil, nil, nil, "added"},
|
|
{"Hendriks", "Brian", 1, "WORKING", nil, nil, nil, "added"},
|
|
{"Son", "Aaron", 2, "WORKING", nil, nil, nil, "added"},
|
|
{"Fitzgerald", "Brian", 3, "WORKING", nil, nil, nil, "added"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select length(dolt_commit('-am', 'Populated tables with data')::text);",
|
|
Expected: []sql.Row{{34}},
|
|
},
|
|
{
|
|
Query: "select * from dolt.status order by table_name;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select message from dolt.log;",
|
|
Expected: []sql.Row{
|
|
{"Populated tables with data"},
|
|
{"Created initial schema"},
|
|
{"CREATE DATABASE"},
|
|
{"Initialize data repository"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select table_name, message, data_change, schema_change from dolt.diff order by date desc, table_name;",
|
|
Expected: []sql.Row{
|
|
{"public.employees", "Populated tables with data", "t", "f"},
|
|
{"public.employees_teams", "Populated tables with data", "t", "f"},
|
|
{"public.teams", "Populated tables with data", "t", "f"},
|
|
{"public.employees", "Created initial schema", "f", "t"},
|
|
{"public.employees_teams", "Created initial schema", "f", "t"},
|
|
{"public.teams", "Created initial schema", "f", "t"},
|
|
},
|
|
},
|
|
|
|
// Oh no! I made a mistake.
|
|
{
|
|
Query: "drop table employees_teams;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select count(*) from employees_teams;",
|
|
ExpectedErr: "table not found: employees_teams",
|
|
},
|
|
{
|
|
Query: "select dolt_reset('--hard');",
|
|
Expected: []sql.Row{{"{0}"}},
|
|
},
|
|
{
|
|
Query: "select count(*) from employees_teams;",
|
|
Expected: []sql.Row{{5}},
|
|
},
|
|
|
|
// Make changes on a branch
|
|
{
|
|
Query: "select dolt_checkout('-b','modifications');",
|
|
Expected: []sql.Row{{"{0,\"Switched to branch 'modifications'\"}"}},
|
|
},
|
|
{
|
|
Query: "update employees SET first_name='Timothy' where first_name='Tim';",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "insert INTO employees (id, first_name, last_name) values (4,'Daylon', 'Wilkins');",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "insert into employees_teams(team_id, employee_id) values (0,4);",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "delete from employees_teams where employee_id=0 and team_id=1;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select length(dolt_commit('-am', 'Modifications on a branch')::text);",
|
|
Expected: []sql.Row{{34}},
|
|
},
|
|
{
|
|
Query: "select dolt_checkout('main');",
|
|
Expected: []sql.Row{{"{0,\"Switched to branch 'main'\"}"}},
|
|
},
|
|
{
|
|
Query: "select name, latest_commit_message from dolt.branches;",
|
|
Expected: []sql.Row{
|
|
{"main", "Populated tables with data"},
|
|
{"modifications", "Modifications on a branch"},
|
|
},
|
|
},
|
|
{
|
|
Query: "select active_branch();",
|
|
Expected: []sql.Row{{"main"}},
|
|
},
|
|
{
|
|
Query: "select * from employees;",
|
|
Expected: []sql.Row{{0, "Sehn", "Tim"}, {1, "Hendriks", "Brian"}, {2, "Son", "Aaron"}, {3, "Fitzgerald", "Brian"}},
|
|
},
|
|
{
|
|
Query: "select * from employees as of 'modifications';",
|
|
Expected: []sql.Row{{0, "Sehn", "Timothy"}, {1, "Hendriks", "Brian"}, {2, "Son", "Aaron"}, {3, "Fitzgerald", "Brian"}, {4, "Wilkins", "Daylon"}},
|
|
},
|
|
{
|
|
Query: `select to_id, to_last_name, to_first_name, to_commit,
|
|
from_id, from_last_name, from_first_name, from_commit, diff_type
|
|
from dolt_diff('main', 'modifications', 'employees');`,
|
|
Expected: []sql.Row{
|
|
{0, "Sehn", "Timothy", "modifications", 0, "Sehn", "Tim", "main", "modified"},
|
|
{4, "Wilkins", "Daylon", "modifications", nil, nil, nil, "main", "added"},
|
|
},
|
|
},
|
|
|
|
// Make a schema change on another branch
|
|
// TODO: Most ALTER TABLE statements aren't supported yet
|
|
{
|
|
Query: "select dolt_checkout('-b', 'schema_changes');",
|
|
Expected: []sql.Row{{"{0,\"Switched to branch 'schema_changes'\"}"}},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
}
|