def test_create_driver(self, mock_driver_class): mock_driver = Mock() mock_driver_class.return_value = mock_driver mock_driver.find_elements_by_id.side_effect = [True, False] create_webdriver() mock_driver.add_cookie.assert_called_once()
def test_create_driver(self, mock_driver_class): mock_driver = Mock() mock_driver_class.return_value = mock_driver mock_driver.find_elements_by_id.side_effect = [True, False] create_webdriver() create_webdriver() mock_driver.add_cookie.assert_called_once()