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'