FbInt128SupportTests.cs

/*
 *    The contents of this file are subject to the Initial
 *    Developer's Public License Version 1.0 (the "License");
 *    you may not use this file except in compliance with the
 *    License. You may obtain a copy of the License at
 *    https://github.com/FirebirdSQL/NETProvider/raw/master/license.txt.
 *
 *    Software distributed under the License is distributed on
 *    an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, either
 *    express or implied. See the License for the specific
 *    language governing rights and limitations under the License.
 *
 *    All Rights Reserved.
 */

//$Authors = Jiri Cincura (jiri@cincura.net)

using System;
using System.Globalization;
using System.Numerics;
using System.Threading.Tasks;
using FirebirdSql.Data.TestsBase;
using NUnit.Framework;

namespace FirebirdSql.Data.FirebirdClient.Tests;

[TestFixtureSource(typeof(FbServerTypeTestFixtureSource), nameof(FbServerTypeTestFixtureSource.Default))]
[TestFixtureSource(typeof(FbServerTypeTestFixtureSource), nameof(FbServerTypeTestFixtureSource.Embedded))]
public class FbInt128SupportTests : FbTestsBase
{
	public FbInt128SupportTests(FbServerType serverType, bool compression, FbWireCrypt wireCrypt)
		: base(serverType, compression, wireCrypt, false)
	{ }

	[SetUp]
	public override async Task SetUp()
	{
		await base.SetUp();

		if (!EnsureServerVersionAtLeast(new Version(4, 0, 0, 0)))
			return;
	}

	static readonly BigInteger[] TestValues = new[]
	{
			BigInteger.Parse("0"),
			BigInteger.Parse("1"),
			BigInteger.Parse("-1"),
			BigInteger.Parse("6"),
			BigInteger.Parse("-6"),
			BigInteger.Parse("184467440737095516190874"),
			BigInteger.Parse("-184467440737095516190874"),
		};

	[TestCaseSource(nameof(TestValues))]
	public async Task ReadsValueCorrectly(BigInteger value)
	{
		await using (var cmd = Connection.CreateCommand())
		{
			var svalue = value.ToString(CultureInfo.InvariantCulture);
			cmd.CommandText = $"select cast({svalue} as int128) from rdb$database";
			var result = (BigInteger)await cmd.ExecuteScalarAsync();
			Assert.AreEqual(value, result);
		}
	}

	[Test]
	public Task CanReadAsByte() => CanReadAsTypeHelper<byte>(6, r => r.GetByte(0));
	[Test]
	public Task CanReadAsInt16() => CanReadAsTypeHelper<short>(6, r => r.GetInt16(0));
	[Test]
	public Task CanReadAsInt32() => CanReadAsTypeHelper<int>(6, r => r.GetInt32(0));
	[Test]
	public Task CanReadAsInt64() => CanReadAsTypeHelper<long>(6, r => r.GetInt64(0));

	[TestCaseSource(nameof(TestValues))]
	public async Task PassesValueCorrectly(BigInteger value)
	{
		await using (var cmd = Connection.CreateCommand())
		{
			cmd.CommandText = "select cast(@value as int128) from rdb$database";
			cmd.Parameters.AddWithValue("value", value);
			var result = (BigInteger)await cmd.ExecuteScalarAsync();
			Assert.AreEqual(value, result);
		}
	}

	[Test]
	public Task CanPassAsByte() => CanPassAsTypeHelper<byte>(6);
	[Test]
	public Task CanPassAsInt16() => CanPassAsTypeHelper<short>(6);
	[Test]
	public Task CanPassAsInt32() => CanPassAsTypeHelper<int>(6);
	[Test]
	public Task CanPassAsInt64() => CanPassAsTypeHelper<long>(6);

	[Test]
	public async Task ReadsValueNullCorrectly()
	{
		await using (var cmd = Connection.CreateCommand())
		{
			cmd.CommandText = "select cast(null as int128) from rdb$database";
			var result = (DBNull)await cmd.ExecuteScalarAsync();
			Assert.AreEqual(DBNull.Value, result);
		}
	}

	[Test]
	public async Task PassesValueNullCorrectly()
	{
		await using (var cmd = Connection.CreateCommand())
		{
			cmd.CommandText = "select cast(@value as int128) from rdb$database";
			cmd.Parameters.AddWithValue("value", DBNull.Value);
			var result = (DBNull)await cmd.ExecuteScalarAsync();
			Assert.AreEqual(DBNull.Value, result);
		}
	}

	[Test]
	public async Task SelectEmptyResultSet()
	{
		await using (var cmd = Connection.CreateCommand())
		{
			cmd.CommandText = "select cast(null as int128) from rdb$database where 0=1";
			await using (var reader = await cmd.ExecuteReaderAsync())
			{
				Assert.DoesNotThrowAsync(reader.ReadAsync);
			}
		}
	}

	[Test]
	public async Task SimpleSelectSchemaTableTest()
	{
		await using (var cmd = Connection.CreateCommand())
		{
			cmd.CommandText = "select cast(null as int128) from rdb$database";
			await using (var reader = await cmd.ExecuteReaderAsync())
			{
				var schema = await reader.GetSchemaTableAsync();
				Assert.AreEqual(typeof(BigInteger), schema.Rows[0].ItemArray[5]);
			}
		}
	}

	async Task CanReadAsTypeHelper<T>(T value, Func<FbDataReader, T> getter)
		where T : IFormattable
	{
		await using (var cmd = Connection.CreateCommand())
		{
			var svalue = value.ToString(null, CultureInfo.InvariantCulture);
			cmd.CommandText = $"select cast({svalue} as int128) from rdb$database";
			await using (var reader = await cmd.ExecuteReaderAsync())
			{
				await reader.ReadAsync();
				Assert.AreEqual(value, getter(reader));
			}
		}
	}

	async Task CanPassAsTypeHelper<T>(T value)
	{
		await using (var cmd = Connection.CreateCommand())
		{
			cmd.CommandText = "select cast(@value as int128) from rdb$database";
			cmd.Parameters.AddWithValue("value", value);
			Assert.DoesNotThrowAsync(cmd.ExecuteScalarAsync);
		}
	}
}