From 53903d6423a91905b34389b5e550298dd8b6c657 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Paulo=20Carvalho?= Date: Sat, 3 Aug 2024 00:10:45 -0300 Subject: [PATCH] style - aplicando ruff --- tests/test_utils.py | 24 ++++++++++-------------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index 8c4cdf0..a639699 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -3,7 +3,6 @@ from unittest.mock import MagicMock - def test_get_browser_options_default_options_class(): options = utils.get_browser_options(['a', 'b']) assert type(options) is ChromeOptions @@ -17,18 +16,16 @@ def test_get_browser_options(): def test_get_session(): - cookies_mock = [{ - 'name': 'test', - 'value': 'test_value', - 'domain': 'www.site.com', - 'path': '/', - 'secure': False - }] - webdriver_mock = MagicMock( - get_cookies=MagicMock( - return_value=cookies_mock - ) - ) + cookies_mock = [ + { + 'name': 'test', + 'value': 'test_value', + 'domain': 'www.site.com', + 'path': '/', + 'secure': False, + } + ] + webdriver_mock = MagicMock(get_cookies=MagicMock(return_value=cookies_mock)) session = utils.get_session(webdriver_mock) assert len(session.cookies) == 1 assert session.cookies.get('test') == 'test_value' @@ -37,4 +34,3 @@ def test_get_session(): def test_get_domain(): webdriver_mock = MagicMock(current_url='https://www.site.com/') assert utils.get_domain(webdriver_mock) == 'www.site.com' -