ceph-disk: update tests to use the new exit flag

Signed-off-by: Alfredo Deza <adeza@redhat.com>

diff --git a/src/ceph-disk/tests/test_prepare.py b/src/ceph-disk/tests/test_prepare.py
index 3f9ff8624992b7daf00b1dac9a4d19a84c00fe06..2e475832d8dc7fde0690d482209eb1e3549c16c5 100644 (file)
--- a/src/ceph-disk/tests/test_prepare.py
+++ b/src/ceph-disk/tests/test_prepare.py
@@ -154,7 +154,7 @@ class TestDevice(Base):
                        partition_number,
                        main.PTYPE['regular']['journal']['ready']),
                    '--mbrtogpt', '--', path]
-        m_command_check_call.assert_called_with(command)
+        m_command_check_call.assert_called_with(command, exit=True)
         m_update_partition.assert_called_with(path, 'created')
 
         actual_partition_number = device.create_partition(
@@ -167,7 +167,7 @@ class TestDevice(Base):
                        partition_number,
                        main.PTYPE['regular']['journal']['ready']),
                    '--mbrtogpt', '--', path]
-        m_command_check_call.assert_called_with(command)
+        m_command_check_call.assert_called_with(command, exit=True)
 
 
 class TestDevicePartition(Base):