attr -> use_profile

dbt-labs · Apr 18, 2019 · b00c824 · b00c824
1 parent 832f1db
commit b00c824
Showing 1 changed file with 15 additions and 16 deletions.
diff --git a/test/integration/004_simple_archive_test/test_simple_archive.py b/test/integration/004_simple_archive_test/test_simple_archive.py
@@ -1,5 +1,4 @@
-from nose.plugins.attrib import attr
-from test.integration.base import DBTIntegrationTest
+from test.integration.base import DBTIntegrationTest, use_profile
 import dbt.exceptions
 
 
@@ -58,7 +57,7 @@ def assert_case_tables_equal(self, actual, expected):
     def assert_expected(self):
         self.assert_case_tables_equal('archive_actual', 'archive_expected')
 
-    @attr(type='postgres')
+    @use_profile('postgres')
     def test__postgres__simple_archive(self):
         self.dbt_run_seed_archive()
 
@@ -72,7 +71,7 @@ def test__postgres__simple_archive(self):
 
         self.assert_expected()
 
-    @attr(type='snowflake')
+    @use_profile('snowflake')
     def test__snowflake__simple_archive(self):
         self.dbt_run_seed_archive()
 
@@ -86,7 +85,7 @@ def test__snowflake__simple_archive(self):
 
         self.assert_expected()
 
-    @attr(type='redshift')
+    @use_profile('redshift')
     def test__redshift__simple_archive(self):
         self.dbt_run_seed_archive()
 
@@ -100,7 +99,7 @@ def test__redshift__simple_archive(self):
 
         self.assert_expected()
 
-    @attr(type='presto')
+    @use_profile('presto')
     def test__presto__simple_archive_disabled(self):
         results = self.run_dbt(["seed"])
         self.assertEqual(len(results),  self.NUM_ARCHIVE_MODELS)
@@ -142,7 +141,7 @@ def project_config(self):
     def assert_expected(self):
         self.assertTablesEqual('archive_actual', 'archive_expected')
 
-    @attr(type='bigquery')
+    @use_profile('bigquery')
     def test__bigquery__simple_archive(self):
         self.use_default_project()
         self.use_profile('bigquery')
@@ -161,7 +160,7 @@ def test__bigquery__simple_archive(self):
         self.assert_expected()
 
 
-    @attr(type='bigquery')
+    @use_profile('bigquery')
     def test__bigquery__archive_with_new_field(self):
         self.use_default_project()
         self.use_profile('bigquery')
@@ -250,7 +249,7 @@ def project_config(self):
     def run_archive(self):
         return self.run_dbt(['archive'])
 
-    @attr(type='snowflake')
+    @use_profile('snowflake')
     def test__snowflake__cross_archive(self):
         self.run_sql_file("test/integration/004_simple_archive_test/seed.sql")
 
@@ -267,7 +266,7 @@ def test__snowflake__cross_archive(self):
 
         self.assertTablesEqual("ARCHIVE_EXPECTED", "ARCHIVE_ACTUAL", table_b_db=self.alternative_database)
 
-    @attr(type='bigquery')
+    @use_profile('bigquery')
     def test__bigquery__cross_archive(self):
         self.run_sql_file("test/integration/004_simple_archive_test/seed_bq.sql")
 
@@ -291,7 +290,7 @@ def project_config(self):
             "archive-paths": ['test/integration/004_simple_archive_test/test-archives-pg'],
         }
 
-    @attr(type='postgres')
+    @use_profile('postgres')
     def test__postgres_ref_archive(self):
         self.dbt_run_seed_archive()
         results = self.run_dbt(['run'])
@@ -315,7 +314,7 @@ def project_config(self):
                               'test/integration/004_simple_archive_test/test-archives-pg'],
         }
 
-    @attr(type='postgres')
+    @use_profile('postgres')
     def test__postgres__select_archives(self):
         self.run_sql_file('test/integration/004_simple_archive_test/seed.sql')
 
@@ -336,7 +335,7 @@ def test__postgres__select_archives(self):
         self.assertTablesEqual('archive_kelly', 'archive_kelly_expected')
         self.assertTablesEqual('archive_actual', 'archive_expected')
 
-    @attr(type='postgres')
+    @use_profile('postgres')
     def test__postgres_exclude_archives(self):
         self.run_sql_file('test/integration/004_simple_archive_test/seed.sql')
         results = self.run_dbt(['archive', '--exclude', 'archive_castillo'])
@@ -346,7 +345,7 @@ def test__postgres_exclude_archives(self):
         self.assertTablesEqual('archive_kelly', 'archive_kelly_expected')
         self.assertTablesEqual('archive_actual', 'archive_expected')
 
-    @attr(type='postgres')
+    @use_profile('postgres')
     def test__postgres_select_archives(self):
         self.run_sql_file('test/integration/004_simple_archive_test/seed.sql')
         results = self.run_dbt(['archive', '--models', 'archive_castillo'])
@@ -395,7 +394,7 @@ def project_config(self):
             "archive-paths": ['test/integration/004_simple_archive_test/test-archives-invalid'],
         }
 
-    @attr(type='postgres')
+    @use_profile('postgres')
     def test__postgres__invalid(self):
         with self.assertRaises(dbt.exceptions.CompilationException) as exc:
             self.run_dbt(['compile'], expect_pass=False)
@@ -451,7 +450,7 @@ def project_config(self):
             "archive-paths": ['test/integration/004_simple_archive_test/test-check-col-archives-bq'],
         }
 
-    @attr(type='bigquery')
+    @use_profile('bigquery')
     def test__bigquery__archive_with_new_field(self):
         self.use_default_project()
         self.use_profile('bigquery')