Fix integration test: explicitly set database and schema for streamlit deploy commands

Author: sfc-gh-svishnu

tests_integration/test_streamlit_old.py

@@ -14,6 +14,7 @@
 
 import uuid
 from pathlib import Path
+from typing import List
 
 import pytest
 
@@ -23,7 +24,6 @@
     rows_from_snowflake_session,
 )
 from tests_integration.testing_utils import assert_that_result_is_successful
-from typing import List
 
 
 @pytest.mark.integration
@@ -141,7 +141,17 @@ def _assert_file_names_on_stage(expected_files: List[str]) -> None:
     with project_directory(f"streamlit_v2") as project_root:
         # deploy streamlit with legacy flag to use ROOT_LOCATION stages
         result = runner.invoke_with_connection(
-            ["streamlit", "deploy", "my_streamlit", "--replace", "--legacy"]
+            [
+                "streamlit",
+                "deploy",
+                "my_streamlit",
+                "--replace",
+                "--legacy",
+                "--database",
+                test_database,
+                "--schema",
+                "public",
+            ]
         )
         assert result.exit_code == 0, result.output
         _assert_file_names_on_stage(["streamlit_app.py"])
@@ -162,7 +172,17 @@ def _assert_file_names_on_stage(expected_files: List[str]) -> None:
 
         # deploy streamlit again without prune - unexpected file should remain on stage
         result = runner.invoke_with_connection(
-            ["streamlit", "deploy", "my_streamlit", "--replace", "--legacy"]
+            [
+                "streamlit",
+                "deploy",
+                "my_streamlit",
+                "--replace",
+                "--legacy",
+                "--database",
+                test_database,
+                "--schema",
+                "public",
+            ]
         )
         assert result.exit_code == 0, result.output
         _assert_file_names_on_stage(
@@ -174,7 +194,18 @@ def _assert_file_names_on_stage(expected_files: List[str]) -> None:
 
         # deploy with --prune flag - unexpected file should be removed
         result = runner.invoke_with_connection(
-            ["streamlit", "deploy", "my_streamlit", "--replace", "--legacy", "--prune"]
+            [
+                "streamlit",
+                "deploy",
+                "my_streamlit",
+                "--replace",
+                "--legacy",
+                "--prune",
+                "--database",
+                test_database,
+                "--schema",
+                "public",
+            ]
         )
         assert result.exit_code == 0, result.output
         _assert_file_names_on_stage(["streamlit_app.py"])