From f3a6e127c313039fdaf19ef8ba64671537d4e214 Mon Sep 17 00:00:00 2001 From: Matt Bertrand Date: Tue, 19 Mar 2024 13:23:09 -0700 Subject: [PATCH] formatting --- ui/utils.py | 4 +++- ui/utils_test.py | 10 +++++----- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/ui/utils.py b/ui/utils.py index 11c0fd68..fb11a407 100644 --- a/ui/utils.py +++ b/ui/utils.py @@ -47,7 +47,9 @@ def get_moira_client(retries=settings.MOIRA_RETRIES): ) for idx in range(retries): try: - return Moira(settings.MIT_WS_CERTIFICATE_FILE, settings.MIT_WS_PRIVATE_KEY_FILE) + return Moira( + settings.MIT_WS_CERTIFICATE_FILE, settings.MIT_WS_PRIVATE_KEY_FILE + ) except Exception as exc: # pylint: disable=broad-except if idx == retries - 1: raise MoiraException( diff --git a/ui/utils_test.py b/ui/utils_test.py index 4375efa4..a9e0f5ba 100644 --- a/ui/utils_test.py +++ b/ui/utils_test.py @@ -70,16 +70,16 @@ def test_get_moira_client_success(mock_moira, settings): settings.MIT_WS_CERTIFICATE_FILE, settings.MIT_WS_PRIVATE_KEY_FILE ) -@pytest.mark.parametrize("retries, raise_error", [ - [3, False], - [2, True] -]) + +@pytest.mark.parametrize("retries, raise_error", [[3, False], [2, True]]) def test_get_moira_client_retries(settings, mocker, retries, raise_error): """Test that the expected # retries are made to get a moira client""" tempfile1, tempfile2 = (NamedTemporaryFile(), NamedTemporaryFile()) settings.MIT_WS_PRIVATE_KEY_FILE = tempfile1.name settings.MIT_WS_CERTIFICATE_FILE = tempfile2.name - mocked_moira = mocker.patch("ui.utils.Moira", side_effect=[Exception, Exception, Mock()]) + mocked_moira = mocker.patch( + "ui.utils.Moira", side_effect=[Exception, Exception, Mock()] + ) if raise_error: with pytest.raises(MoiraException): get_moira_client(retries=retries)