from test.integration.base import DBTIntegrationTest, use_profile


class TestRuntimeMaterialization(DBTIntegrationTest):

    def setUp(self):
        DBTIntegrationTest.setUp(self)
        self.run_dbt(['seed'])

    @property
    def project_config(self):
        return {
            'config-version': 2,
            'data-paths': ['data'],
            'seeds': {
                'quote_columns': False,
            }
        }

    @property
    def schema(self):
        return "runtime_materialization_017"

    @property
    def models(self):
        return "models"

    def run_dbt_full_refresh(self):
        return self.run_dbt(['run', '--full-refresh'])

    @use_profile('postgres')
    def test_postgres_full_refresh(self):
        # initial full-refresh should have no effect
        results = self.run_dbt_full_refresh()
        self.assertEqual(len(results), 3)

        self.assertTablesEqual("seed", "view")
        self.assertTablesEqual("seed", "incremental")
        self.assertTablesEqual("seed", "materialized")

        # adds one record to the incremental model. full-refresh should truncate then re-run
        self.run_sql_file("invalidate_incremental.sql")
        results = self.run_dbt_full_refresh()
        self.assertEqual(len(results), 3)
        self.assertTablesEqual("seed", "incremental")

        self.run_sql_file("update.sql")

        results = self.run_dbt_full_refresh()
        self.assertEqual(len(results), 3)

        self.assertTablesEqual("seed", "view")
        self.assertTablesEqual("seed", "incremental")
        self.assertTablesEqual("seed", "materialized")

    @use_profile('postgres')
    def test_postgres_delete__dbt_tmp_relation(self):
        # This creates a __dbt_tmp view - make sure it doesn't interfere with the dbt run
        self.run_sql_file("create_view__dbt_tmp.sql")
        results = self.run_dbt(['run', '--model', 'view'])
        self.assertEqual(len(results), 1)

        self.assertTableDoesNotExist('view__dbt_tmp')
        self.assertTablesEqual("seed", "view")


class TestRuntimeMaterializationWithConfig(TestRuntimeMaterialization):
    @property
    def project_config(self):
        result = super().project_config
        result.update({'models': {'full_refresh': True}})
        return result

    def run_dbt_full_refresh(self):
        return self.run_dbt(['run'])