NH-3919 - More date types tests, and refactoring of them.

Author: fredericDelaporte

src/NHibernate.Test/Async/TypesTest/AbstractDateTimeTypeFixture.cs

@@ -0,0 +1,385 @@
+//------------------------------------------------------------------------------
+// <auto-generated>
+//     This code was generated by AsyncGenerator.
+//
+//     Changes to this file may cause incorrect behavior and will be lost if
+//     the code is regenerated.
+// </auto-generated>
+//------------------------------------------------------------------------------
+
+
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Data.Common;
+using System.Linq;
+using System.Reflection;
+using NHibernate.Cfg;
+using NHibernate.Driver;
+using NHibernate.Engine;
+using NHibernate.SqlCommand;
+using NHibernate.SqlTypes;
+using NHibernate.Tool.hbm2ddl;
+using NHibernate.Type;
+using NHibernate.Util;
+using NUnit.Framework;
+
+namespace NHibernate.Test.TypesTest
+{
+	using System.Threading.Tasks;
+	using System.Threading;
+	[TestFixture]
+	public abstract class AbstractDateTimeTypeFixtureAsync : TypeFixtureBase
+	{
+		protected abstract AbstractDateTimeType Type { get; }
+		protected virtual bool RevisionCheck => true;
+
+		protected const int DateId = 1;
+		protected const int AdditionalDateId = 2;
+
+		protected override void Configure(Configuration configuration)
+		{
+			base.Configure(configuration);
+
+			var driverClass = ReflectHelper.ClassForName(configuration.GetProperty(Cfg.Environment.ConnectionDriver));
+			ClientDriverWithParamsStats.DriverClass = driverClass;
+
+			configuration.SetProperty(
+				Cfg.Environment.ConnectionDriver,
+				typeof(ClientDriverWithParamsStats).AssemblyQualifiedName);
+		}
+
+		protected override void OnSetUp()
+		{
+			base.OnSetUp();
+
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				var d = new DateTimeClass
+				{
+					Id = DateId,
+					Value = Now
+				};
+				s.Save(d);
+				t.Commit();
+			}
+		}
+
+		protected override void OnTearDown()
+		{
+			base.OnTearDown();
+
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				s.CreateQuery("delete from DateTimeClass").ExecuteUpdate();
+				t.Commit();
+			}
+		}
+
+		protected override void DropSchema()
+		{
+			(Sfi.ConnectionProvider.Driver as ClientDriverWithParamsStats)?.CleanUp();
+			base.DropSchema();
+		}
+
+		[Test]
+		public async Task NextAsync()
+		{
+			var current = DateTime.Parse("2004-01-01");
+			var next = await (Type.NextAsync(current, null, CancellationToken.None));
+
+			Assert.That(next, Is.TypeOf<DateTime>(), "next should be DateTime");
+			Assert.That(next, Is.GreaterThan(current), "next should be greater than current");
+		}
+
+		[Test]
+		public async Task SeedAsync()
+		{
+			Assert.That(await (Type.SeedAsync(null, CancellationToken.None)), Is.TypeOf<DateTime>(), "seed should be DateTime");
+		}
+
+		[Test]
+		[TestCase(DateTimeKind.Unspecified)]
+		[TestCase(DateTimeKind.Local)]
+		[TestCase(DateTimeKind.Utc)]
+		public virtual async Task ReadWriteAsync(DateTimeKind kind)
+		{
+			var entity = new DateTimeClass
+			{
+				Id = AdditionalDateId,
+				Value = GetTestDate(kind)
+			};
+
+			DateTime beforeNow, afterNow;
+
+			// Save
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				// Account db accuracy
+				beforeNow = Now.AddTicks(-DateAccuracyInTicks);
+				await (s.SaveAsync(entity));
+				await (t.CommitAsync());
+				afterNow = Now.AddTicks(DateAccuracyInTicks);
+			}
+
+			var typeKind = GetTypeKind();
+			if (RevisionCheck)
+			{
+				Assert.That(entity.Revision, Is.GreaterThan(beforeNow).And.LessThan(afterNow), "Revision not correctly seeded.");
+				if (typeKind != DateTimeKind.Unspecified)
+					Assert.That(entity.Revision.Kind, Is.EqualTo(typeKind), "Revision kind not correctly seeded.");
+				Assert.That(entity.NullableValue, Is.Null, "NullableValue unexpectedly seeded.");
+			}
+
+			// Retrieve, compare then update
+			DateTimeClass retrieved;
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				retrieved = await (s.GetAsync<DateTimeClass>(AdditionalDateId));
+
+				Assert.That(retrieved, Is.Not.Null, "Entity not saved or cannot be retrieved by its key.");
+				Assert.That(retrieved.Value, Is.EqualTo(entity.Value), "Value should be the same.");
+				if (RevisionCheck)
+					Assert.That(retrieved.Revision, Is.EqualTo(entity.Revision), "Revision should be the same.");
+				Assert.That(retrieved.NullableValue, Is.EqualTo(entity.NullableValue), "NullableValue should be the same.");
+				if (typeKind != DateTimeKind.Unspecified)
+				{
+					Assert.That(retrieved.Value.Kind, Is.EqualTo(typeKind), "Value kind not correctly retrieved.");
+					if (RevisionCheck)
+						Assert.That(retrieved.Revision.Kind, Is.EqualTo(typeKind), "Revision kind not correctly retrieved.");
+				}
+
+				retrieved.NullableValue = GetTestDate(kind);
+				retrieved.Value = GetTestDate(kind).AddMonths(-1);
+
+				beforeNow = Now.AddTicks(-DateAccuracyInTicks);
+				await (t.CommitAsync());
+				afterNow = Now.AddTicks(DateAccuracyInTicks);
+			}
+
+			if (RevisionCheck)
+			{
+				Assert.That(
+					retrieved.Revision,
+					Is.GreaterThan(beforeNow).And.LessThan(afterNow).And.GreaterThanOrEqualTo(entity.Revision),
+					"Revision not correctly incremented.");
+				if (typeKind != DateTimeKind.Unspecified)
+					Assert.That(retrieved.Revision.Kind, Is.EqualTo(typeKind), "Revision kind incorrectly changed.");
+			}
+
+			// Retrieve and compare again
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				var retrievedAgain = await (s.GetAsync<DateTimeClass>(AdditionalDateId));
+
+				Assert.That(retrievedAgain, Is.Not.Null, "Entity deleted or cannot be retrieved again by its key.");
+				Assert.That(retrievedAgain.Value, Is.EqualTo(retrieved.Value), "Value should be the same again.");
+				if (RevisionCheck)
+					Assert.That(retrievedAgain.Revision, Is.EqualTo(retrieved.Revision), "Revision should be the same again.");
+				Assert.That(
+					retrievedAgain.NullableValue,
+					Is.EqualTo(retrieved.NullableValue),
+					"NullableValue should be the same again.");
+				if (typeKind != DateTimeKind.Unspecified)
+				{
+					Assert.That(retrievedAgain.Value.Kind, Is.EqualTo(typeKind), "Value kind not correctly retrieved again.");
+					if (RevisionCheck)
+						Assert.That(retrievedAgain.Revision.Kind, Is.EqualTo(typeKind), "Revision kind not correctly retrieved again.");
+					Assert.That(
+						retrievedAgain.NullableValue.Value.Kind,
+						Is.EqualTo(typeKind),
+						"NullableValue kind not correctly retrieved again.");
+				}
+				await (t.CommitAsync());
+			}
+		}
+
+		[Test]
+		public Task DbHasExpectedTypeAsync()
+		{
+			try
+			{
+				var validator = new SchemaValidator(cfg);
+				return validator.ValidateAsync();
+			}
+			catch (Exception ex)
+			{
+				return Task.FromException<object>(ex);
+			}
+		}
+
+		[Test]
+		public async Task SaveUseExpectedSqlTypeAsync()
+		{
+			var driver = (ClientDriverWithParamsStats) Sfi.ConnectionProvider.Driver;
+
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				var d = new DateTimeClass
+				{
+					Id = 2,
+					Value = Now,
+					NullableValue = Now
+				};
+				driver.ClearStats();
+				await (s.SaveAsync(d));
+				await (t.CommitAsync());
+			}
+
+			// 2 properties + revision
+			AssertSqlType(driver, 3);
+		}
+
+		[Test]
+		public async Task UpdateUseExpectedSqlTypeAsync()
+		{
+			var driver = (ClientDriverWithParamsStats) Sfi.ConnectionProvider.Driver;
+
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				var d = await (s.GetAsync<DateTimeClass>(DateId));
+				d.Value = Now;
+				d.NullableValue = Now;
+				driver.ClearStats();
+				await (t.CommitAsync());
+			}
+
+			// 2 properties + revision x 2 (check + update)
+			AssertSqlType(driver, 4);
+		}
+
+		[Test]
+		public virtual async Task QueryUseExpectedSqlTypeAsync()
+		{
+			if (!TestDialect.SupportsNonDataBoundCondition)
+				Assert.Ignore("Dialect does not support the test query");
+
+			var driver = (ClientDriverWithParamsStats) Sfi.ConnectionProvider.Driver;
+
+			using (var s = OpenSession())
+			using (var t = s.BeginTransaction())
+			{
+				var q = s
+					.CreateQuery(
+						"from DateTimeClass d where d.Value = :value and " +
+						"d.NullableValue = :nullableValue and " +
+						"d.Revision = :revision and " +
+						":other1 = :other2")
+					.SetDateTime("value", Now)
+					.SetDateTime("nullableValue", Now)
+					.SetDateTime("revision", Now)
+					.SetDateTime("other1", Now)
+					.SetDateTime("other2", Now);
+				driver.ClearStats();
+				await (q.ListAsync<DateTimeClass>());
+				await (t.CommitAsync());
+			}
+
+			AssertSqlType(driver, 5);
+		}
+
+		private void AssertSqlType(ClientDriverWithParamsStats driver, int expectedCount)
+		{
+			if (Type.SqlTypes(Sfi).Any(t => Equals(t, SqlTypeFactory.DateTime2)))
+			{
+				Assert.That(
+					driver.GetCount(SqlTypeFactory.DateTime),
+					Is.EqualTo(0),
+					"Found unexpected SqlTypeFactory.DateTime usages.");
+				Assert.That(
+					driver.GetCount(SqlTypeFactory.DateTime2),
+					Is.EqualTo(expectedCount),
+					"Unexpected SqlTypeFactory.DateTime2 usage count.");
+				Assert.That(driver.GetCount(DbType.DateTime), Is.EqualTo(0), "Found unexpected DbType.DateTime usages.");
+				Assert.That(
+					driver.GetCount(DbType.DateTime2),
+					Is.EqualTo(expectedCount),
+					"Unexpected DbType.DateTime2 usage count.");
+			}
+			else if (Type.SqlTypes(Sfi).Any(t => Equals(t, SqlTypeFactory.DateTime)))
+			{
+				Assert.That(
+					driver.GetCount(SqlTypeFactory.DateTime2),
+					Is.EqualTo(0),
+					"Found unexpected SqlTypeFactory.DateTime2 usages.");
+				Assert.That(
+					driver.GetCount(SqlTypeFactory.DateTime),
+					Is.EqualTo(expectedCount),
+					"Unexpected SqlTypeFactory.DateTime usage count.");
+				Assert.That(driver.GetCount(DbType.DateTime2), Is.EqualTo(0), "Found unexpected DbType.DateTime2 usages.");
+				Assert.That(driver.GetCount(DbType.DateTime), Is.EqualTo(expectedCount), "Unexpected DbType.DateTime usage count.");
+			}
+			else if (Type.SqlTypes(Sfi).Any(t => Equals(t, SqlTypeFactory.Date)))
+			{
+				Assert.That(
+					driver.GetCount(SqlTypeFactory.DateTime),
+					Is.EqualTo(0),
+					"Found unexpected SqlTypeFactory.DateTime usages.");
+				Assert.That(
+					driver.GetCount(SqlTypeFactory.Date),
+					Is.EqualTo(expectedCount),
+					"Unexpected SqlTypeFactory.Date usage count.");
+				Assert.That(driver.GetCount(DbType.DateTime), Is.EqualTo(0), "Found unexpected DbType.DateTime usages.");
+				Assert.That(driver.GetCount(DbType.Date), Is.EqualTo(expectedCount), "Unexpected DbType.Date usage count.");
+			}
+		}
+
+		protected virtual long DateAccuracyInTicks => Dialect.TimestampResolutionInTicks;
+
+		protected DateTime Now => GetTypeKind() == DateTimeKind.Utc ? DateTime.UtcNow : DateTime.Now;
+
+		protected virtual DateTime GetTestDate(DateTimeKind kind)
+		{
+			return RoundForDialect(
+					kind == DateTimeKind.Utc ? DateTime.UtcNow : DateTime.SpecifyKind(DateTime.Now, kind))
+				// Take another date than now for checking the value do not get overridden by seeding.
+				.AddDays(1);
+		}
+
+		/// <summary>
+		/// Return a date time still considered equal but as different as possible.
+		/// </summary>
+		/// <param name="original">The originale date time.</param>
+		/// <returns>An equal date time.</returns>
+		protected virtual DateTime GetSameDate(DateTime original)
+		{
+			if (GetTypeKind() != DateTimeKind.Unspecified)
+				return new DateTime(original.Ticks, original.Kind);
+
+			switch (original.Kind)
+			{
+				case DateTimeKind.Local:
+					return DateTime.SpecifyKind(original, DateTimeKind.Unspecified);
+				case DateTimeKind.Unspecified:
+					return DateTime.SpecifyKind(original, DateTimeKind.Utc);
+				default:
+					return DateTime.SpecifyKind(original, DateTimeKind.Local);
+			}
+		}
+
+		/// <summary>
+		/// Return a different date time but as few different as possible.
+		/// </summary>
+		/// <param name="original">The originale date time.</param>
+		/// <returns>An inequal date time.</returns>
+		protected virtual DateTime GetDifferentDate(DateTime original)
+		{
+			return original.AddTicks(DateAccuracyInTicks);
+		}
+
+		private static readonly PropertyInfo _kindProperty =
+			typeof(AbstractDateTimeType).GetProperty("Kind", BindingFlags.Instance | BindingFlags.NonPublic);
+
+		protected DateTimeKind GetTypeKind()
+		{
+			return (DateTimeKind) _kindProperty.GetValue(Type);
+		}
+	}
+}