def test_set_virt_pxe_boot(test_input, expected_exception): # Arrange test_api = CobblerAPI() test_manager = CollectionManager(test_api) test_system = System(test_manager) # Act with expected_exception: result = utils.set_virt_pxe_boot(test_system, test_input) # Assert assert test_system.virt_pxe_boot == 0 or test_system.virt_pxe_boot == 1
def set_virt_pxe_boot(self, num): return utils.set_virt_pxe_boot(self, num)