|
3 | 3 | from mindsdb_sql_parser.ast import *
|
4 | 4 | from mindsdb_sql_parser.lexer import MindsDBLexer
|
5 | 5 |
|
6 |
| -class TestCreateView: |
| 6 | +class TestViews: |
7 | 7 | def test_create_view_lexer(self):
|
8 | 8 | sql = "CREATE VIEW my_view FROM integration AS ( SELECT * FROM pred )"
|
9 | 9 | tokens = list(MindsDBLexer().tokenize(sql))
|
@@ -34,6 +34,27 @@ def test_create_view_nofrom(self):
|
34 | 34 | assert str(ast) == str(expected_ast)
|
35 | 35 | assert ast.to_tree() == expected_ast.to_tree()
|
36 | 36 |
|
| 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 | + |
37 | 58 | # def test_create_dataset_full(self):
|
38 | 59 | # sql = "CREATE DATASET my_view FROM integr AS ( SELECT * FROM pred )"
|
39 | 60 | # ast = parse_sql(sql)
|
|
0 commit comments