receive test. """ import logging import asyncio import argparse import time import os from urllib.parse import quote_plus from azure.eventhub import Offset from azure.eventhub.async import EventHubClientAsync try: import tests logger = tests.get_logger("recv_test.log", logging.INFO) except ImportError: logger = logging.getLogger("uamqp") logger.setLevel(logging.INFO) async def pump(_pid, receiver, _args, _dl): total = 0 iteration = 0 deadline = time.time() + _dl try: while time.time() < deadline: batch = await receiver.receive(timeout=5) size = len(batch) total += size iteration += 1
import sys from tests import get_logger from azure import eventhub from azure.eventhub import EventHubClient, Receiver, Offset from azure.eventprocessorhost import EventProcessorHost from azure.eventprocessorhost import EventHubPartitionPump from azure.eventprocessorhost import AzureStorageCheckpointLeaseManager from azure.eventprocessorhost import AzureBlobLease from azure.eventprocessorhost import EventHubConfig from azure.eventprocessorhost.lease import Lease from azure.eventprocessorhost.partition_pump import PartitionPump from azure.eventprocessorhost.partition_manager import PartitionManager from azure.eventprocessorhost.abstract_event_processor import AbstractEventProcessor log = get_logger(None, logging.DEBUG) @pytest.fixture() def live_eventhub_config(): try: config = {} config['hostname'] = os.environ['EVENT_HUB_HOSTNAME'] config['event_hub'] = os.environ['EVENT_HUB_NAME'] config['key_name'] = os.environ['EVENT_HUB_SAS_POLICY'] config['access_key'] = os.environ['EVENT_HUB_SAS_KEY'] config['consumer_group'] = "$Default" config['partition'] = "0" except KeyError: pytest.skip("Live EventHub configuration not found.") else:
""" send test """ import logging import argparse import time import threading import os import asyncio from azure.eventhub import EventHubClientAsync, EventData try: import tests logger = tests.get_logger("send_test.log", logging.INFO) except ImportError: logger = logging.getLogger("uamqp") logger.setLevel(logging.INFO) def check_send_successful(outcome, condition): if outcome.value != 0: print("Send failed {}".format(condition)) async def get_partitions(args): #client = EventHubClientAsync.from_connection_string( # args.conn_str, # eventhub=args.eventhub, debug=True) eh_data = await args.get_eventhub_info_async()
import sys from tests import get_logger from azure import eventhub from azure.eventhub import EventHubClient, Receiver, Offset from azure.eventprocessorhost import EventProcessorHost from azure.eventprocessorhost import EventHubPartitionPump from azure.eventprocessorhost import AzureStorageCheckpointLeaseManager from azure.eventprocessorhost import AzureBlobLease from azure.eventprocessorhost import EventHubConfig from azure.eventprocessorhost.lease import Lease from azure.eventprocessorhost.partition_pump import PartitionPump from azure.eventprocessorhost.partition_manager import PartitionManager from azure.eventprocessorhost.abstract_event_processor import AbstractEventProcessor log = get_logger(None, logging.INFO) @pytest.fixture() def live_eventhub_config(): try: config = {} config['hostname'] = os.environ['EVENT_HUB_HOSTNAME'] config['event_hub'] = os.environ['EVENT_HUB_NAME'] config['key_name'] = os.environ['EVENT_HUB_SAS_POLICY'] config['access_key'] = os.environ['EVENT_HUB_SAS_KEY'] config['consumer_group'] = "$Default" config['partition'] = "0" except KeyError: pytest.skip("Live EventHub configuration not found.") else: