def test_before_scenario_no_driver(start_driver, add_assert_screenshot_methods):
# Create context mock
context = mock.MagicMock()
context.toolium_config = ExtendedConfigParser()
scenario = mock.MagicMock()
scenario.tags = ['a', 'no_driver', 'b']
before_scenario(context, scenario)
# Check that start_driver is called
start_driver.assert_called_once_with(context, True)
def test_before_scenario_reset_driver(start_driver, DriverWrappersPool, add_assert_screenshot_methods):
# Create context mock
context = mock.MagicMock()
context.toolium_config = ExtendedConfigParser()
scenario = mock.MagicMock()
scenario.tags = ['a', 'reset_driver', 'b']
before_scenario(context, scenario)
# Check that start_driver and stop drivers are called
start_driver.assert_called_once_with(context, False)
DriverWrappersPool.stop_drivers.assert_called_once_with()