Materialize/misc/dbt-materialize/tests/adapter/test_custom_materializations.py

# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License in the LICENSE file at the
# root of this repository, or online at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import pytest
from dbt.tests.util import check_relations_equal, run_dbt
from fixtures import (
    actual_indexes,
    expected_indexes,
    test_materialized_view,
    test_materialized_view_index,
    test_materialized_view_retain_history,
    test_relation_name_length,
    test_sink,
    test_source,
    test_source_index,
    test_source_table,
    test_source_table_index,
    test_subsources,
    test_table_index,
    test_view_index,
)


class TestCustomMaterializations:
    @pytest.fixture(scope="class")
    def project_config_update(self):
        return {"name": "custom_materializations"}

    @pytest.fixture(scope="class")
    def seeds(self):
        return {
            "expected_indexes.csv": expected_indexes,
        }

    @pytest.fixture(scope="class")
    def models(self):
        return {
            "actual_indexes.sql": actual_indexes,
            "test_materialized_view.sql": test_materialized_view,
            "test_materialized_view_retain_history.sql": test_materialized_view_retain_history,
            "test_materialized_view_index.sql": test_materialized_view_index,
            "test_relation_name_loooooooooooooooooonger_than_postgres_63_limit.sql": test_relation_name_length,
            "test_source.sql": test_source,
            "test_source_index.sql": test_source_index,
            "test_source_table.sql": test_source_table,
            "test_source_table_index.sql": test_source_table_index,
            "test_subsources.sql": test_subsources,
            "test_sink.sql": test_sink,
            "test_table_index.sql": test_table_index,
            "test_view_index.sql": test_view_index,
        }

    def test_custom_materializations(self, project):
        # seed seeds
        results = run_dbt(["seed"])
        # seed result length
        assert len(results) == 1
        # run models
        results = run_dbt(["run"])
        # run result length
        assert len(results) == 13
        # re-run models to ensure there are no lingering errors in recreating
        # the materializations
        results = run_dbt(["run"])
        # re-run result length
        assert len(results) == 13
        # relations_equal
        check_relations_equal(
            project.adapter,
            ["test_materialized_view_index", "test_table_index"],
        )

        check_relations_equal(project.adapter, ["actual_indexes", "expected_indexes"])

        # TODO(morsapaes): add test that ensures that the source/sink emit the
        # correct data once sinks land.
        # From benesch: Ideally we'd just ingest the sink back into Materialize
        # with the source and then use `relations_equal` with `test_materialized_view`.