wehub-resource-sync
324 lines
8.5 KiB
Go
324 lines
8.5 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"
|
|
)
|
|
|
|
func TestCreateViewStatements(t *testing.T) {
|
|
RunScripts(t, createViewStmts)
|
|
}
|
|
|
|
var createViewStmts = []ScriptTest{
|
|
{
|
|
Name: "basic create view statements",
|
|
SetUpScript: []string{
|
|
"create table t1 (pk int);",
|
|
"insert into t1 values (1), (2), (3), (1);",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "create view v as select * from t1 order by pk;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select * from v order by pk;",
|
|
Expected: []sql.Row{{1}, {1}, {2}, {3}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "views on different schemas",
|
|
SetUpScript: []string{
|
|
"CREATE SCHEMA testschema;",
|
|
"SET search_path TO testschema;",
|
|
"CREATE TABLE testing (pk INT primary key, v2 TEXT);",
|
|
"INSERT INTO testing VALUES (1,'a'), (2,'b'), (3,'c');",
|
|
"CREATE VIEW testview AS SELECT * FROM testing;",
|
|
"CREATE SCHEMA myschema;",
|
|
"SET search_path TO myschema;",
|
|
"CREATE TABLE mytable (pk INT primary key, v1 INT);",
|
|
"INSERT INTO mytable VALUES (1,4), (2,5), (3,6);",
|
|
"CREATE VIEW myview AS SELECT * FROM mytable;",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "SHOW search_path;",
|
|
Expected: []sql.Row{{"myschema"}},
|
|
},
|
|
{
|
|
Query: "select v1 from myview order by pk;",
|
|
Expected: []sql.Row{{4}, {5}, {6}},
|
|
},
|
|
{
|
|
Query: "select v2 from testview order by pk;",
|
|
ExpectedErr: "table not found: testview",
|
|
},
|
|
{
|
|
Query: "select v2 from testschema.testview order by pk;",
|
|
Expected: []sql.Row{{"a"}, {"b"}, {"c"}},
|
|
},
|
|
{
|
|
Query: "select name from dolt_schemas;",
|
|
Expected: []sql.Row{{"myview"}},
|
|
},
|
|
{
|
|
Query: "SET search_path = 'testschema';",
|
|
},
|
|
{
|
|
Query: "SHOW search_path;",
|
|
Expected: []sql.Row{{"testschema"}},
|
|
},
|
|
{
|
|
Query: "select * from myview order by pk; /* err */",
|
|
ExpectedErr: "table not found: myview",
|
|
},
|
|
{
|
|
Query: "select v1 from myschema.myview order by pk;",
|
|
Expected: []sql.Row{{4}, {5}, {6}},
|
|
},
|
|
{
|
|
Query: "select v2 from testview order by pk;",
|
|
Expected: []sql.Row{{"a"}, {"b"}, {"c"}},
|
|
},
|
|
{
|
|
Query: "select name from dolt_schemas;",
|
|
Expected: []sql.Row{{"testview"}},
|
|
},
|
|
{
|
|
Query: "SET search_path = testschema, myschema;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "SHOW search_path;",
|
|
Expected: []sql.Row{{"testschema, myschema"}},
|
|
},
|
|
{
|
|
Skip: true, // TODO: Should be able to resolve views from all schema in search_path
|
|
Query: "select v1 from myview order by pk;",
|
|
Expected: []sql.Row{{4}, {5}, {6}},
|
|
},
|
|
{
|
|
Query: "select v2 from testview order by pk;",
|
|
Expected: []sql.Row{{"a"}, {"b"}, {"c"}},
|
|
},
|
|
{
|
|
Skip: true, // TODO: Should be able to resolve views from all schema in search_path
|
|
Query: "select name from dolt_schemas;",
|
|
Expected: []sql.Row{{"testview"}, {"myview"}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "create view from view",
|
|
SetUpScript: []string{
|
|
"create table t1 (pk int);",
|
|
"insert into t1 values (1), (2), (3), (1);",
|
|
"create view v as select * from t1 where pk > 1;",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "create view v1 as select * from v order by pk;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select * from v1 order by pk;",
|
|
Expected: []sql.Row{{2}, {3}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "view with expression name",
|
|
SetUpScript: []string{
|
|
"create view v as select 2+2",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "SELECT * from v;",
|
|
Expected: []sql.Row{{4}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "view with column names",
|
|
SetUpScript: []string{
|
|
`CREATE TABLE xy (x int primary key, y int);`,
|
|
`insert into xy values (1, 4), (4, 9)`,
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "create view v_today(today) as select 2",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "CREATE VIEW xyv (u,v) AS SELECT * from xy",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "SELECT v from xyv;",
|
|
Expected: []sql.Row{{4}, {9}},
|
|
},
|
|
{
|
|
Query: "SELECT today from v_today;",
|
|
Expected: []sql.Row{{2}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Skip: true, // TODO: getting subquery alias not supported error
|
|
Name: "nested view",
|
|
SetUpScript: []string{
|
|
"create table t1 (pk int);",
|
|
"insert into t1 values (1), (2), (3), (4);",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "create view unionView as (select * from t1 order by pk desc limit 1) union all (select * from t1 order by pk limit 1)",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select * from unionView order by pk;",
|
|
Expected: []sql.Row{{1}, {4}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "cast (postgres-specific syntax)",
|
|
SetUpScript: []string{
|
|
"create table t1 (pk int);",
|
|
"insert into t1 values (1), (2), (3), (4);",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "CREATE VIEW v AS SELECT pk::INT2 FROM t1 ORDER BY pk;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select * from v order by pk;",
|
|
Expected: []sql.Row{{1}, {2}, {3}, {4}},
|
|
},
|
|
{
|
|
Query: "CREATE VIEW v_text AS SELECT pk::int2, (pk)::text AS pk_text FROM t1;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "select pk_text from v_text order by pk;",
|
|
Expected: []sql.Row{{"1"}, {"2"}, {"3"}, {"4"}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "not yet supported create view queries",
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "CREATE TEMPORARY VIEW v AS SELECT 1;",
|
|
Skip: true,
|
|
},
|
|
{
|
|
Query: "CREATE RECURSIVE VIEW v AS SELECT 1;",
|
|
Skip: true,
|
|
},
|
|
{
|
|
Query: "CREATE VIEW v AS SELECT 1 WITH LOCAL CHECK OPTION;",
|
|
Skip: true,
|
|
},
|
|
{
|
|
Query: "CREATE VIEW v WITH (check_option = 'local') AS SELECT 1;",
|
|
Skip: true,
|
|
},
|
|
{
|
|
Query: "CREATE VIEW v WITH (security_barrier = true) AS SELECT 1;",
|
|
Skip: true,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "create view with CTE",
|
|
SetUpScript: []string{
|
|
"CREATE TABLE public.t1 (id integer NOT NULL);",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: `create view public.v1 as with table1 as (select * from t1) select id from table1;`,
|
|
Expected: []sql.Row{},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "create view with custom type in its select statement",
|
|
SetUpScript: []string{
|
|
"CREATE TYPE e AS ENUM ('sched', 'busy', 'final', 'help');",
|
|
"CREATE TABLE t (id integer NOT NULL, t e);",
|
|
"INSERT INTO t VALUES (1, 'busy'), (2, 'final'), (3, 'busy'), (4, 'help');",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: `create view v as select * from t where (t = 'busy'::e);`,
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: `select * from v;`,
|
|
Expected: []sql.Row{{1, "busy"}, {3, "busy"}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "View names must be unique across all relation types",
|
|
SetUpScript: []string{
|
|
"CREATE TABLE tbl1 (pk int PRIMARY KEY, v1 int);",
|
|
"CREATE SEQUENCE seq1;",
|
|
"CREATE VIEW existing_view AS SELECT pk FROM tbl1;",
|
|
"CREATE INDEX idx1 ON tbl1 (v1);",
|
|
},
|
|
Assertions: []ScriptTestAssertion{
|
|
{
|
|
Query: "CREATE VIEW tbl1 AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `relation "tbl1" already exists`,
|
|
},
|
|
{
|
|
Query: "CREATE OR REPLACE VIEW tbl1 AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `"tbl1" is not a view`,
|
|
},
|
|
{
|
|
Query: "CREATE VIEW seq1 AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `relation "seq1" already exists`,
|
|
},
|
|
{
|
|
Query: "CREATE OR REPLACE VIEW seq1 AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `"seq1" is not a view`,
|
|
},
|
|
{
|
|
Query: "CREATE VIEW existing_view AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `relation "existing_view" already exists`,
|
|
},
|
|
{
|
|
Query: "CREATE OR REPLACE VIEW existing_view AS SELECT pk FROM tbl1;",
|
|
Expected: []sql.Row{},
|
|
},
|
|
{
|
|
Query: "CREATE VIEW idx1 AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `relation "idx1" already exists`,
|
|
},
|
|
{
|
|
Query: "CREATE OR REPLACE VIEW idx1 AS SELECT pk FROM tbl1;",
|
|
ExpectedErr: `"idx1" is not a view`,
|
|
},
|
|
},
|
|
},
|
|
}
|