def test_get_driver(self, mock_drv_manager): mock_drv_manager.return_value = self.drv_manager self.assertEqual('fake_obj', utils.get_driver('fake_driver'))
def __init__(self, data): self.driver = utils.get_driver(CONF.data_driver)(data) self.partition_scheme = None self.configdrive_scheme = None self.image_scheme = None
def __init__(self, data): self.driver = utils.get_driver(CONF.data_driver)(data)
def test_get_driver(self, mock_drv_manager): mock_drv_manager.return_value = self.drv_manager self.assertEqual(mock.MagicMock.__name__, utils.get_driver('fake_driver').__name__)