Mocking SDKs in python [4]

test_redirect_to_bank_controller.py

from httpx import Response, Client
from app import app
from factories import UserFactory, BasketFactory
from unittest.mock import MagicMock

def test_purchase_basket_controller():
    get_gateway_url = payment_sdk.get_gateway_url
    payment_sdk.get_gateway_url = MagicMock(return_value='https://sample_bank_url.test')

    with Client(app=app, base_url="http://testserver.test", timeout=0.1) as client:
        user = UserFactory.create_one()
        basket = BasketFactory.create_one(user_id=user.id)
        response = client.post(
            url='/basket/purchase',
            headers=dict(Accept='application/json')
        )

        assert response.status_code == 307

        payment_sdk.get_gateway_url = get_gateway_url