allow single name tags in test configs

dbt-labs · Jan 27, 2020 · 8b722c7 · 8b722c7
1 parent 5c80d6a
commit 8b722c7
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/core/dbt/parser/schema_test_builders.py b/core/dbt/parser/schema_test_builders.py
@@ -249,6 +249,8 @@ def severity(self) -> str:
 
     def tags(self) -> List[str]:
         tags = self.modifiers.get('tags', [])
+        if isinstance(tags, str):
+            tags = [tags]
         if not isinstance(tags, list):
             raise_compiler_error(
                 f'got {tags} ({type(tags)}) for tags, expected a list of '

diff --git a/test/integration/008_schema_tests_test/models-v2/models/schema.yml b/test/integration/008_schema_tests_test/models-v2/models/schema.yml
@@ -32,7 +32,11 @@ models:
         - name: favorite_color
           description: "The user's favorite color"
           tests:
-            - accepted_values: { values: ['blue', 'green'], quote: true }
+            - accepted_values: {
+                values: ['blue', 'green'],
+                quote: true,
+                tags: table_copy_favorite_color  # tags can be a single string
+            }
           tags:
             - table_favorite_color
         - name: fav_number
@@ -41,7 +45,7 @@ models:
             - accepted_values:
                 values: [3.14159265]
                 quote: false
-                tags:
+                tags:  # tags can be a list of strings
                   - favorite_number_is_pi
 
 

diff --git a/test/integration/008_schema_tests_test/test_schema_v2_tests.py b/test/integration/008_schema_tests_test/test_schema_v2_tests.py
@@ -74,6 +74,11 @@ def test_postgres_schema_test_selection(self):
         self.assertEqual(len(test_results), 1)
         self.assertTestPassed(test_results[0])
 
+        test_results = self.run_dbt(['test', '--models', 'tag:table_copy_favorite_color'])
+        self.assertEqual(len(test_results), 1)
+        self.assertTestPassed(test_results[0])
+
+
     @use_profile('postgres')
     def test_postgres_schema_test_exclude_failures(self):
         results = self.run_dbt()