Skip to content

Commit b2c3fd2

Browse files
added a unit test
1 parent 41faaa2 commit b2c3fd2

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/test_mindsdb/test_create_integration.py renamed to tests/test_mindsdb/test_databases.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from mindsdb_sql_parser.lexer import MindsDBLexer
77

88

9-
class TestCreateDatabase:
9+
class TestDatabases:
1010
def test_create_database_lexer(self):
1111
sql = "CREATE DATABASE IF NOT EXISTS db WITH ENGINE = 'mysql', PARAMETERS = {\"user\": \"admin\", \"password\": \"admin\"}"
1212
tokens = list(MindsDBLexer().tokenize(sql))
@@ -130,3 +130,11 @@ def test_create_database_using(self):
130130
assert str(ast).lower() == str(expected_ast).lower()
131131
assert ast.to_tree() == expected_ast.to_tree()
132132

133+
def test_update_database(self):
134+
sql = "UPDATE DATABASE db SET PARAMETERS = {'A': 1}"
135+
ast = parse_sql(sql)
136+
137+
expected_ast = UpdateDatabase(name=Identifier('db'), updated_params={'PARAMETERS': {'A': 1}})
138+
139+
assert str(ast).lower() == str(expected_ast).lower()
140+
assert ast.to_tree() == expected_ast.to_tree()

0 commit comments

Comments
 (0)