[CT-1970]: convert custom_schema tests to functional (#454)

* convert custom_schema tests to functional

* Replace bigserial with bigint

* nit newline

* import ordering

* Pr comments cleanup

* scope of setUp to function

* Simplify seeds addition

* Nits

Author: nssalian

tests/functional/adapter/custom_schema_tests/seeds.py

@@ -0,0 +1,13 @@
+seed_csv = """
+id,first_name,last_name,email,gender,ip_address
+1,Jack,Hunter,[email protected],Male,59.80.20.168
+2,Kathryn,Walker,[email protected],Female,194.121.179.35
+3,Gerald,Ryan,[email protected],Male,11.3.212.243
+""".lstrip()
+
+seed_agg_csv = """
+last_name,count
+Hunter,2
+Walker,2
+Ryan,2
+""".lstrip()

tests/functional/adapter/custom_schema_tests/test_custom_database.py

@@ -0,0 +1,117 @@
+import pytest
+import os
+from dbt.tests.util import (
+    check_relations_equal,
+    check_table_does_exist,
+    run_dbt
+)
+from tests.functional.adapter.custom_schema_tests.seeds import (
+    seed_agg_csv,
+    seed_csv
+)
+
+_VIEW_1_SQL = """
+select * from {{ ref('seed') }}
+""".lstrip()
+
+_VIEW_2_SQL = """
+{{ config(database='alt') }}
+select * from {{ ref('view_1') }}
+""".lstrip()
+
+_VIEW_3_SQL = """
+{{ config(database='alt', materialized='table') }}
+
+
+with v1 as (
+
+    select * from {{ ref('view_1') }}
+
+),
+
+v2 as (
+
+    select * from {{ ref('view_2') }}
+
+),
+
+combined as (
+
+    select last_name from v1
+    union all
+    select last_name from v2
+
+)
+
+select
+    last_name,
+    count(*) as count
+
+from combined
+group by 1
+""".lstrip()
+
+_CUSTOM_DB_SQL = """
+{% macro generate_database_name(database_name, node) %}
+    {% if database_name == 'alt' %}
+        {{ env_var('SNOWFLAKE_TEST_ALT_DATABASE') }}
+    {% elif database_name %}
+        {{ database_name }}
+    {% else %}
+        {{ target.database }}
+    {% endif %}
+{% endmacro %}
+""".lstrip()
+
+ALT_DATABASE = os.getenv("SNOWFLAKE_TEST_ALT_DATABASE")
+
+class TestOverrideDatabase:
+    @pytest.fixture(scope="class")
+    def macros(self):
+        return {
+            "custom_db.sql": _CUSTOM_DB_SQL,
+        }
+
+    @pytest.fixture(scope="class")
+    def seeds(self):
+        return {
+            "seed.csv" : seed_csv,
+            "agg.csv": seed_agg_csv
+        }
+
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "view_1.sql": _VIEW_1_SQL,
+            "view_2.sql": _VIEW_2_SQL,
+            "view_3.sql": _VIEW_3_SQL,
+        }
+
+    def test_snowflake_override_generate_db_name(self, project):
+        seed_results = run_dbt(["seed", "--full-refresh"])
+        assert len(seed_results) == 2
+
+        db_with_schema = f"{project.database}.{project.test_schema}"
+        alt_db_with_schema = f"{ALT_DATABASE}.{project.test_schema}"
+        seed_table = "SEED"
+        agg_table = "AGG"
+        view_1 = "VIEW_1"
+        view_2 = "VIEW_2"
+        view_3 = "VIEW_3"
+
+        check_table_does_exist(project.adapter, f"{db_with_schema}.{seed_table}")
+        check_table_does_exist(project.adapter, f"{db_with_schema}.{agg_table}")
+
+        results = run_dbt()
+        assert len(results) == 3
+
+        check_table_does_exist(project.adapter, f"{db_with_schema}.{view_1}")
+        check_table_does_exist(project.adapter, f"{alt_db_with_schema}.{view_2}")
+        check_table_does_exist(project.adapter, f"{alt_db_with_schema}.{view_3}")
+
+        # not overridden
+        check_relations_equal(project.adapter, [f"{db_with_schema}.{seed_table}", f"{db_with_schema}.{view_1}"])
+
+        # overridden
+        check_relations_equal(project.adapter, [f"{db_with_schema}.{seed_table}", f"{alt_db_with_schema}.{view_2}"])
+        check_relations_equal(project.adapter, [f"{db_with_schema}.{agg_table}", f"{alt_db_with_schema}.{view_3}"])

tests/functional/adapter/custom_schema_tests/test_custom_schema.py

@@ -0,0 +1,87 @@
+import pytest
+from dbt.tests.util import (
+    check_relations_equal,
+    run_dbt
+)
+from tests.functional.adapter.custom_schema_tests.seeds import (
+    seed_agg_csv,
+    seed_csv
+)
+
+_VIEW_1_SQL = """
+select * from {{ ref('seed') }}
+""".lstrip()
+
+_VIEW_2_SQL = """
+{{ config(schema='custom') }}
+
+select * from {{ ref('view_1') }}
+""".lstrip()
+
+_VIEW_3_SQL = """
+{{ config(schema='test', materialized='table') }}
+
+
+with v1 as (
+
+    select * from {{ ref('view_1') }}
+
+),
+
+v2 as (
+
+    select * from {{ ref('view_2') }}
+
+),
+
+combined as (
+
+    select last_name from v1
+    union all
+    select last_name from v2
+
+)
+
+select
+    last_name,
+    count(*) as count
+
+from combined
+group by 1
+""".lstrip()
+
+
+class TestCustomProjectSchemaWithPrefix:
+    @pytest.fixture(scope="class")
+    def seeds(self):
+        return {
+            "seed.csv": seed_csv,
+            "agg.csv": seed_agg_csv
+        }
+
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "view_1.sql": _VIEW_1_SQL,
+            "view_2.sql": _VIEW_2_SQL,
+            "view_3.sql": _VIEW_3_SQL
+        }
+
+    @pytest.fixture(scope="class")
+    def project_config_update(self):
+        return {
+            "models": {
+                "schema": "dbt_test"
+            }
+        }
+
+    def test__snowflake__custom_schema_with_prefix(self, project):
+        seed_results = run_dbt(["seed"])
+        assert len(seed_results) == 2
+        results = run_dbt()
+        assert len(results) == 3
+
+        db_with_schema = f"{project.database}.{project.test_schema}"
+        check_relations_equal(project.adapter, [f"{db_with_schema}.SEED", f"{db_with_schema}_DBT_TEST.VIEW_1"])
+        check_relations_equal(project.adapter, [f"{db_with_schema}.SEED", f"{db_with_schema}_CUSTOM.VIEW_2"])
+        check_relations_equal(project.adapter, [f"{db_with_schema}.AGG", f"{db_with_schema}_TEST.VIEW_3"])