From 2d855719dd768dcbed388e0650ae0b76ed154e9b Mon Sep 17 00:00:00 2001
From: R1kaB3rN <100738684+R1kaB3rN@users.noreply.github.com>
Date: Sun, 28 Apr 2024 00:40:28 -0700
Subject: [PATCH] umu_test: add tests for check_runtime

---
 umu/umu_test.py | 35 ++++++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/umu/umu_test.py b/umu/umu_test.py
index f0f05da5..a3be98de 100644
--- a/umu/umu_test.py
+++ b/umu/umu_test.py
@@ -14,6 +14,7 @@
 from pathlib import Path
 from shutil import rmtree, copytree, copy
 from pwd import getpwuid
+from subprocess import CompletedProcess
 
 
 class TestGameLauncher(unittest.TestCase):
@@ -111,8 +112,9 @@ def setUp(self):
         Path(self.test_user_share, "umu").touch()
 
         # Mock pressure vessel
-        Path(self.test_user_share, "pressure-vessel").mkdir()
+        Path(self.test_user_share, "pressure-vessel", "bin").mkdir(parents=True)
         Path(self.test_user_share, "pressure-vessel", "foo").touch()
+        Path(self.test_user_share, "pressure-vessel", "bin", "pv-verify").touch()
 
         # Mock the proton file in the dir
         self.test_proton_dir.joinpath("proton").touch(exist_ok=True)
@@ -155,6 +157,37 @@ def tearDown(self):
         if self.test_local_share.exists():
             rmtree(self.test_local_share.as_posix())
 
+    def test_check_runtime(self):
+        """Test check_runtime when pv-verify does not exist.
+
+        check_runtime calls pv-verify to verify the integrity of the runtime
+        archive's contents, and will only be called when restoring or setting
+        up the runtime
+
+        If the pv-verify binary does not exist, a warning should be logged and
+        the function should return
+        """
+        json_root = umu_util._get_json(self.test_user_share, "umu_version.json")
+        self.test_user_share.joinpath("pressure-vessel", "bin", "pv-verify").unlink()
+        result = umu_util.check_runtime(self.test_user_share, json_root)
+        self.assertEqual(result, 1, "Expected the exit code 1")
+
+    def test_check_runtime_fail(self):
+        """Test check_runtime when runtime validation fails."""
+        json_root = umu_util._get_json(self.test_user_share, "umu_version.json")
+        mock = CompletedProcess(["foo"], 1)
+        with patch.object(umu_util, "run", return_value=mock):
+            result = umu_util.check_runtime(self.test_user_share, json_root)
+            self.assertEqual(result, 1, "Expected the exit code 1")
+
+    def test_check_runtime_success(self):
+        """Test check_runtime when runtime validation succeeds."""
+        json_root = umu_util._get_json(self.test_user_share, "umu_version.json")
+        mock = CompletedProcess(["foo"], 0)
+        with patch.object(umu_util, "run", return_value=mock):
+            result = umu_util.check_runtime(self.test_user_share, json_root)
+            self.assertEqual(result, 0, "Expected the exit code 0")
+
     def test_move(self):
         """Test _move when copying a directory or a file.