test: yaml file as text

andreoliwa · Jan 14, 2021 · d179976 · d179976
1 parent f9090c0
commit d179976
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/nitpick/plugins/text.py b/src/nitpick/plugins/text.py
@@ -73,6 +73,6 @@ def plugin_class() -> Type["NitpickPlugin"]:
 @hookimpl
 def handler(file_name: str, tags: Set[str]) -> Optional["NitpickPlugin"]:
     """Handle text files."""
-    if "plain-text" in tags:
+    if "text" in tags:
         return TextPlugin(file_name)
     return None
diff --git a/tests/test_text.py b/tests/test_text.py
@@ -67,3 +67,17 @@ def test_text_file_contains_line(request):
         qqq\x1b[0m
         """
     )
+
+
+def test_yaml_file_as_text(request):
+    ProjectMock(request).style(
+        """
+        [[".gitlab-ci.yml".contains]]
+        line = "    - mypy -p ims --junit-xml report-mypy.xml"
+        """
+    ).save_file(".gitlab-ci.yml", "def\nghi\nwww").flake8().assert_errors_contain(
+        """
+        NIP352 File .gitlab-ci.yml has missing lines:\x1b[32m
+            - mypy -p ims --junit-xml report-mypy.xml\x1b[0m
+        """
+    )