Added extra testcases for subscription and create server per testcase

sahb1239 · Aug 3, 2019 · 0b400af · 0b400af
1 parent 6baf43b
commit 0b400af
Show file tree

Hide file tree

Showing 3 changed files with 230 additions and 9 deletions.
diff --git a/src/SAHB.GraphQL.Client.Subscription/GraphQLSubscriptionClient.cs b/src/SAHB.GraphQL.Client.Subscription/GraphQLSubscriptionClient.cs
@@ -17,6 +17,7 @@ namespace SAHB.GraphQLClient.Subscription
     public class GraphQLSubscriptionClient : IGraphQLSubscriptionClient
     {
         private long _operationCounter = 1;
+        private object _locker = new object();
         private readonly Dictionary<string, GraphQLOperationSource> _operations = new Dictionary<string, GraphQLOperationSource>();
 
         private const int ReceiveChunkSize = 1024;
@@ -91,7 +92,11 @@ public async Task<IGraphQLSubscriptionOperation<T>> ExecuteOperation<T>(params G
                 throw new InvalidOperationException("GraphQLSubscriptionClient is not initilized");
 
             // Get operationId
-            var operationId = _operationCounter++;
+            long operationId;
+            lock (_locker)
+            {
+                operationId = _operationCounter++;
+            }
 
             // Get query
             var selectionSet = FieldBuilder.GenerateSelectionSet(typeof(T));

diff --git a/tests/SAHB.GraphQL.Client.Subscription.Integration.Tests/Chat/TestChat.cs b/tests/SAHB.GraphQL.Client.Subscription.Integration.Tests/Chat/TestChat.cs
@@ -13,13 +13,13 @@
 
 namespace SAHB.GraphQLClient.Subscription.Integration.Tests
 {
-    public class TestChat : IClassFixture<GraphQLWebApplicationFactory<ChatSchema>>
+    public class TestChat
     {
         private readonly GraphQLWebApplicationFactory<ChatSchema> _factory;
 
-        public TestChat(GraphQLWebApplicationFactory<ChatSchema> factory)
+        public TestChat()
         {
-            _factory = factory;
+            _factory = new GraphQLWebApplicationFactory<ChatSchema>();
         }
 
         [Fact]
@@ -48,9 +48,9 @@ public async Task Test_Recieve_Message()
 
             // Act
             // Registrer operation
-            var operation = await graphQLSubscriptionClient.ExecuteOperation<MessageSubscription>();
+            var operation = await graphQLSubscriptionClient.ExecuteOperation<MessageAddedSubscription>();
 
-            List<GraphQLDataResult<MessageSubscription>> recievedData = new List<GraphQLDataResult<MessageSubscription>>();
+            List<GraphQLDataResult<MessageAddedSubscription>> recievedData = new List<GraphQLDataResult<MessageAddedSubscription>>();
             operation.DataRecieved += (sender, e) =>
             {
                 recievedData.Add(e.ReceivedData);
@@ -65,17 +65,233 @@ public async Task Test_Recieve_Message()
                 sentAt = DateTime.Now.AddDays(-1)
             })).Execute();
 
+            await Task.Delay(TimeSpan.FromSeconds(2));
+
+            // Assert
+            Assert.Single(recievedData);
+        }
+
+        [Fact]
+        public async Task Test_Recieve_Multiple_Messages()
+        {
+            // Arrange
+            var tokenSource = new CancellationTokenSource();
+
+            var client = _factory.CreateClient();
+            var graphQLClient = GraphQLHttpClient.Default(client);
+
+            // Create websocket GraphQL client
+            var wsClient = _factory.Server.CreateWebSocketClient();
+            wsClient.ConfigureRequest = request => request.Headers.Add("Sec-WebSocket-Protocol", "graphql-ws");
+
+            var wsUri = new UriBuilder(_factory.Server.BaseAddress)
+            {
+                Scheme = "wss",
+                Path = "graphql"
+            }.Uri;
+
+            var websocket = await wsClient.ConnectAsync(wsUri, tokenSource.Token);
+
+            var graphQLSubscriptionClient = new GraphQLSubscriptionClient(websocket, tokenSource.Token, new GraphQLFieldBuilder(), new GraphQLQueryGeneratorFromFields(), new GraphQLDeserilization());
+            await graphQLSubscriptionClient.Initilize();
+
+            // Act
+            // Registrer operation
+            var operation = await graphQLSubscriptionClient.ExecuteOperation<MessageAddedSubscription>();
+
+            List<GraphQLDataResult<MessageAddedSubscription>> recievedData = new List<GraphQLDataResult<MessageAddedSubscription>>();
+            operation.DataRecieved += (sender, e) =>
+            {
+                recievedData.Add(e.ReceivedData);
+            };
             await Task.Delay(TimeSpan.FromSeconds(1));
 
+            // Send 2 messages
+            await graphQLClient.CreateMutation<SendMessageMutation>("http://localhost/graphql", arguments: new GraphQLQueryArgument("message", new MessageInputType
+            {
+                content = "Message 1",
+                fromId = "SAHB",
+                sentAt = DateTime.Now.AddDays(-1)
+            })).Execute();
+            await graphQLClient.CreateMutation<SendMessageMutation>("http://localhost/graphql", arguments: new GraphQLQueryArgument("message", new MessageInputType
+            {
+                content = "Message 1",
+                fromId = "SAHB",
+                sentAt = DateTime.Now.AddDays(-1)
+            })).Execute();
+
+            await Task.Delay(TimeSpan.FromSeconds(2));
+
             // Assert
+            Assert.Equal(2, recievedData.Count);
+        }
+
+        [Fact]
+        public async Task Test_Recieve_Message_By_User()
+        {
+            // Arrange
+            var tokenSource = new CancellationTokenSource();
+
+            var client = _factory.CreateClient();
+            var graphQLClient = GraphQLHttpClient.Default(client);
+
+            // Create websocket GraphQL client
+            var wsClient = _factory.Server.CreateWebSocketClient();
+            wsClient.ConfigureRequest = request => request.Headers.Add("Sec-WebSocket-Protocol", "graphql-ws");
+
+            var wsUri = new UriBuilder(_factory.Server.BaseAddress)
+            {
+                Scheme = "wss",
+                Path = "graphql"
+            }.Uri;
+
+            var websocket = await wsClient.ConnectAsync(wsUri, tokenSource.Token);
+
+            var graphQLSubscriptionClient = new GraphQLSubscriptionClient(websocket, tokenSource.Token, new GraphQLFieldBuilder(), new GraphQLQueryGeneratorFromFields(), new GraphQLDeserilization());
+            await graphQLSubscriptionClient.Initilize();
+
+            // Act
+            // Registrer operation
+            var operation = await graphQLSubscriptionClient.ExecuteOperation<MessageAddedByUserSubscription>(new GraphQLQueryArgument("fromId", "SAHB"));
+
+            List<GraphQLDataResult<MessageAddedByUserSubscription>> recievedData = new List<GraphQLDataResult<MessageAddedByUserSubscription>>();
+            operation.DataRecieved += (sender, e) =>
+            {
+                recievedData.Add(e.ReceivedData);
+            };
+            await Task.Delay(TimeSpan.FromSeconds(1));
+
+            // Send 2 message from 2 users only one should be retrieved in subscription
+            await graphQLClient.CreateMutation<SendMessageMutation>("http://localhost/graphql", arguments: new GraphQLQueryArgument("message", new MessageInputType
+            {
+                content = "Message 1",
+                fromId = "SAHB",
+                sentAt = DateTime.Now.AddDays(-1)
+            })).Execute();
+
+            await graphQLClient.CreateMutation<SendMessageMutation>("http://localhost/graphql", arguments: new GraphQLQueryArgument("message", new MessageInputType
+            {
+                content = "Message 1",
+                fromId = "OtherUser",
+                sentAt = DateTime.Now.AddDays(-1)
+            })).Execute();
+
+            await Task.Delay(TimeSpan.FromSeconds(2));
+
+            // Assert
+            Assert.Single(recievedData);
+        }
+
+        [Fact]
+        public async Task Test_Recieve_Multiple_Subscribers()
+        {
+            // Arrange
+            var tokenSource = new CancellationTokenSource();
+
+            var client = _factory.CreateClient();
+            var graphQLClient = GraphQLHttpClient.Default(client);
+
+            // Create websocket GraphQL client
+            var wsClient = _factory.Server.CreateWebSocketClient();
+            wsClient.ConfigureRequest = request => request.Headers.Add("Sec-WebSocket-Protocol", "graphql-ws");
+
+            var wsUri = new UriBuilder(_factory.Server.BaseAddress)
+            {
+                Scheme = "wss",
+                Path = "graphql"
+            }.Uri;
+
+            var websocket = await wsClient.ConnectAsync(wsUri, tokenSource.Token);
+
+            var graphQLSubscriptionClient = new GraphQLSubscriptionClient(websocket, tokenSource.Token, new GraphQLFieldBuilder(), new GraphQLQueryGeneratorFromFields(), new GraphQLDeserilization());
+            await graphQLSubscriptionClient.Initilize();
+
+            // Act
+            // Registrer operations
+            var operationByUser = await graphQLSubscriptionClient.ExecuteOperation<MessageAddedByUserSubscription>(new GraphQLQueryArgument("fromId", "SAHB"));
+
+            List<GraphQLDataResult<MessageAddedByUserSubscription>> recievedDataByUser = new List<GraphQLDataResult<MessageAddedByUserSubscription>>();
+            operationByUser.DataRecieved += (sender, e) =>
+            {
+                recievedDataByUser.Add(e.ReceivedData);
+            };
+
+            var operation = await graphQLSubscriptionClient.ExecuteOperation<MessageAddedSubscription>();
+
+            List<GraphQLDataResult<MessageAddedSubscription>> recievedData = new List<GraphQLDataResult<MessageAddedSubscription>>();
+            operation.DataRecieved += (sender, e) =>
+            {
+                recievedData.Add(e.ReceivedData);
+            };
+            await Task.Delay(TimeSpan.FromSeconds(1));
+
+            // Send 2 message from 2 users only one should be retrieved in subscription
+            await graphQLClient.CreateMutation<SendMessageMutation>("http://localhost/graphql", arguments: new GraphQLQueryArgument("message", new MessageInputType
+            {
+                content = "Message 1",
+                fromId = "SAHB",
+                sentAt = DateTime.Now.AddDays(-1)
+            })).Execute();
+
+            await Task.Delay(TimeSpan.FromSeconds(1));
+
+            // Assert
+            Assert.Single(recievedDataByUser);
             Assert.Single(recievedData);
         }
 
-        public class MessageSubscription
+        [Fact]
+        public async Task Test_Complete()
+        {
+            // Arrange
+            var tokenSource = new CancellationTokenSource();
+
+            var client = _factory.CreateClient();
+            var graphQLClient = GraphQLHttpClient.Default(client);
+
+            // Create websocket GraphQL client
+            var wsClient = _factory.Server.CreateWebSocketClient();
+            wsClient.ConfigureRequest = request => request.Headers.Add("Sec-WebSocket-Protocol", "graphql-ws");
+
+            var wsUri = new UriBuilder(_factory.Server.BaseAddress)
+            {
+                Scheme = "wss",
+                Path = "graphql"
+            }.Uri;
+
+            var websocket = await wsClient.ConnectAsync(wsUri, tokenSource.Token);
+
+            var graphQLSubscriptionClient = new GraphQLSubscriptionClient(websocket, tokenSource.Token, new GraphQLFieldBuilder(), new GraphQLQueryGeneratorFromFields(), new GraphQLDeserilization());
+            await graphQLSubscriptionClient.Initilize();
+
+            // Act
+            List<bool> completedTimes = new List<bool>();
+
+            // Registrer operations and stop it
+            var operation = await graphQLSubscriptionClient.ExecuteOperation<MessageAddedSubscription>();
+            operation.Completed += (sender, e) =>
+            {
+                completedTimes.Add(true);
+            };
+            await operation.Stop();
+
+            await Task.Delay(TimeSpan.FromSeconds(2));
+
+            // Assert
+            Assert.Single(completedTimes);
+        }
+
+        public class MessageAddedSubscription
         {
             public Message MessageAdded { get; set; }
         }
 
+        public class MessageAddedByUserSubscription
+        {
+            [GraphQLArguments("id", "String", "fromId", IsRequired = true)]
+            public Message messageAddedByUser { get; set; }
+        }
+
         public class Message
         {
             public Author From { get; set; }

diff --git a/....Subscription.Integration.Tests/SAHB.GraphQL.Client.Subscription.Integration.Tests.csproj b/....Subscription.Integration.Tests/SAHB.GraphQL.Client.Subscription.Integration.Tests.csproj
@@ -7,8 +7,8 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="GraphQL.Server.Transports.AspNetCore" Version="3.4.0" />
-    <PackageReference Include="GraphQL.Server.Transports.WebSockets" Version="3.4.0" />
+    <PackageReference Include="GraphQL.Server.Transports.AspNetCore" Version="3.2.0" />
+    <PackageReference Include="GraphQL.Server.Transports.WebSockets" Version="3.2.0" />
     <PackageReference Include="Microsoft.AspNetCore.App" Version="2.2.4" />
     <PackageReference Include="Microsoft.AspNetCore.Mvc.Testing" Version="2.2.0" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.0.1" />