Add tests for DisableDirectStreaming

Author: gmarz

src/Tests/ClientConcepts/LowLevel/DirectStreaming.cs

@@ -0,0 +1,83 @@
+using Elasticsearch.Net;
+using FluentAssertions;
+using Nest;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Tests.Framework;
+
+namespace Tests.ClientConcepts.LowLevel
+{
+	public class DirectStreaming
+	{
+		[U]
+		public void DisableDirectStreamingOnError()
+		{
+			Action<IResponse> assert = r =>
+			{
+				r.ApiCall.Should().NotBeNull();
+				r.ApiCall.RequestBodyInBytes.Should().NotBeNull();
+				r.ApiCall.ResponseBodyInBytes.Should().NotBeNull();
+			};
+
+			var client = TestClient.GetFixedReturnClient(new { }, 404, s => s.DisableDirectStreaming());
+			var response = client.Search<object>(s => s);
+			assert(response);
+			response = client.SearchAsync<object>(s => s).Result;
+			assert(response);
+		}
+
+		[U]
+		public void EnableDirectStreamingOnError()
+		{
+			Action<IResponse> assert = r =>
+			{
+				r.ApiCall.Should().NotBeNull();
+				r.ApiCall.RequestBodyInBytes.Should().BeNull();
+				r.ApiCall.ResponseBodyInBytes.Should().BeNull();
+			};
+
+			var client = TestClient.GetFixedReturnClient(new { }, 404);
+			var response = client.Search<object>(s => s);
+			assert(response);
+			response = client.SearchAsync<object>(s => s).Result;
+			assert(response);
+		}
+
+		[U]
+		public void DisableDirectStreamingOnSuccess()
+		{
+			Action<IResponse> assert = r =>
+			{
+				r.ApiCall.Should().NotBeNull();
+				r.ApiCall.RequestBodyInBytes.Should().NotBeNull();
+				r.ApiCall.ResponseBodyInBytes.Should().NotBeNull();
+			};
+
+			var client = TestClient.GetFixedReturnClient(new { }, 200, s => s.DisableDirectStreaming());
+			var response = client.Search<object>(s => s);
+			assert(response);
+			response = client.SearchAsync<object>(s => s).Result;
+			assert(response);
+		}
+
+		[U]
+		public void EnableDirectStreamingOnSuccess()
+		{
+			Action<IResponse> assert = r =>
+			{
+				r.ApiCall.Should().NotBeNull();
+				r.ApiCall.RequestBodyInBytes.Should().BeNull();
+				r.ApiCall.ResponseBodyInBytes.Should().BeNull();
+			};
+
+			var client = TestClient.GetFixedReturnClient(new { });
+			var response = client.Search<object>(s => s);
+			assert(response);
+			response = client.SearchAsync<object>(s => s).Result;
+			assert(response);
+		}
+	}
+}

src/Tests/Framework/TestClient.cs

@@ -76,7 +76,8 @@ public static IConnection CreateConnection(ConnectionSettings settings = null, b
 				? ((IConnection)new HttpConnection())
 				: new InMemoryConnection();
 
-		public static IElasticClient GetFixedReturnClient(object responseJson)
+		public static IElasticClient GetFixedReturnClient(
+			object responseJson, int statusCode = 200, Func<ConnectionSettings, ConnectionSettings> modifySettings = null)
 		{
 			var serializer = new JsonNetSerializer(new ConnectionSettings());
 			byte[] fixedResult;
@@ -85,9 +86,10 @@ public static IElasticClient GetFixedReturnClient(object responseJson)
 				serializer.Serialize(responseJson, ms);
 				fixedResult = ms.ToArray();
 			}
-			var connection = new InMemoryConnection(fixedResult);
+			var connection = new InMemoryConnection(fixedResult, statusCode);
 			var connectionPool = new SingleNodeConnectionPool(new Uri("http://localhost:9200"));
-			var settings = new ConnectionSettings(connectionPool, connection);
+			var defaultSettings = new ConnectionSettings(connectionPool, connection);
+			var settings = (modifySettings != null) ? modifySettings(defaultSettings) : defaultSettings;
 			return new ElasticClient(settings);
 		}