From e7f94059f5a141354de945d11d0eac65b199781d Mon Sep 17 00:00:00 2001 From: Rakshith Bhyravabhotla Date: Fri, 16 Jul 2021 11:58:33 -0700 Subject: [PATCH] emdpoint --- sdk/eventgrid/azure-eventgrid/tests/eventgrid_preparer.py | 3 +++ .../azure-eventgrid/tests/test_eg_publisher_client.py | 3 ++- .../azure-eventgrid/tests/test_eg_publisher_client_async.py | 3 ++- 3 files changed, 7 insertions(+), 2 deletions(-) diff --git a/sdk/eventgrid/azure-eventgrid/tests/eventgrid_preparer.py b/sdk/eventgrid/azure-eventgrid/tests/eventgrid_preparer.py index 0ea1558a240b..c686eb1cefdf 100644 --- a/sdk/eventgrid/azure-eventgrid/tests/eventgrid_preparer.py +++ b/sdk/eventgrid/azure-eventgrid/tests/eventgrid_preparer.py @@ -115,6 +115,9 @@ class EventGridTest(AzureMgmtTestCase): def __init__(self, method_name): super(EventGridTest, self).__init__(method_name) + def get_oauth_endpoint(self): + return os.getenv("EG_TOPIC_HOSTNAME") + def generate_oauth_token(self): if self.is_live: from azure.identity import ClientSecretCredential diff --git a/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client.py b/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client.py index fa55c9b90db8..31ff141f3586 100644 --- a/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client.py +++ b/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client.py @@ -347,11 +347,12 @@ def test_send_throws_with_bad_credential(self): with pytest.raises(ValueError, match="The provided credential should be an instance of a TokenCredential, AzureSasCredential or AzureKeyCredential"): client = EventGridPublisherClient("eventgrid_endpoint", bad_credential) + @pytest.mark.live_test_only @CachedResourceGroupPreparer(name_prefix='eventgridtest') @CachedEventGridTopicPreparer(name_prefix='eventgridtest') def test_send_token_credential(self, resource_group, eventgrid_topic, eventgrid_topic_primary_key, eventgrid_topic_endpoint): credential = self.generate_oauth_token() - client = EventGridPublisherClient(eventgrid_topic_endpoint, credential) + client = EventGridPublisherClient(self.get_oauth_endpoint(), credential) eg_event = EventGridEvent( subject="sample", data={"sample": "eventgridevent"}, diff --git a/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client_async.py b/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client_async.py index 165ffa52aeb5..0ef2e24b0c40 100644 --- a/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client_async.py +++ b/sdk/eventgrid/azure-eventgrid/tests/test_eg_publisher_client_async.py @@ -332,12 +332,13 @@ def test_send_NONE_credential_async(self, resource_group, eventgrid_topic, event with pytest.raises(ValueError, match="Parameter 'self._credential' must not be None."): client = EventGridPublisherClient(eventgrid_topic_endpoint, None) + @pytest.mark.live_test_only @CachedResourceGroupPreparer(name_prefix='eventgridtest') @CachedEventGridTopicPreparer(name_prefix='eventgridtest') @pytest.mark.asyncio async def test_send_token_credential(self, resource_group, eventgrid_topic, eventgrid_topic_primary_key, eventgrid_topic_endpoint): credential = self.generate_oauth_token() - client = EventGridPublisherClient(eventgrid_topic_endpoint, credential) + client = EventGridPublisherClient(self.get_oauth_endpoint(), credential) eg_event = EventGridEvent( subject="sample", data={"sample": "eventgridevent"},