Add test cases of PR #47

Author: svengiegerich

tests/integration/test_multiple_statements.py

@@ -161,3 +161,44 @@ def get_table(
     engine: Engine, table_name: str, schema: Optional[str] = None
 ) -> sa.Table:
     return sa.Table(table_name, sa.MetaData(), autoload_with=engine, schema=schema)
+
+
+def test_stored_procedure_declaration(engine):
+    statement = """
+DROP DATABASE IF EXISTS stored_procedure_declaration
+CREATE DATABASE stored_procedure_declaration
+USE stored_procedure_declaration
+GO
+/****** Object:  Table [dbo].[table1]    Script Date: 2/23/2021 2:48:02 PM ******/
+CREATE PROCEDURE CREATEALLDATES
+    (
+        @StartDate AS DATE, @EndDate AS DATE
+    ) AS
+    DECLARE @Current AS DATE = DATEADD(DD, 0, @StartDate); DROP TABLE IF EXISTS ##alldates CREATE TABLE ##alldates (
+        dt DATE PRIMARY KEY
+    ) WHILE @Current <= @EndDate BEGIN
+    INSERT INTO ##alldates
+    VALUES (@Current);
+    SET @Current = DATEADD(DD, 1, @Current) -- add 1 to current day
+END
+GO
+IF OBJECT_ID ( N'dbo.get_db_sampling_factor' , N'FN' ) IS NOT NULL DROP FUNCTION get_db_sampling_factor ;
+GO
+"""
+    executes(statement, engine, None)
+
+
+def test_top_level_declaration(engine):
+    statement = """
+DROP DATABASE IF EXISTS top_level_declaration
+CREATE DATABASE top_level_declaration
+USE top_level_declaration
+GO
+DECLARE @Current AS DATE = '2022-01-01'
+GO
+SELECT @Current as a INTO dummy01
+GO
+SELECT @Current as b INTO dummy02
+GO
+"""
+    executes(statement, engine, None)