Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
test update
  • Loading branch information
l0lawrence committed May 26, 2024
commit 6e2e53d325f52e963ea4c7ab205a0ef222ded4f6
31 changes: 14 additions & 17 deletions sdk/eventgrid/azure-eventgrid/tests/test_eg_consumer_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,9 @@
from msrest.serialization import UTC
import datetime as dt

from devtools_testutils import AzureRecordedTestCase, recorded_by_proxy
from devtools_testutils import AzureRecordedTestCase
from azure.core.messaging import CloudEvent
from azure.core.credentials import AzureKeyCredential
from azure.eventgrid import EventGridConsumerClient, EventGridPublisherClient
from eventgrid_preparer import (
EventGridPreparer,
Expand Down Expand Up @@ -44,12 +45,11 @@ def create_eg_consumer_client(self, endpoint, topic, subscription):
@EventGridPreparer()
def test_receive_data(self, **kwargs):
eventgrid_endpoint = kwargs["eventgrid_endpoint"]
eventgrid_key = kwargs["eventgrid_key"]
eventgrid_topic_name = kwargs["eventgrid_topic_name"]
eventgrid_event_subscription_name = kwargs["eventgrid_event_subscription_name"]
publisher = self.create_eg_publisher_client(eventgrid_endpoint, eventgrid_topic_name)
consumer = self.create_eg_consumer_client(
eventgrid_endpoint, eventgrid_topic_name, eventgrid_event_subscription_name
)
publisher = EventGridPublisherClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name)
consumer = EventGridConsumerClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name, subscription=eventgrid_event_subscription_name)
cloud_event = CloudEvent(
source="http://samplesource.dev",
data={"sample": "cloudevent"},
Expand All @@ -68,12 +68,11 @@ def test_receive_data(self, **kwargs):
@EventGridPreparer()
def test_receive_renew_data(self, **kwargs):
eventgrid_endpoint = kwargs["eventgrid_endpoint"]
eventgrid_key = kwargs["eventgrid_key"]
eventgrid_topic_name = kwargs["eventgrid_topic_name"]
eventgrid_event_subscription_name = kwargs["eventgrid_event_subscription_name"]
publisher = self.create_eg_publisher_client(eventgrid_endpoint, eventgrid_topic_name)
consumer = self.create_eg_consumer_client(
eventgrid_endpoint, eventgrid_topic_name, eventgrid_event_subscription_name
)
publisher = EventGridPublisherClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name)
consumer = EventGridConsumerClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name, subscription=eventgrid_event_subscription_name)
cloud_event = CloudEvent(
source="http://samplesource.dev",
data={"sample": "cloudevent"},
Expand All @@ -93,12 +92,11 @@ def test_receive_renew_data(self, **kwargs):
@EventGridPreparer()
def test_receive_release_data(self, **kwargs):
eventgrid_endpoint = kwargs["eventgrid_endpoint"]
eventgrid_key = kwargs["eventgrid_key"]
eventgrid_topic_name = kwargs["eventgrid_topic_name"]
eventgrid_event_subscription_name = kwargs["eventgrid_event_subscription_name"]
publisher = self.create_eg_publisher_client(eventgrid_endpoint, eventgrid_topic_name)
consumer = self.create_eg_consumer_client(
eventgrid_endpoint, eventgrid_topic_name, eventgrid_event_subscription_name
)
publisher = EventGridPublisherClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name)
consumer = EventGridConsumerClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name, subscription=eventgrid_event_subscription_name)
cloud_event = CloudEvent(
source="http://samplesource.dev",
data={"sample": "cloudevent"},
Expand All @@ -117,12 +115,11 @@ def test_receive_release_data(self, **kwargs):
@EventGridPreparer()
def test_receive_reject_data(self, **kwargs):
eventgrid_endpoint = kwargs["eventgrid_endpoint"]
eventgrid_key = kwargs["eventgrid_key"]
eventgrid_topic_name = kwargs["eventgrid_topic_name"]
eventgrid_event_subscription_name = kwargs["eventgrid_event_subscription_name"]
publisher = self.create_eg_publisher_client(eventgrid_endpoint, eventgrid_topic_name)
consumer = self.create_eg_consumer_client(
eventgrid_endpoint, eventgrid_topic_name, eventgrid_event_subscription_name
)
publisher = EventGridPublisherClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name)
consumer = EventGridConsumerClient(eventgrid_endpoint, AzureKeyCredential(eventgrid_key), namespace_topic=eventgrid_topic_name, subscription=eventgrid_event_subscription_name)
cloud_event = CloudEvent(
source="http://samplesource.dev",
data={"sample": "cloudevent"},
Expand Down