Include underlying error when AST safety check parsing fails (#2693)

[etc/vim.git] / tests / test_primer.py

diff --git a/tests/test_primer.py b/tests/test_primer.py
index 8d00d8353a7efeba3f994d86b9edf72a966d8d3f..9bb401574ca1c2bb296170f7b9dce84dca9ca4dd 100644 (file)
--- a/tests/test_primer.py
+++ b/tests/test_primer.py
@@ -9,6 +9,7 @@ from io import StringIO
 from os import getpid
 from pathlib import Path
 from platform import system
 from os import getpid
 from pathlib import Path
 from platform import system

+from pytest import LogCaptureFixture

 from subprocess import CalledProcessError
 from tempfile import TemporaryDirectory, gettempdir
 from typing import Any, Callable, Iterator, List, Tuple, TypeVar
 from subprocess import CalledProcessError
 from tempfile import TemporaryDirectory, gettempdir
 from typing import Any, Callable, Iterator, List, Tuple, TypeVar


@@ -20,6 +21,14 @@ from black_primer import cli, lib
 
 
 EXPECTED_ANALYSIS_OUTPUT = """\
 
 
 EXPECTED_ANALYSIS_OUTPUT = """\

+
+Failed projects:
+
+## black:
+ - Returned 69
+ - stdout:
+Black didn't work
+

 -- primer results 📊 --
 
 68 / 69 succeeded (98.55%) ✅
 -- primer results 📊 --
 
 68 / 69 succeeded (98.55%) ✅


@@ -28,12 +37,7 @@ EXPECTED_ANALYSIS_OUTPUT = """\
  - 0 projects skipped due to Python version
  - 0 skipped due to long checkout
 
  - 0 projects skipped due to Python version
  - 0 skipped due to long checkout
 

-Failed projects:
-
-## black:
- - Returned 69
- - stdout:
-Black didn't work
+Failed projects: black

 
 """
 FAKE_PROJECT_CONFIG = {
 
 """
 FAKE_PROJECT_CONFIG = {


@@ -264,18 +268,23 @@ class PrimerCLITests(unittest.TestCase):
         result = runner.invoke(cli.main, ["--help"])
         self.assertEqual(result.exit_code, 0)
 
         result = runner.invoke(cli.main, ["--help"])
         self.assertEqual(result.exit_code, 0)
 

-    def test_projects(self) -> None:
+
+def test_projects(caplog: LogCaptureFixture) -> None:
+    with event_loop():
+        runner = CliRunner()
+        result = runner.invoke(cli.main, ["--projects=STDIN,asdf"])
+        assert result.exit_code == 0
+        assert "1 / 1 succeeded" in result.output
+        assert "Projects not found: {'asdf'}" in caplog.text
+
+    caplog.clear()
+
+    with event_loop():

         runner = CliRunner()
         runner = CliRunner()

-        with event_loop():
-            result = runner.invoke(cli.main, ["--projects=tox,asdf"])
-            self.assertEqual(result.exit_code, 0)
-            assert "1 / 1 succeeded" in result.output
-
-        with event_loop():
-            runner = CliRunner()
-            result = runner.invoke(cli.main, ["--projects=tox,attrs"])
-            self.assertEqual(result.exit_code, 0)
-            assert "2 / 2 succeeded" in result.output
+        result = runner.invoke(cli.main, ["--projects=fdsa,STDIN"])
+        assert result.exit_code == 0
+        assert "1 / 1 succeeded" in result.output
+        assert "Projects not found: {'fdsa'}" in caplog.text

 
 
 if __name__ == "__main__":
 
 
 if __name__ == "__main__":