Skip to content

Commit ed7bb2c

Browse files
added unit tests
1 parent 260c5f6 commit ed7bb2c

File tree

1 file changed

+22
-1
lines changed

1 file changed

+22
-1
lines changed

tests/test_mindsdb/test_create_view.py renamed to tests/test_mindsdb/test_views.py

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from mindsdb_sql_parser.ast import *
44
from mindsdb_sql_parser.lexer import MindsDBLexer
55

6-
class TestCreateView:
6+
class TestViews:
77
def test_create_view_lexer(self):
88
sql = "CREATE VIEW my_view FROM integration AS ( SELECT * FROM pred )"
99
tokens = list(MindsDBLexer().tokenize(sql))
@@ -34,6 +34,27 @@ def test_create_view_nofrom(self):
3434
assert str(ast) == str(expected_ast)
3535
assert ast.to_tree() == expected_ast.to_tree()
3636

37+
def test_alter_view_full(self):
38+
sql = "ALTER VIEW my_view AS ( SELECT * FROM pred ) FROM integr"
39+
ast = parse_sql(sql)
40+
expected_ast = AlterView(
41+
name=Identifier('my_view'),
42+
from_table=Identifier('integr'),
43+
query_str="SELECT * FROM pred"
44+
)
45+
assert str(ast) == str(expected_ast)
46+
assert ast.to_tree() == expected_ast.to_tree()
47+
48+
def test_alter_view_nofrom(self):
49+
sql = "ALTER VIEW my_view AS ( SELECT * FROM pred )"
50+
ast = parse_sql(sql)
51+
expected_ast = AlterView(
52+
name=Identifier('my_view'),
53+
query_str="SELECT * FROM pred"
54+
)
55+
assert str(ast) == str(expected_ast)
56+
assert ast.to_tree() == expected_ast.to_tree()
57+
3758
# def test_create_dataset_full(self):
3859
# sql = "CREATE DATASET my_view FROM integr AS ( SELECT * FROM pred )"
3960
# ast = parse_sql(sql)

0 commit comments

Comments
 (0)