async def test_client_secret_credential_async(aad_credential, live_eventhub):
try:
from azure.identity.aio import ClientSecretCredential
except ImportError:
pytest.skip("No azure identity library")
client_id, secret, tenant_id = aad_credential
credential = ClientSecretCredential(client_id=client_id,
client_secret=secret,
tenant_id=tenant_id)
client = EventHubClient(host=live_eventhub['hostname'],
event_hub_path=live_eventhub['event_hub'],
credential=credential,
user_agent='customized information')
sender = client.create_producer(partition_id='0')
receiver = client.create_consumer(consumer_group="$default",
partition_id='0',
event_position=EventPosition("@latest"))
async with receiver:
received = await receiver.receive(timeout=3)
assert len(received) == 0
async with sender:
event = EventData(body='A single message')
await sender.send(event)
await asyncio.sleep(1)
received = await receiver.receive(timeout=3)
assert len(received) == 1
assert list(received[0].body)[0] == 'A single message'.encode('utf-8')
async def test_long_running_partition_send_async(connection_str):
parser = argparse.ArgumentParser()
parser.add_argument("--duration", help="Duration in seconds of the test", type=int, default=30)
parser.add_argument("--payload", help="payload size", type=int, default=1024)
parser.add_argument("--partitions", help="Comma separated partition IDs")
parser.add_argument("--conn-str", help="EventHub connection string", default=connection_str)
parser.add_argument("--eventhub", help="Name of EventHub")
parser.add_argument("--address", help="Address URI to the EventHub entity")
parser.add_argument("--sas-policy", help="Name of the shared access policy to authenticate with")
parser.add_argument("--sas-key", help="Shared access key")
parser.add_argument("--logger-name", help="Unique log file ID")
loop = asyncio.get_event_loop()
args, _ = parser.parse_known_args()
if args.conn_str:
client = EventHubClient.from_connection_string(
args.conn_str,
event_hub_path=args.eventhub, network_tracing=False)
elif args.address:
client = EventHubClient(host=args.address,
event_hub_path=args.eventhub,
credential=EventHubSharedKeyCredential(args.sas_policy, args.sas_key),
network_tracing=False)
else:
try:
import pytest
pytest.skip("Must specify either '--conn-str' or '--address'")
except ImportError:
raise ValueError("Must specify either '--conn-str' or '--address'")
try:
if not args.partitions:
partitions = await client.get_partition_ids()
else:
pid_range = args.partitions.split("-")
if len(pid_range) > 1:
partitions = [str(i) for i in range(int(pid_range[0]), int(pid_range[1]) + 1)]
else:
partitions = args.partitions.split(",")
pumps = []
for pid in partitions:
sender = client.create_producer(partition_id=pid, send_timeout=0)
pumps.append(pump(pid, sender, args, args.duration))
results = await asyncio.gather(*pumps, return_exceptions=True)
assert not results
except Exception as e:
logger.error("EventHubProducer failed: {}".format(e))
