[Web PubSub] Adding live test for web pubsub (Azure#25106)

* Adding live test for web pubsub

* Add live test

* Adding arm template

* Ignore live test until it is supported by CI

* Update code snnipet

Author: vicancy

sdk/webpubsub/Azure.Messaging.WebPubSub/README.md

@@ -73,15 +73,15 @@ When a client is connected, it can send messages to the upstream application, or
 ### Broadcast a text message to all clients
 
 ```C# Snippet:WebPubSubHelloWorld
-var serviceClient = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+var serviceClient = new WebPubSubServiceClient(connectionString, "some_hub");
 
 serviceClient.SendToAll("Hello World!");
 ```
 
 ### Broadcast a JSON message to all clients
 
 ```C# Snippet:WebPubSubSendJson
-var serviceClient = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+var serviceClient = new WebPubSubServiceClient(connectionString, "some_hub");
 
 serviceClient.SendToAll(RequestContent.Create(
         new
@@ -95,7 +95,7 @@ serviceClient.SendToAll(RequestContent.Create(
 ### Broadcast a binary message to all clients
 
 ```C# Snippet:WebPubSubSendBinary
-var serviceClient = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+var serviceClient = new WebPubSubServiceClient(connectionString, "some_hub");
 
 Stream stream = BinaryData.FromString("Hello World!").ToStream();
 serviceClient.SendToAll(RequestContent.Create(stream), ContentType.ApplicationOctetStream);

sdk/webpubsub/Azure.Messaging.WebPubSub/tests/Samples/WebPubSubSamples.HelloWorld.cs

@@ -2,7 +2,10 @@
 // Licensed under the MIT License.
 
 using System;
+using System.Collections.Generic;
 using System.IO;
+using System.Linq;
+using System.Web.UI;
 
 using Azure.Core;
 using Azure.Core.TestFramework;
@@ -15,20 +18,20 @@ public class WebPubSubSamples : SamplesBase<WebPubSubTestEnvironment>
     {
         public void HelloWorld()
         {
-            var endpoint = TestEnvironment.Endpoint;
-            var key = TestEnvironment.Key;
+            var connectionString = TestEnvironment.ConnectionString;
 
             #region Snippet:WebPubSubHelloWorld
-            var serviceClient = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+            var serviceClient = new WebPubSubServiceClient(connectionString, "some_hub");
 
             serviceClient.SendToAll("Hello World!");
             #endregion
         }
 
         public void Authenticate()
         {
-            var endpoint = TestEnvironment.Endpoint;
-            var key = TestEnvironment.Key;
+            var connectionString = TestEnvironment.ConnectionString;
+            var endpoint = ParseConnectionString(connectionString)["Endpoint"];
+            var key = ParseConnectionString(connectionString)["AccessKey"];
 
             #region Snippet:WebPubSubAuthenticate
             // Create a WebPubSubServiceClient that will authenticate using a key credential.
@@ -47,11 +50,10 @@ public void HelloWorldWithConnectionString()
 
         public void JsonMessage()
         {
-            var endpoint = TestEnvironment.Endpoint;
-            var key = TestEnvironment.Key;
+            var connectionString = TestEnvironment.ConnectionString;
 
             #region Snippet:WebPubSubSendJson
-            var serviceClient = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+            var serviceClient = new WebPubSubServiceClient(connectionString, "some_hub");
 
             serviceClient.SendToAll(RequestContent.Create(
                     new
@@ -65,11 +67,10 @@ public void JsonMessage()
 
         public void BinaryMessage()
         {
-            var endpoint = TestEnvironment.Endpoint;
-            var key = TestEnvironment.Key;
+            var connectionString = TestEnvironment.ConnectionString;
 
             #region Snippet:WebPubSubSendBinary
-            var serviceClient = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+            var serviceClient = new WebPubSubServiceClient(connectionString, "some_hub");
 
             Stream stream = BinaryData.FromString("Hello World!").ToStream();
             serviceClient.SendToAll(RequestContent.Create(stream), ContentType.ApplicationOctetStream);
@@ -78,11 +79,10 @@ public void BinaryMessage()
 
         public void AddUserToGroup()
         {
-            var endpoint = TestEnvironment.Endpoint;
-            var key = TestEnvironment.Key;
-
-            var client = new WebPubSubServiceClient(new Uri(endpoint), "some_hub", new AzureKeyCredential(key));
+            var connectionString = TestEnvironment.ConnectionString;
+            var client = new WebPubSubServiceClient(connectionString, "some_hub");
 
+            #region Snippet:WebPubSubAddUserToGroup
             client.AddUserToGroup("some_group", "some_user");
 
             // Avoid sending messages to users who do not exist.
@@ -92,6 +92,20 @@ public void AddUserToGroup()
             }
 
             client.RemoveUserFromGroup("some_group", "some_user");
+            #endregion
+        }
+
+        private static Dictionary<string, string> ParseConnectionString(string connectionString)
+        {
+            return connectionString.Split(new char[] { ';' }, StringSplitOptions.RemoveEmptyEntries).Select(pair =>
+            {
+                var map = pair.Split('=');
+                if (map.Length != 2)
+                {
+                    return default;
+                }
+                return new KeyValuePair<string, string>(map[0], map[1]);
+            }).Where(s => !string.IsNullOrEmpty(s.Key)).ToDictionary(p => p.Key, p => p.Value, StringComparer.OrdinalIgnoreCase);
         }
     }
 }

sdk/webpubsub/Azure.Messaging.WebPubSub/tests/SessionRecords/WebPubSubGeneralTests/ServiceClientCanBroadcastMessages.json

@@ -0,0 +1,20 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System;
+using System.Threading.Tasks;
+
+namespace Azure.Rest.WebPubSub.Tests
+{
+    public static class TaskExtensions
+    {
+        public static async Task OrTimeout(this Task task, int millisecondsDelay = 5000)
+        {
+            var completed = await Task.WhenAny(task, Task.Delay(millisecondsDelay));
+            if (!task.IsCompleted)
+            {
+                throw new TimeoutException();
+            }
+        }
+    }
+}

sdk/webpubsub/Azure.Messaging.WebPubSub/tests/SessionRecords/WebPubSubGeneralTests/ServiceClientCanBroadcastMessagesAsync.json

@@ -0,0 +1,47 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Net.WebSockets;
+using System.Text;
+using System.Text.Json;
+using System.Threading;
+using System.Threading.Tasks;
+
+using Azure.Core;
+using Azure.Core.TestFramework;
+using Azure.Messaging.WebPubSub;
+
+using NUnit.Framework;
+
+namespace Azure.Rest.WebPubSub.Tests
+{
+    public class WebPubSubGeneralTests : RecordedTestBase<WebPubSubTestEnvironment>
+    {
+        public WebPubSubGeneralTests(bool isAsync) : base(isAsync)
+        {
+        }
+
+        [Test]
+        public async Task ServiceClientCanBroadcastMessages()
+        {
+            WebPubSubServiceClientOptions options = InstrumentClientOptions(new WebPubSubServiceClientOptions());
+
+            var serviceClient = new WebPubSubServiceClient(TestEnvironment.ConnectionString, nameof(ServiceClientCanBroadcastMessages), options);
+            // broadcast messages
+            var textContent = "Hello";
+            var response = await serviceClient.SendToAllAsync(textContent, ContentType.TextPlain);
+
+            Assert.AreEqual(202, response.Status);
+
+            var jsonContent = BinaryData.FromObjectAsJson(new { hello = "world" });
+            response = await serviceClient.SendToAllAsync(RequestContent.Create(jsonContent), ContentType.ApplicationJson);
+            Assert.AreEqual(202, response.Status);
+            var binaryContent = BinaryData.FromString("Hello");
+            response = await serviceClient.SendToAllAsync(RequestContent.Create(binaryContent), ContentType.ApplicationOctetStream);
+            Assert.AreEqual(202, response.Status);
+        }
+    }
+}