Update tests for mount_disk_image_and_copy_files

CCA-Public · Aug 5, 2023 · 764daf7 · 764daf7
1 parent b7fabbf
commit 764daf7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/disk_image_toolkit/tests/test_disk_image.py b/disk_image_toolkit/tests/test_disk_image.py
@@ -2,6 +2,7 @@
 import os
 import pytest
 import subprocess
+from unittest.mock import call
 
 from disk_image_toolkit.dfxml import objects
 
@@ -380,8 +381,11 @@ def test_mount_image_and_copy_files(mocker, raw_image, create_dfxml):
     else:
         assert convert_raw.call_count == 0
 
-    cmd = "sudo mount -t udf -o loop '{}' /".format(raw_image)
-    subprocess_call.assert_called_with(cmd, shell=True)
+    mount_cmds = [
+        call("sudo mount -t udf -o loop 'udf-example.dd' /mnt/diskid/", shell=True),
+        call("sudo umount /mnt/diskid/", shell=True),
+    ]
+    subprocess_call.assert_has_calls(mount_cmds)
 
     if create_dfxml:
         write_dfxml.call_count == 1