zulip/zerver/tests/test_internet.py

import requests
import responses

from zerver.lib.test_classes import ZulipTestCase


class ResponsesTest(ZulipTestCase):
    def test_responses(self) -> None:
        # With our test setup, accessing the internet should be blocked.
        with self.assertRaises(Exception):
            result = requests.request("GET", "https://www.google.com")

        # A test can invoke its own responses.RequestsMock context manager
        # and register URLs to mock, accessible from within the context.
        with responses.RequestsMock() as requests_mock:
            requests_mock.add(
                responses.GET,
                "https://www.google.com",
                body="{}",
                status=200,
                content_type="application/json",
            )
            result = requests.request("GET", "https://www.google.com")
            self.assertEqual(result.status_code, 200)
            self.assertEqual(result.text, "{}")