Tests for unicode identifiers (#2816)

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

diff --git a/tests/test_primer.py b/tests/test_primer.py
index 2aac2d09ff53a0f6124040f67b161fa061e75043..0a9d2aec495042229c7e93ea011d361ba7f48302 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


@@ -180,7 +181,7 @@ class PrimerLibTests(unittest.TestCase):
         stdout, stderr = loop.run_until_complete(
             lib._gen_check_output([lib.BLACK_BINARY, "--help"])
         )
         stdout, stderr = loop.run_until_complete(
             lib._gen_check_output([lib.BLACK_BINARY, "--help"])
         )

-        self.assertTrue("The uncompromising code formatter" in stdout.decode("utf8"))
+        self.assertIn("The uncompromising code formatter", stdout.decode("utf8"))

         self.assertEqual(None, stderr)
 
         # TODO: Add a test to see failure works on Windows
         self.assertEqual(None, stderr)
 
         # TODO: Add a test to see failure works on Windows


@@ -267,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__":