SerializerTestBase.cs

// Licensed to Elasticsearch B.V under one or more agreements.
// Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
// See the LICENSE file in the project root for more information.

using System.IO;
using System.Text.Json;
using System.Threading.Tasks;
using Tests.Domain.Extensions;
using VerifyXunit;
using Xunit.Abstractions;

namespace Tests.Serialization;

public abstract class VerifySerializerTestBase : VerifyBase
{
	protected static readonly Serializer _requestResponseSerializer;
	protected static readonly IElasticsearchClientSettings _settings;

	static VerifySerializerTestBase() 
	{
		var settings = new ElasticsearchClientSettings();
		settings.ApplyDomainSettings();

		var client = new ElasticsearchClient(settings);

		_requestResponseSerializer = client.RequestResponseSerializer;
		_settings = client.ElasticsearchClientSettings;
	}

	protected static Stream WrapInStream(string json)
	{
		var stream = new MemoryStream();
		var writer = new StreamWriter(stream);
		writer.Write(json);
		writer.Flush();
		stream.Position = 0;
		return stream;
	}

	protected static string SerializeAndGetJsonString<T>(T data)
	{
		var stream = new MemoryStream();
		_requestResponseSerializer.Serialize(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		return reader.ReadToEnd();
	}

	/// <summary>
	/// Serialises the <paramref name="data"/> using the sync and async request/response serializer methods, comparing the results.
	/// </summary>
	/// <returns>
	/// The JSON as a string for further comparisons and assertions.
	/// </returns>
	protected static async Task<string> SerializeAndGetJsonStringAsync<T>(T data)
	{
		var stream = new MemoryStream();
		await _requestResponseSerializer.SerializeAsync(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		var asyncJsonString = await reader.ReadToEndAsync();

		stream.SetLength(0);
		_requestResponseSerializer.Serialize(data, stream);
		stream.Position = 0;
		reader = new StreamReader(stream);
		var syncJsonString = await reader.ReadToEndAsync();

		syncJsonString.Should().Be(asyncJsonString);

		return asyncJsonString;
	}
}

public abstract class InstanceSerializerTestBase
{
	protected readonly Serializer _requestResponseSerializer;
	protected readonly Serializer _sourceSerializer;
	protected readonly IElasticsearchClientSettings _settings;

	protected InstanceSerializerTestBase(ElasticsearchClientSettings settings, bool applyDomainSettings = false)
	{
		if (applyDomainSettings)
			settings.ApplyDomainSettings();

		var client = new ElasticsearchClient(settings);

		_requestResponseSerializer = client.RequestResponseSerializer;
		_sourceSerializer = client.SourceSerializer;

		_settings = client.ElasticsearchClientSettings;
	}

	protected string SerializeAndGetJsonString<T>(T data)
	{
		var stream = new MemoryStream();
		_requestResponseSerializer.Serialize(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		return reader.ReadToEnd();
	}

	protected string SerializeAndGetJsonString<T>(T data, ElasticsearchClientSettings settings)
	{
		var client = new ElasticsearchClient(settings);
		var serializer = client.RequestResponseSerializer;
		var stream = new MemoryStream();
		serializer.Serialize(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		return reader.ReadToEnd();
	}
}

public abstract class SerializerTestBase
{
	protected static readonly Serializer _requestResponseSerializer;
	protected static readonly Serializer _sourceSerializer;
	protected static readonly IElasticsearchClientSettings _settings;

	static SerializerTestBase()
	{
		var settings = new ElasticsearchClientSettings();
		settings.ApplyDomainSettings();

		var client = new ElasticsearchClient(settings);
		
		_requestResponseSerializer = client.RequestResponseSerializer;
		_sourceSerializer = client.SourceSerializer;
		_settings = client.ElasticsearchClientSettings;
	}

	protected static Inferrer Inferrer => _settings.Inferrer;

	public static async Task<string> SerializeAndVerifyJsonAsync<T>(T data)
	{
		var serialisedJson = await SerializeAndGetJsonStringAsync(data);
		await Verifier.VerifyJson(serialisedJson);
		return serialisedJson;
	}

	public static async Task<T> RoundTripAndVerifyJsonAsync<T>(T data)
	{
		var serialisedJson = await SerializeAndGetJsonStringAsync(data);
		await Verifier.VerifyJson(serialisedJson);
		return DeserializeJsonString<T>(serialisedJson);
	}

	protected static Stream WrapInStream(string json)
	{
		var stream = new MemoryStream();
		var writer = new StreamWriter(stream);
		writer.Write(json);
		writer.Flush();
		stream.Position = 0;
		return stream;
	}

	protected static string SerializeAndGetJsonString<T>(T data)
	{
		var stream = new MemoryStream();
		_requestResponseSerializer.Serialize(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		return reader.ReadToEnd();
	}

	protected static string SourceSerializeAndGetJsonString<T>(T data)
	{
		var stream = new MemoryStream();
		_sourceSerializer.Serialize(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		return reader.ReadToEnd();
	}

	protected static T DeserializeJsonString<T>(string json)
	{
		var stream = WrapInStream(json);
		return _requestResponseSerializer.Deserialize<T>(stream);
	}

	protected static T SourceDeserializeJsonString<T>(string json)
	{
		var stream = WrapInStream(json);
		return _sourceSerializer.Deserialize<T>(stream);
	}

	/// <summary>
	/// Serialises the <paramref name="data"/> using the sync and async request/response serializer methods, comparing the results.
	/// </summary>
	/// <returns>
	/// The JSON as a string for further comparisons and assertions.
	/// </returns>
	protected static async Task<string> SerializeAndGetJsonStringAsync<T>(T data)
	{
		var stream = new MemoryStream();
		await _requestResponseSerializer.SerializeAsync(data, stream);
		stream.Position = 0;
		var reader = new StreamReader(stream);
		var asyncJsonString = await reader.ReadToEndAsync();

		stream.SetLength(0);
		_requestResponseSerializer.Serialize(data, stream);
		stream.Position = 0;
		reader = new StreamReader(stream);
		var syncJsonString = await reader.ReadToEndAsync();

		syncJsonString.Should().Be(asyncJsonString);

		return asyncJsonString;
	}
}