def test_select_driver(mock_pkg_resources, mock_input):
driver = Mock()
driver.name = "testdriver"
mock_pkg_resources.iter_entry_points.return_value = (driver, )
mock_input.side_effect = ('a', '1')
assert cli._select_driver() == 'testdriver'
def test_select_driver(mock_input):
driver = Mock()
driver.name = "testdriver"
mock_input.side_effect = ('1')
assert cli._select_driver(None) == 'testdriver'
def test_select_driver(mock_input):
driver = Mock()
driver.name = "testdriver"
mock_input.side_effect = ('1')
assert cli._select_driver(None) == 'testdriver'
