Merge pull request #170 from graphql-dotnet/test-request-cancellation

Test request cancellation

Author: rose-a

tests/GraphQL.Client.Tests.Common/Chat/Schema/ChatQuery.cs

@@ -1,5 +1,7 @@
+using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Threading.Tasks;
 using GraphQL.Types;
 
 namespace GraphQL.Client.Tests.Common.Chat.Schema {
@@ -21,6 +23,13 @@ public ChatQuery(IChat chat) {
 					context.Errors.Add(new ExecutionError("this error contains extension fields", TestExtensions));
 					return null;
 				});
+
+			Field<StringGraphType>()
+				.Name("longRunning")
+				.ResolveAsync(async context => {
+					await Task.Delay(TimeSpan.FromSeconds(5));
+					return "finally returned";
+				});
 		}
 	}
 }

tests/GraphQL.Integration.Tests/QueryAndMutationTests/Base.cs

@@ -1,4 +1,8 @@
+using System;
 using System.Net.Http;
+using System.Threading;
+using System.Threading.Tasks;
+using FluentAssertions;
 using GraphQL.Client.Abstractions;
 using GraphQL.Client.Abstractions.Websocket;
 using GraphQL.Client.Http;
@@ -155,5 +159,22 @@ public async void PreprocessHttpRequestMessageIsCalled() {
 				Assert.Equal("Luke", response.Data.Human.Name);
 			}
 		}
+
+		[Fact]
+		public void PostRequestCanBeCancelled() {
+			var graphQLRequest = new GraphQLRequest(@"
+				query Long {
+					longRunning
+				}");
+
+			using (var setup = WebHostHelpers.SetupTest<StartupChat>(false, serializer)) {
+				var cts = new CancellationTokenSource(TimeSpan.FromSeconds(1));
+
+				Func<Task> requestTask = () => setup.Client.SendQueryAsync(graphQLRequest, () => new {longRunning = string.Empty}, cts.Token);
+				Action timeMeasurement = () => requestTask.Should().Throw<TaskCanceledException>();
+
+				timeMeasurement.ExecutionTime().Should().BeCloseTo(TimeSpan.FromSeconds(2), TimeSpan.FromMilliseconds(200));
+			}
+		}
 	}
 }

tests/GraphQL.Integration.Tests/WebsocketTests/Base.cs

@@ -3,6 +3,7 @@
 using System.Diagnostics;
 using System.Net.WebSockets;
 using System.Threading;
+using System.Threading.Tasks;
 using FluentAssertions;
 using GraphQL.Client.Abstractions;
 using GraphQL.Client.Abstractions.Websocket;
@@ -38,7 +39,6 @@ public async void AssertTestingHarness() {
 			}
 		}
 
-
 		[Fact]
 		public async void CanSendRequestViaWebsocket() {
 			var port = NetworkHelpers.GetFreeTcpPortNumber();
@@ -51,6 +51,23 @@ public async void CanSendRequestViaWebsocket() {
 			}
 		}
 
+		[Fact]
+		public void WebsocketRequestCanBeCancelled() {
+			var graphQLRequest = new GraphQLRequest(@"
+				query Long {
+					longRunning
+				}");
+
+			using (var setup = WebHostHelpers.SetupTest<StartupChat>(true, Serializer)) {
+				var cts = new CancellationTokenSource(TimeSpan.FromSeconds(1));
+
+				Func<Task> requestTask = () => setup.Client.SendQueryAsync(graphQLRequest, () => new { longRunning = string.Empty }, cts.Token);
+				Action timeMeasurement = () => requestTask.Should().Throw<TaskCanceledException>();
+
+				timeMeasurement.ExecutionTime().Should().BeCloseTo(TimeSpan.FromSeconds(2), TimeSpan.FromMilliseconds(200));
+			}
+		}
+
 		[Fact]
 		public async void CanHandleRequestErrorViaWebsocket() {
 			var port = NetworkHelpers.GetFreeTcpPortNumber();