From dbd1a9e4d6713c80885b0a1b68ea1dcb75556b86 Mon Sep 17 00:00:00 2001 From: Dmitry Tantsur Date: Thu, 25 Nov 2021 09:58:54 +0100 Subject: Do not assume sushy constants are strings - part 2 Change-Id: I42a314dd4d338b169fc5fb9be375ef004e153eac (cherry picked from commit 2ddd8ef2ca2e14287bd36c59f381138bc2a3b7b7) --- ironic/tests/unit/drivers/modules/drac/test_boot.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'ironic/tests/unit') diff --git a/ironic/tests/unit/drivers/modules/drac/test_boot.py b/ironic/tests/unit/drivers/modules/drac/test_boot.py index 66d01d8c3..c6ab480e9 100644 --- a/ironic/tests/unit/drivers/modules/drac/test_boot.py +++ b/ironic/tests/unit/drivers/modules/drac/test_boot.py @@ -77,7 +77,7 @@ class DracBootTestCase(test_utils.BaseDracTest): task, boot_devices.CDROM, persistent=True) mock_manager_oem.set_virtual_boot_device.assert_called_once_with( - 'cd', persistent=True, system=mock_system) + sushy.VIRTUAL_MEDIA_CD, persistent=True, system=mock_system) def test__set_boot_device_cd(self, mock_get_system): mock_system = mock_get_system.return_value @@ -90,7 +90,7 @@ class DracBootTestCase(test_utils.BaseDracTest): task.driver.boot._set_boot_device(task, boot_devices.CDROM) mock_manager_oem.set_virtual_boot_device.assert_called_once_with( - 'cd', persistent=False, system=mock_system) + sushy.VIRTUAL_MEDIA_CD, persistent=False, system=mock_system) def test__set_boot_device_floppy(self, mock_get_system): mock_system = mock_get_system.return_value @@ -103,7 +103,8 @@ class DracBootTestCase(test_utils.BaseDracTest): task.driver.boot._set_boot_device(task, boot_devices.FLOPPY) mock_manager_oem.set_virtual_boot_device.assert_called_once_with( - 'floppy', persistent=False, system=mock_system) + sushy.VIRTUAL_MEDIA_FLOPPY, persistent=False, + system=mock_system) def test__set_boot_device_disk(self, mock_get_system): mock_system = mock_get_system.return_value -- cgit v1.2.1