tests

Author: StpMax

tests/test_mindsdb/test_jobs.py

@@ -109,4 +109,26 @@ def test_drop_job(self):
             if_exists=True
         )
         assert str(ast) == str(expected_ast)
+        assert ast.to_tree() == expected_ast.to_tree()
+
+    def test_create_job_with_variable(self):
+        """Test that @ prefix is preserved in job query_str"""
+        query_str = "INSERT INTO tbl SELECT * FROM source WHERE date > @last_date and @@condition is True"
+        sql = f"""
+            CREATE JOB myjob (
+                {query_str}
+            )
+            EVERY hour
+        """
+        ast = parse_sql(sql)
+        
+        expected_ast = CreateJob(
+            name=Identifier('myjob'),
+            query_str=query_str,
+            repeat_str="hour"
+        )
+        
+        # Check that query_str preserves the @ sign
+        assert ast.query_str == query_str
+        assert str(ast) == str(expected_ast)
         assert ast.to_tree() == expected_ast.to_tree()

tests/test_mindsdb/test_variables.py

@@ -60,3 +60,35 @@ def test_mysql(self):
 
         assert str(ast).lower() == sql.lower()
         assert str(ast) == str(expected_ast)
+
+    def test_variables_in_select(self):
+        """Test multiple variables in single query"""
+        sql = "SELECT @var1, @var2 FROM tbl WHERE x = @var3 AND y = @@system_var"
+        ast = parse_sql(sql)
+
+        expected_ast = Select(
+                targets=[Variable("var1"), Variable("var2")],
+                from_table=Identifier("tbl"),
+                where=BinaryOperation(
+                    op="AND",
+                    args=[
+                        BinaryOperation(
+                            op="=",
+                            args=[Identifier("x"), Variable("var3")]
+                        ),
+                        BinaryOperation(
+                            op="=",
+                            args=[Identifier("y"), Variable("system_var", is_system_var=True)]
+                        )
+                    ]
+                )
+            )
+
+        # Verify the string representation preserves all @ signs
+        assert ast.to_tree() == expected_ast.to_tree()
+        result = str(ast)
+        assert result == str(expected_ast)
+        assert "@var1" in result
+        assert "@var2" in result
+        assert "@var3" in result
+        assert "@@system_var" in result

tests/test_mindsdb/test_views.py

@@ -55,6 +55,33 @@ def test_alter_view_nofrom(self):
         assert str(ast) == str(expected_ast)
         assert ast.to_tree() == expected_ast.to_tree()
 
+    def test_create_view_with_variables(self):
+        """Test multiple variables in view query"""
+        query_str = "SELECT @var1, @@sys_var FROM tbl WHERE x = @var2"
+        sql = f"CREATE VIEW v1 AS ({query_str})"
+        ast = parse_sql(sql)
+        
+        # Check that all variables preserve their prefixes
+        assert '@var1' in ast.query_str
+        assert '@@sys_var' in ast.query_str
+        assert '@var2' in ast.query_str
+        assert ast.query_str == query_str
+
+    def test_alter_view_with_variableы(self):
+        """Test that @ prefix is preserved in ALTER VIEW query_str"""
+        query_str = "SELECT @var1, @@sys_var FROM tbl WHERE x = @var2"
+        sql = f"ALTER VIEW myview AS ({query_str})"
+        ast = parse_sql(sql)
+        
+        expected_ast = AlterView(
+            name=Identifier('myview'),
+            query_str=query_str
+        )
+
+        assert ast.query_str == query_str
+        assert str(ast) == str(expected_ast)
+        assert ast.to_tree() == expected_ast.to_tree()
+
     # def test_create_dataset_full(self):
     #     sql = "CREATE DATASET my_view FROM integr AS ( SELECT * FROM pred )"
     #     ast = parse_sql(sql)