Add test for --keep

Author: thatch

squatter/__main__.py

@@ -12,9 +12,7 @@ def cli() -> None:  # pragma: no cover
 
 
 @cli.command(help="Generate a package")
-@click.option(
-    "--keep", is_flag=True, default=False, help="Do not delete temp dir"
-)
+@click.option("--keep", is_flag=True, default=False, help="Do not delete temp dir")
 @click.option(
     "--upload", is_flag=True, default=False, help="Whether to invoke twine upload"
 )
@@ -24,7 +22,11 @@ def cli() -> None:  # pragma: no cover
 )
 @click.argument("package_name")
 def generate(
-    package_name: str, upload: bool, keep: bool, author: Optional[str], author_email: Optional[str]
+    package_name: str,
+    upload: bool,
+    keep: bool,
+    author: Optional[str],
+    author_email: Optional[str],
 ) -> None:
     # TODO flag for delete
     with tempfile.TemporaryDirectory(prefix=package_name, delete=not keep) as d:

squatter/tests/__init__.py

@@ -1,3 +1,5 @@
+import os.path
+import re
 import unittest
 from pathlib import Path
 from subprocess import check_call
@@ -87,3 +89,13 @@ def patched_check_call(cmd: List[str], **kwargs: Any) -> Any:
         self.assertEqual("", result.output)
         self.assertFalse(result.exception)
         self.assertEqual(1, uploads)
+
+    @patch("squatter.templates.check_output")
+    @patch("squatter.templates.check_call")
+    def test_cli_keep_flag(self, check_call_mock: Any, check_output_mock: Any) -> None:
+        runner = CliRunner()
+        result = runner.invoke(generate, ["--keep", "foo"])
+        m = re.match("Generating in (.+)", result.output)
+        assert os.path.exists(m.group(1))
+        assert os.path.exists(os.path.join(m.group(1), "setup.py"))
+        self.assertFalse(result.exception)