fixing the tests

Author: fshabashev

mindsdb_sql_parser/ast/mindsdb/knowledge_base.py

@@ -122,3 +122,7 @@ def to_tree(self, *args, level=0, **kwargs):
         ind = indent(level)
         out_str = f"{ind}CreateIndexBase(name={self.name.to_string()})"
         return out_str
+
+    def get_string(self, *args, **kwargs):
+        out_str = f'CREATE INDEX ON KNOWLEDGE_BASE {self.name.to_string()}'
+        return out_str

mindsdb_sql_parser/parser.py

@@ -94,7 +94,8 @@ class MindsDBParser(Parser):
        'update_skill',
        'create_agent',
        'drop_agent',
-       'update_agent'
+       'update_agent',
+       'create_index'
        )
     def query(self, p):
         return p[0]

tests/test_mindsdb/test_knowledgebase_index.py

@@ -1,12 +1,18 @@
 from mindsdb_sql_parser import parse_sql
-
+from mindsdb_sql_parser.ast.mindsdb.knowledge_base import CreateKnowledgeBaseIndexBase
+from mindsdb_sql_parser.ast import *
 
 class TestKB:
 
     def test_create_knowledge_base_index(self):
         # create without select
-        sql = """
-            CREATE INDEX ON KNOWLEDGE_BASE my_knowledge_base;
-        """
+
+        sql = """CREATE INDEX ON KNOWLEDGE_BASE my_index"""
         ast = parse_sql(sql)
         print(ast)
+        expected_ast = CreateKnowledgeBaseIndexBase(
+            name=Identifier('my_index'),
+        )
+        assert str(ast).lower() == sql.lower()
+        assert ast.to_tree() == expected_ast.to_tree()
+