From 97baa4c184da214eb0631ca72b6131b60f98144f Mon Sep 17 00:00:00 2001 From: Kenneth Kehl <@kkehl@flexion.us> Date: Wed, 10 Jan 2024 11:38:23 -0800 Subject: [PATCH] flake8 --- tests/app/user/test_rest_verify.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/app/user/test_rest_verify.py b/tests/app/user/test_rest_verify.py index d7d42a019..f6f29f518 100644 --- a/tests/app/user/test_rest_verify.py +++ b/tests/app/user/test_rest_verify.py @@ -206,6 +206,10 @@ def test_send_user_sms_code(client, sample_user, sms_code_template, mocker): """ notify_service = dao_fetch_service_by_id(current_app.config["NOTIFY_SERVICE_ID"]) + mock_redis_get = mocker.patch("app.celery.scheduled_tasks.redis_store.raw_get") + mock_redis_get.return_value = "foo" + + mocker.patch("app.celery.scheduled_tasks.redis_store.raw_set") auth_header = create_admin_authorization_header() mocked = mocker.patch("app.user.rest.create_secret_code", return_value="11111") mocker.patch("app.celery.provider_tasks.deliver_sms.apply_async") @@ -238,6 +242,11 @@ def test_send_user_code_for_sms_with_optional_to_field( """ Tests POST endpoint /user//sms-code with optional to field """ + + mock_redis_get = mocker.patch("app.celery.scheduled_tasks.redis_store.raw_get") + mock_redis_get.return_value = "foo" + + mocker.patch("app.celery.scheduled_tasks.redis_store.raw_set") to_number = "+447119876757" mocked = mocker.patch("app.user.rest.create_secret_code", return_value="11111") mocker.patch("app.celery.provider_tasks.deliver_sms.apply_async") @@ -482,6 +491,11 @@ def test_send_user_email_code_with_urlencoded_next_param( ): mocker.patch("app.celery.provider_tasks.deliver_email.apply_async") + mock_redis_get = mocker.patch("app.celery.scheduled_tasks.redis_store.raw_get") + mock_redis_get.return_value = "foo" + + mocker.patch("app.celery.scheduled_tasks.redis_store.raw_set") + data = {"to": None, "next": "/services"} admin_request.post( "user.send_user_2fa_code", @@ -556,6 +570,13 @@ def test_user_verify_email_code_fails_if_code_already_used( def test_send_user_2fa_code_sends_from_number_for_international_numbers( client, sample_user, mocker, sms_code_template ): + + + mock_redis_get = mocker.patch("app.celery.scheduled_tasks.redis_store.raw_get") + mock_redis_get.return_value = "foo" + + mocker.patch("app.celery.scheduled_tasks.redis_store.raw_set") + sample_user.mobile_number = "+601117224412" auth_header = create_admin_authorization_header() mocker.patch("app.user.rest.create_secret_code", return_value="11111")