From 888b128a73b12c50cfd2e2d863438ce31c548359 Mon Sep 17 00:00:00 2001
From: Guy Bloom <guy.bloom@snowflake.com>
Date: Fri, 10 May 2024 13:34:00 -0400
Subject: [PATCH] read warehouse and role env vars in integration tests

---
 .gitignore                               | 1 +
 tests_integration/snowflake_connector.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/.gitignore b/.gitignore
index 825fe1b359..b123504497 100644
--- a/.gitignore
+++ b/.gitignore
@@ -15,6 +15,7 @@ dist/
 .coverage
 gen_docs/
 /venv/
+.env
 
 ^app.zip
 ^snowflake.yml
diff --git a/tests_integration/snowflake_connector.py b/tests_integration/snowflake_connector.py
index 4c87c059e5..b4da6a2e39 100644
--- a/tests_integration/snowflake_connector.py
+++ b/tests_integration/snowflake_connector.py
@@ -79,6 +79,8 @@ def snowflake_session():
         "user": _get_from_env("USER"),
         "password": _get_from_env("PASSWORD"),
         "host": _get_from_env("HOST", allow_none=True),
+        "warehouse": _get_from_env("WAREHOUSE", allow_none=True),
+        "role": _get_from_env("ROLE", allow_none=True),
     }
     config = {k: v for k, v in config.items() if v is not None}
     connection = connector.connect(**config)