Add: Unit tests for uv mode

Author: n-thumann

tests/test_config.py

@@ -129,6 +129,26 @@ def test_get_mode_poetry_multiline(self):
 
         self.assertEqual(len(config.get_pre_commit_script_names()), 0)
 
+    def test_get_mode_uv_multiline(self):
+        config = AutohooksConfig.from_dict(
+            {"tool": {"autohooks": {"mode": "uv_multiline"}}}
+        )
+
+        self.assertTrue(config.has_autohooks_config())
+        self.assertEqual(config.get_mode(), Mode.UV_MULTILINE)
+
+        self.assertEqual(len(config.get_pre_commit_script_names()), 0)
+
+    def test_get_mode_uv(self):
+        config = AutohooksConfig.from_dict(
+            {"tool": {"autohooks": {"mode": "uv"}}}
+        )
+
+        self.assertTrue(config.has_autohooks_config())
+        self.assertEqual(config.get_mode(), Mode.UV)
+
+        self.assertEqual(len(config.get_pre_commit_script_names()), 0)
+
     def test_get_mode_pythonpath(self):
         config = AutohooksConfig.from_dict(
             {"tool": {"autohooks": {"mode": "pythonpath"}}}

tests/test_hooks.py

@@ -19,6 +19,8 @@
     POETRY_SHEBANG,
     PYTHON3_SHEBANG,
     TEMPLATE_VERSION,
+    UV_MULTILINE_SHEBANG,
+    UV_SHEBANG,
     PreCommitTemplate,
 )
 from autohooks.utils import exec_git
@@ -161,6 +163,12 @@ def test_poetry_mode(self):
 
         self.assertEqual(pre_commit_hook.read_mode(), Mode.POETRY)
 
+    def test_uv_mode(self):
+        path = FakeReadPath(f"#!{UV_SHEBANG}")
+        pre_commit_hook = PreCommitHook(path)
+
+        self.assertEqual(pre_commit_hook.read_mode(), Mode.UV)
+
     def test_poetry_mode_with_python3(self):
         path = FakeReadPath(f"#!{POETRY_SHEBANG}3")
         pre_commit_hook = PreCommitHook(path)
@@ -179,6 +187,12 @@ def test_poetry_multiline_mode(self):
 
         self.assertEqual(pre_commit_hook.read_mode(), Mode.POETRY_MULTILINE)
 
+    def test_uv_multiline_mode(self):
+        path = FakeReadPath(f"#!{UV_MULTILINE_SHEBANG}")
+        pre_commit_hook = PreCommitHook(path)
+
+        self.assertEqual(pre_commit_hook.read_mode(), Mode.UV_MULTILINE)
+
     def test_pythonpath_mode(self):
         path = FakeReadPath(f"#!{PYTHON3_SHEBANG}")
         pre_commit_hook = PreCommitHook(path)
@@ -211,6 +225,18 @@ def test_poetry_mode(self):
         text = args[0]
         self.assertRegex(text, f"^#!{POETRY_SHEBANG} *")
 
+    def test_uv_mode(self):
+        write_path = Mock()
+        pre_commit_hook = PreCommitHook(write_path)
+        pre_commit_hook.write(mode=Mode.UV)
+
+        write_path.chmod.assert_called_with(0o775)
+        self.assertTrue(write_path.write_text.called)
+
+        args, _kwargs = write_path.write_text.call_args
+        text = args[0]
+        self.assertRegex(text, f"^#!{UV_SHEBANG} *")
+
     def test_pythonpath_mode(self):
         write_path = Mock()
         pre_commit_hook = PreCommitHook(write_path)

tests/test_settings.py

@@ -22,6 +22,10 @@ def test_get_effective_mode(self):
         self.assertEqual(
             Mode.POETRY_MULTILINE.get_effective_mode(), Mode.POETRY_MULTILINE
         )
+        self.assertEqual(Mode.UV.get_effective_mode(), Mode.UV)
+        self.assertEqual(
+            Mode.UV_MULTILINE.get_effective_mode(), Mode.UV_MULTILINE
+        )
         self.assertEqual(Mode.UNDEFINED.get_effective_mode(), Mode.PYTHONPATH)
         self.assertEqual(Mode.UNKNOWN.get_effective_mode(), Mode.PYTHONPATH)
 
@@ -53,6 +57,18 @@ def test_get_poetry_multiline_mode_from_string(self):
             Mode.from_string("POETRY_MULTILINE"), Mode.POETRY_MULTILINE
         )
 
+    def test_get_uv_mode_from_string(self):
+        self.assertEqual(Mode.from_string("uv"), Mode.UV)
+        self.assertEqual(Mode.from_string("UV"), Mode.UV)
+
+    def test_get_uv_multiline_mode_from_string(self):
+        self.assertEqual(
+            Mode.from_string("uv_multiline"), Mode.UV_MULTILINE
+        )
+        self.assertEqual(
+            Mode.from_string("UV_MULTILINE"), Mode.UV_MULTILINE
+        )
+
     def test_get_invalid_mode_from_string(self):
         self.assertEqual(Mode.from_string("foo"), Mode.UNKNOWN)
         self.assertEqual(Mode.from_string(None), Mode.UNDEFINED)

tests/test_template.py

@@ -60,6 +60,14 @@ def test_should_render_mode_poetry(self):
             "/usr/bin/env -S poetry run python",
         )
 
+    def test_should_render_mode_uv(self):
+        path = FakeTemplatePath("$SHEBANG")
+        template = PreCommitTemplate(path)
+        self.assertEqual(
+            template.render(mode=Mode.UV),
+            "/usr/bin/env -S uv run python",
+        )
+
     def test_should_render_mode_pipenv_multiline(self):
         path = FakeTemplatePath("$SHEBANG")
         template = PreCommitTemplate(path)
@@ -88,6 +96,20 @@ def test_should_render_mode_poetry_multiline(self):
             ),
         )
 
+    def test_should_render_mode_uv_multiline(self):
+        path = FakeTemplatePath("$SHEBANG")
+        template = PreCommitTemplate(path)
+        self.assertEqual(
+            template.render(mode=Mode.UV_MULTILINE),
+            (
+                "/bin/sh\n"
+                "\"true\" ''':'\n"
+                'uv run python "$0" "$@"\n'
+                'exit "$?"\n'
+                "'''"
+            ),
+        )
+
     def test_should_render_mode_unknown(self):
         path = FakeTemplatePath("$SHEBANG")
         template = PreCommitTemplate(path)