Python requests_mock 模块，mock() 实例源码

我们从Python开源项目中，提取了以下50个代码示例，用于说明如何使用requests_mock.mock()。

项目：abodepy 作者：MisterWil | 项目源码 | 文件源码

def tests_event_registration(self):
        """Tests that events register correctly."""
        # Get the event controller
        events = self.abode.events
        self.assertIsNotNone(events)

        # Create mock callback
        callback = Mock()

        # Test that a valid event registers
        self.assertTrue(
            events.add_event_callback(TIMELINE.ALARM_GROUP, callback))

        # Test that no event group returns false
        self.assertFalse(events.add_event_callback(None, callback))

        # Test that an invalid event throws exception
        with self.assertRaises(abodepy.AbodeException):
            events.add_event_callback("lol", callback)

项目：abodepy 作者：MisterWil | 项目源码 | 文件源码

def tests_timeline_registration(self):
        """Tests that timeline events register correctly."""
        # Get the event controller
        events = self.abode.events
        self.assertIsNotNone(events)

        # Create mock callback
        callback = Mock()

        # Test that a valid timeline event registers
        self.assertTrue(
            events.add_timeline_callback(
                TIMELINE.CAPTURE_IMAGE, callback))

        # Test that no timeline event returns false
        self.assertFalse(events.add_timeline_callback(None, callback))

        # Test that an invalid timeline event string throws exception
        with self.assertRaises(abodepy.AbodeException):
            events.add_timeline_callback("lol", callback)

        # Test that an invalid timeline event dict throws exception
        with self.assertRaises(abodepy.AbodeException):
            events.add_timeline_callback({"lol": "lol"}, callback)

项目：abodepy 作者：MisterWil | 项目源码 | 文件源码

def tests_multi_events_callback(self):
        """Tests that multiple event updates callback correctly."""
        # Get the event controller
        events = self.abode.events
        self.assertIsNotNone(events)

        # Create mock callback
        callback = Mock()

        # Register our events
        self.assertTrue(
            events.add_event_callback(
                [TIMELINE.ALARM_GROUP, TIMELINE.CAPTURE_GROUP],
                callback))

        # Call our events callback method and trigger a capture group event
        # pylint: disable=protected-access
        event_json = json.loads(IRCAMERA.timeline_event())
        events._on_timeline_update(event_json)

        # Ensure our callback was called
        callback.assert_called_with(event_json)

项目：abodepy 作者：MisterWil | 项目源码 | 文件源码

def tests_multi_timeline_callback(self):
        """Tests that multiple timeline updates callback correctly."""
        # Get the event controller
        events = self.abode.events
        self.assertIsNotNone(events)

        # Create mock callback
        callback = Mock()

        # Register our events
        self.assertTrue(
            events.add_timeline_callback(
                [TIMELINE.CAPTURE_IMAGE, TIMELINE.OPENED], callback))

        # Call our events callback method and trigger a capture group event
        # pylint: disable=protected-access
        event_json = json.loads(IRCAMERA.timeline_event())
        events._on_timeline_update(event_json)

        # Ensure our callback was called
        callback.assert_called_with(event_json)

项目：abodepy 作者：MisterWil | 项目源码 | 文件源码

def tests_automations_callback(self):
        """Tests that automation updates callback correctly."""
        # Get the event controller
        events = self.abode.events
        self.assertIsNotNone(events)

        # Create mock callbacks
        automation_callback = Mock()

        # Register our events
        self.assertTrue(
            events.add_event_callback(
                TIMELINE.AUTOMATION_EDIT_GROUP, automation_callback))

        # Call our events callback method and trigger a capture group event
        # pylint: disable=protected-access
        events._on_automation_update('{}')

        # Our capture callback should get one, but our alarm should not
        automation_callback.assert_called_with('{}')

项目：h1-python 作者：uber-common | 项目源码 | 文件源码

def test_find_resources_yield_pages(self):
        page_one = json.dumps(load_resource_blob("responses/report-list-twopage-0"))
        page_two = json.dumps(load_resource_blob("responses/report-list-twopage-1"))
        with requests_mock.mock() as m:
            filter_str = "filter%5Bprogram%5D%5B%5D=foo"
            m.get(HackerOneClient.BASE_URL + "/reports?" + filter_str, text=page_one)
            m.get(HackerOneClient.BASE_URL + "/reports/nextpage", text=page_two)

            client = HackerOneClient("a", "b")
            listing = client.find_resources(Report, program=["foo"], yield_pages=True)

            # The listing should contain two pages each with one report
            self.assertEqual(len(listing), 2)
            # Flatten the pages into a single page
            all_items = list(itertools.chain(*listing))
            self.assertEqual(len(all_items), 2)
            self.assertTrue(all(isinstance(r, Report) for r in all_items))