Unit Testing with Azure Event Hub Receive Event

receive_event_test.py

from azure.eventhub import EventDat

class TestMethods(unittest.TestCase):
    @patch('azure.eventhub.EventHubConsumerClient.from_connection_string')
    @patch('azure.eventhub.extensions.checkpointstoreblob.BlobCheckpointStore.from_connection_string')
    def test_eventhub_receive_events(self,
                                     mock_blob_checkpoint_store,
                                     mock_event_hub_consumer_client):
        '''Test Case 1: Receive events using MyEventHub'''
        def receive(on_event, **kwargs):
            partition_context = MagicMock()
            partition_context.update_checkpoint = MagicMock()
            on_event(partition_context, EventData("Mock"))
        mock_event_hub_consumer_client.receive = receive
        eh_obj = MyEventHub("foo", "bar", "hi", "there", "blah")
        received_events = eh_obj.receive_events(receive_duration=1)
        self.assertEqual([], received_events)