FixtureByCode.cs

//------------------------------------------------------------------------------
// <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.Linq;
using NHibernate.Cfg;
using NHibernate.Cfg.MappingSchema;
using NHibernate.Linq;
using NHibernate.Mapping.ByCode;
using NUnit.Framework;

namespace NHibernate.Test.NHSpecificTest.GH3643
{
	using System.Threading.Tasks;
	using System.Threading;
	[TestFixture]
	public class FixtureByCodeAsync : TestCaseMappingByCode
	{
		protected override void Configure(Configuration configuration)
		{
			configuration.SetProperty(Environment.UseQueryCache, "true");
			configuration.SetProperty(Environment.GenerateStatistics, "true");
		}

		protected override HbmMapping GetMappings()
		{
			var mapper = new ModelMapper();

			mapper.Class<Entity>(
				rc =>
				{
					rc.Id(x => x.Id);
					rc.Bag(
						x => x.Children,
						m =>
						{
							m.Access(Accessor.Field);
							m.Key(k => k.Column("EntityId"));
						},
						r => r.OneToMany());
			
					rc.Cache(
						cm =>
						{
							cm.Include(CacheInclude.All);
							cm.Usage(CacheUsage.ReadWrite);
						});
				});

			mapper.Class<ChildEntity>(
				rc =>
				{
					rc.Id(x => x.Id);
					rc.Cache(
						cm =>
						{
							cm.Include(CacheInclude.All);
							cm.Usage(CacheUsage.ReadWrite);
						});
				});

			return mapper.CompileMappingForAllExplicitlyAddedEntities();
		}

		protected override void OnSetUp()
		{
			using var session = OpenSession();
			using var transaction = session.BeginTransaction();
			
			session.CreateSQLQuery(
				"INSERT INTO Entity (Id) VALUES (0)"
			).ExecuteUpdate();

			session.CreateSQLQuery(
				"INSERT INTO ChildEntity (Id, EntityId) VALUES (0, 0)"
			).ExecuteUpdate();

			session.CreateSQLQuery(
				"INSERT INTO ChildEntity (Id, EntityId) VALUES (1, 0)"
			).ExecuteUpdate();
			
			transaction.Commit();
		}

		protected override void OnTearDown()
		{
			using var session = OpenSession();
			using var transaction = session.BeginTransaction();

			session.CreateSQLQuery("DELETE FROM ChildEntity").ExecuteUpdate();
			session.CreateSQLQuery("DELETE FROM Entity").ExecuteUpdate();

			transaction.Commit();
		}
	
		[Test]
		public async Task LoadsEntityWithEnumIdAndChildrenUsingQueryCacheAsync()
		{
			await (LoadEntityWithQueryCacheAsync()); // warm up cache

			var entity = await (LoadEntityWithQueryCacheAsync());

			Assert.That(entity.Children.Count(), Is.EqualTo(2));
			
			Assert.That(Sfi.Statistics.QueryExecutionCount, Is.EqualTo(1), "Unexpected execution count");
			Assert.That(Sfi.Statistics.QueryCachePutCount, Is.EqualTo(1), "Unexpected cache put count");
			Assert.That(Sfi.Statistics.QueryCacheHitCount, Is.EqualTo(1), "Unexpected cache hit count");
		}
		
		private async Task<Entity> LoadEntityWithQueryCacheAsync(CancellationToken cancellationToken = default(CancellationToken))
		{
			using var session = OpenSession();
			using var transaction = session.BeginTransaction();
			var entity = (await (session
			          .Query<Entity>()
			          .FetchMany(x => x.Children)
			          .WithOptions(opt => opt.SetCacheable(true))
			          .ToListAsync(cancellationToken)))[0];
			
			await (transaction.CommitAsync(cancellationToken));
			return entity;
		}
	}
}