Add Async tests for GitHub issue #2552

Author: David Ellingsworth

src/NHibernate.Test/Async/NHSpecificTest/GH2552/Fixture.cs

@@ -0,0 +1,194 @@
+//------------------------------------------------------------------------------
+// <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.Reflection;
+using NHibernate.Cache;
+using NHibernate.Cfg;
+using NHibernate.Test.CacheTest.Caches;
+using NUnit.Framework;
+
+namespace NHibernate.Test.NHSpecificTest.GH2552
+{
+	using System.Threading.Tasks;
+	using System.Threading;
+	[TestFixture]
+	public class FixtureAsync : BugTestCase
+	{
+		protected override string CacheConcurrencyStrategy => null;
+
+		protected override void Configure(Configuration configuration)
+		{
+			configuration.SetProperty(Environment.UseSecondLevelCache, "true");
+			configuration.SetProperty(Environment.UseQueryCache, "true");
+			configuration.SetProperty(Environment.GenerateStatistics, "true");
+			configuration.SetProperty(Environment.CacheProvider, typeof(BatchableCacheProvider).AssemblyQualifiedName);
+		}
+
+		protected override void OnSetUp()
+		{
+			// Initialize database with data
+			using (var s = Sfi.OpenSession())
+			using (var tx =s.BeginTransaction())
+			{
+				for (var i = 0; i < 6; i++)
+				{
+					var person = new PersonByFK()
+					{
+						Name = $"PersonByFK{i}",
+						Details = (i % 2 == 0)
+							? new DetailsByFK() { Data = $"DetailsByFK{i}" }
+							: null,
+					};
+
+					s.Save(person);
+				}
+
+				for (var i = 0; i < 6; i++)
+				{
+					var person = new PersonByRef()
+					{
+						Name = $"PersonByRef{i}",
+						Details = (i % 2 == 1)
+							? new DetailsByRef() { Data = $"DetailsByRef{i}" }
+							: null,
+					};
+
+					s.Save(person);
+				}
+
+				tx.Commit();
+			}
+		}
+
+		protected override void OnTearDown()
+		{
+			using (var s = OpenSession())
+			using (var tx = s.BeginTransaction())
+			{
+				s.CreateQuery("delete from DetailsByFK").ExecuteUpdate();
+				s.CreateQuery("delete from PersonByFK").ExecuteUpdate();
+				s.CreateQuery("delete from DetailsByRef").ExecuteUpdate();
+				s.CreateQuery("delete from PersonByRef").ExecuteUpdate();
+
+				tx.Commit();
+			}
+
+			Sfi.Evict(typeof(PersonByFK));
+			Sfi.Evict(typeof(DetailsByFK));
+			Sfi.Evict(typeof(PersonByRef));
+			Sfi.Evict(typeof(DetailsByRef));
+		}
+
+		private BatchableCache GetDefaultQueryCache()
+		{
+			var queryCache = Sfi.GetQueryCache(null);
+			var field = typeof(StandardQueryCache).GetField(
+				"_cache",
+				BindingFlags.NonPublic | BindingFlags.Instance);
+			Assert.That(field, Is.Not.Null, "Unable to find _cache field");
+			var cache = (BatchableCache) field.GetValue(queryCache);
+			Assert.That(cache, Is.Not.Null, "_cache is null");
+
+			return cache;
+		}
+
+		private async Task OneToOneTestAsync<TPerson, TDetails>(CancellationToken cancellationToken = default(CancellationToken)) where TPerson:Person where TDetails : Details
+		{
+			var personPersister = Sfi.GetEntityPersister(typeof(TPerson).FullName);
+			var detailsPersister = Sfi.GetEntityPersister(typeof(TDetails).FullName);
+
+			var personCache = personPersister.Cache.Cache as BatchableCache;
+			var detailsCache = detailsPersister.Cache.Cache as BatchableCache;
+			var queryCache = GetDefaultQueryCache();
+
+			Assert.That(personCache, Is.Not.Null);
+			Assert.That(detailsCache, Is.Not.Null);
+
+			await (queryCache.ClearAsync(cancellationToken));
+			await (personCache.ClearAsync(cancellationToken));
+			await (detailsCache.ClearAsync(cancellationToken));
+
+			queryCache.ClearStatistics();
+			personCache.ClearStatistics();
+			detailsCache.ClearStatistics();
+
+			// Fill the empty caches with data.
+			using (var s = OpenSession())
+			using (var tx = s.BeginTransaction())
+			{
+				await (s.QueryOver<TPerson>()
+					.Cacheable()
+					.ListAsync(cancellationToken));
+
+				await (tx.CommitAsync(cancellationToken));
+			}
+
+			// Verify query cache statistics
+			Assert.That(queryCache.PutCalls, Has.Count.EqualTo(1));
+			Assert.That(queryCache.GetCalls, Has.Count.EqualTo(1));
+
+			// Verify person cache statistics
+			Assert.That(personCache.GetCalls, Has.Count.EqualTo(0));
+			Assert.That(personCache.PutCalls, Has.Count.EqualTo(0));
+			Assert.That(personCache.GetMultipleCalls, Has.Count.EqualTo(0));
+			Assert.That(personCache.PutMultipleCalls, Has.Count.EqualTo(1));
+
+			// Verify details cache statistics
+			Assert.That(detailsCache.GetCalls, Has.Count.EqualTo(0));
+			Assert.That(detailsCache.PutCalls, Has.Count.EqualTo(3));
+			Assert.That(detailsCache.GetMultipleCalls, Has.Count.EqualTo(0));
+			Assert.That(detailsCache.PutMultipleCalls, Has.Count.EqualTo(0));
+
+			Sfi.Statistics.Clear();
+
+			using (var s = OpenSession())
+			using (var tx = s.BeginTransaction())
+			{
+				await (s.QueryOver<TPerson>()
+					.Cacheable()
+					.ListAsync(cancellationToken));
+
+				await (tx.CommitAsync(cancellationToken));
+			}
+
+			// Verify query cache statistics
+			Assert.That(queryCache.GetCalls, Has.Count.EqualTo(2));
+			Assert.That(queryCache.PutCalls, Has.Count.EqualTo(1));
+
+			// Verify person cache statistics
+			Assert.That(personCache.GetCalls, Has.Count.EqualTo(0));
+			Assert.That(personCache.PutCalls, Has.Count.EqualTo(0));
+			Assert.That(personCache.GetMultipleCalls, Has.Count.EqualTo(1));
+			Assert.That(personCache.PutMultipleCalls, Has.Count.EqualTo(1));
+
+			// Verify details cache statistics
+			Assert.That(detailsCache.GetCalls, Has.Count.EqualTo(0));
+			Assert.That(detailsCache.PutCalls, Has.Count.EqualTo(3));
+			Assert.That(detailsCache.GetMultipleCalls, Has.Count.EqualTo(1));
+			Assert.That(detailsCache.PutMultipleCalls, Has.Count.EqualTo(0));
+
+			// Verify that no second level cache misses occured.
+			Assert.That(Sfi.Statistics.QueryCacheMissCount, Is.EqualTo(0));
+			Assert.That(Sfi.Statistics.SecondLevelCacheMissCount, Is.EqualTo(0));
+		}
+
+		[Test]
+		public async Task OneToOneCacheByForeignKeyAsync()
+		{
+			await (OneToOneTestAsync<PersonByFK, DetailsByFK>());
+		}
+
+		[Test]
+		public async Task OneToOneCacheByRefAsync()
+		{
+			await (OneToOneTestAsync<PersonByRef, DetailsByRef>());
+		}
+	}
+}