test: add remove_files tests

tests/decryption/test_decrypt.py

@@ -11,7 +11,8 @@
     get_private_keys,
     decrypt_files,
     move_files,
-    remove_files
+    remove_files,
+    get_args
 )
 from tests.utils import patch_cli
 
@@ -179,3 +180,41 @@ def test_non_directory_path(self, tmp_path):
         non_dir_path.touch()
         with pytest.raises(ValueError):
             remove_files(non_dir_path)
+
+
+class TestGetArgs:
+    """Test get_args."""
+
+    def test_get_args(self):
+        """Test that the arguments are parsed correctly."""
+        with patch_cli(["decrypt.py", "--output-dir", "dir", "file.txt"]):
+            args = get_args()
+            assert args.output_dir == Path("dir")
+            assert args.file_paths == [Path("file.txt")]
+
+    def test_multiple_files(self):
+        """Test that multiple file paths can be passed."""
+        files = ["file.txt", "file2.txt", "file3.txt"]
+        with patch_cli(["decrypt.py"] + files):
+            args = get_args()
+            assert args.file_paths == [Path(file) for file in files]
+
+    def test_default_output_dir(self):
+        """Test that output_dir defaults to $TMPDIR when no directory is passed."""
+        with (patch_cli(["decrypt.py", "file.txt"]),
+              mock.patch.dict(os.environ, {"TMPDIR": "/mock/tmpdir"})):
+            args = get_args()
+            assert args.output_dir == Path("/mock/tmpdir")
+            assert args.file_paths == [Path("file.txt")]
+
+    def test_invalid_argument(self):
+        """Test that a system exit occurs when an invalid argument is passed."""
+        with (patch_cli(["decrypt.py", "--bad-argument", "dir", "file.txt"]),
+              pytest.raises(SystemExit)):
+            get_args()
+
+    def test_no_file_paths(self):
+        """Test that a system exit occurs when no file paths are passed."""
+        with (patch_cli(["decrypt.py", "--output-dir", "dir"]),
+              pytest.raises(SystemExit)):
+            get_args()