From 5797be9b149f75fbcf52fa9d62a02c271bee1777 Mon Sep 17 00:00:00 2001
From: Jacob Beck <jake@fishtownanalytics.com>
Date: Wed, 4 Dec 2019 14:45:16 -0700
Subject: [PATCH] make tests honor environment variables

---
 test.env.sample                                                 | 1 +
 .../050_warehouse_test/models/expected_warehouse.sql            | 2 +-
 .../050_warehouse_test/models/override_warehouse.sql            | 2 +-
 .../project-config-models/expected_warehouse.sql                | 2 +-
 test/integration/050_warehouse_test/test_warehouses.py          | 2 +-
 5 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/test.env.sample b/test.env.sample
index e078b4e0990..b0f0e95ce45 100644
--- a/test.env.sample
+++ b/test.env.sample
@@ -6,6 +6,7 @@ SNOWFLAKE_TEST_DATABASE=
 SNOWFLAKE_TEST_ALT_DATABASE=
 SNOWFLAKE_TEST_QUOTED_DATABASE=
 SNOWFLAKE_TEST_WAREHOUSE=
+SNOWFLAKE_TEST_ALT_WAREHOUSE=
 
 BIGQUERY_TYPE=
 BIGQUERY_PROJECT_ID=
diff --git a/test/integration/050_warehouse_test/models/expected_warehouse.sql b/test/integration/050_warehouse_test/models/expected_warehouse.sql
index e0799874332..a012035e6b3 100644
--- a/test/integration/050_warehouse_test/models/expected_warehouse.sql
+++ b/test/integration/050_warehouse_test/models/expected_warehouse.sql
@@ -1,2 +1,2 @@
 {{ config(materialized='table') }}
-select 'DBT_TEST_ALT' as warehouse
+select '{{ env_var("SNOWFLAKE_TEST_ALT_WAREHOUSE", "DBT_TEST_ALT") }}' as warehouse
diff --git a/test/integration/050_warehouse_test/models/override_warehouse.sql b/test/integration/050_warehouse_test/models/override_warehouse.sql
index c3391480fab..957718d2539 100644
--- a/test/integration/050_warehouse_test/models/override_warehouse.sql
+++ b/test/integration/050_warehouse_test/models/override_warehouse.sql
@@ -1,2 +1,2 @@
-{{ config(snowflake_warehouse='DBT_TEST_ALT', materialized='table') }}
+{{ config(snowflake_warehouse=env_var('SNOWFLAKE_TEST_ALT_WAREHOUSE', 'DBT_TEST_ALT'), materialized='table') }}
 select current_warehouse() as warehouse
diff --git a/test/integration/050_warehouse_test/project-config-models/expected_warehouse.sql b/test/integration/050_warehouse_test/project-config-models/expected_warehouse.sql
index e0799874332..a012035e6b3 100644
--- a/test/integration/050_warehouse_test/project-config-models/expected_warehouse.sql
+++ b/test/integration/050_warehouse_test/project-config-models/expected_warehouse.sql
@@ -1,2 +1,2 @@
 {{ config(materialized='table') }}
-select 'DBT_TEST_ALT' as warehouse
+select '{{ env_var("SNOWFLAKE_TEST_ALT_WAREHOUSE", "DBT_TEST_ALT") }}' as warehouse
diff --git a/test/integration/050_warehouse_test/test_warehouses.py b/test/integration/050_warehouse_test/test_warehouses.py
index 6a612976a3d..3ac84333ca9 100644
--- a/test/integration/050_warehouse_test/test_warehouses.py
+++ b/test/integration/050_warehouse_test/test_warehouses.py
@@ -39,7 +39,7 @@ def project_config(self):
             'source-paths': ['project-config-models'],
             'models': {
                 'test': {
-                    'snowflake_warehouse': 'DBT_TEST_ALT',
+                    'snowflake_warehouse': os.getenv('SNOWFLAKE_TEST_ALT_WAREHOUSE', 'DBT_TEST_ALT'),
                 },
             },
         }