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

Signed-off-by: Alfredo Deza <[email protected]>
lesserevil · Dec 9, 2016 · 2cb9bfe · 2cb9bfe
1 parent 9be64fb
commit 2cb9bfe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ceph-disk/tests/test_prepare.py b/src/ceph-disk/tests/test_prepare.py
@@ -154,7 +154,7 @@ def test_create_partition(self,
                        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 @@ def test_create_partition(self,
                        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):