From aa64bbcbcd40cbac1b30bee08105679c975bbae3 Mon Sep 17 00:00:00 2001 From: Chris MacNaughton Date: Mon, 12 Sep 2022 14:24:29 -0400 Subject: [PATCH] Add mock to make unit tests work on Kinetic Change-Id: I0740c4ad27da89b9dbfbc49af4aaf3e3443af012 --- unit_tests/test_actions_remove_disk.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/unit_tests/test_actions_remove_disk.py b/unit_tests/test_actions_remove_disk.py index 369d3f1f..8482290d 100644 --- a/unit_tests/test_actions_remove_disk.py +++ b/unit_tests/test_actions_remove_disk.py @@ -69,6 +69,7 @@ class RemoveDiskActionTests(CharmTestCase): obj = remove_disk.ActionOSD(dev_map, osd_id='1') self.assertEqual(obj.device, '/dev/sdx1') + @mock.patch.object(remove_disk.charms_ceph.utils, 'disable_osd') @mock.patch.object(remove_disk, 'device_size') @mock.patch.object(remove_disk.charms_ceph.utils, 'stop_osd') @mock.patch.object(remove_disk, 'bcache_remove') @@ -76,7 +77,8 @@ class RemoveDiskActionTests(CharmTestCase): @mock.patch.object(remove_disk.subprocess, 'check_call') @mock.patch.object(remove_disk, 'get_bcache_names') def test_action_osd_remove(self, get_bcache_names, check_call, - call, bcache_remove, stop_osd, device_size): + call, bcache_remove, stop_osd, device_size, + disable_osd): call.return_value = 0 get_bcache_names.return_value = ('/dev/backing', '/dev/caching') device_size.side_effect = lambda x: 1 if x == '/dev/caching' else 0