KeyEntityPropertyRefFixture.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.Linq;
using NUnit.Framework;

namespace NHibernate.Test.PropertyRef
{
	using System.Threading.Tasks;
	[TestFixture]
	public class KeyEntityPropertyRefFixtureAsync : TestCase
	{
		protected override string[] Mappings => new string[] { "PropertyRef.KeyEntityPropertyRef.hbm.xml" };

		protected override string MappingsAssembly => "NHibernate.Test";

		private int _aWithItemsId;

		protected override void OnSetUp()
		{
			using (var s = OpenSession())
			using (var t = s.BeginTransaction())
			{
				var c1 = new C { Name = "Third" };
				s.Save(c1);
				var c2 = new C { Name = "ThirdBis" };
				s.Save(c2);

				var b1 = new B { Id = 1, Name = "SecondC1" };
				var b2 = new B { Id = 2, Name = "SecondC2" };
				s.Save(b1);
				s.Save(b2);
				var a = new A { Id = 3, Name = "First", C = c1 };
				a.CItems.Add(b1);
				a.CItems.Add(b2);
				_aWithItemsId = (int) s.Save(a);

				s.Save(new A { Id = 4, Name = "FirstBis", C = c2 });

				t.Commit();
			}
		}

		protected override void OnTearDown()
		{
			using (var s = OpenSession())
			using (var t = s.BeginTransaction())
			{
				s.CreateQuery("delete from B").ExecuteUpdate();
				s.CreateQuery("delete from A").ExecuteUpdate();
				s.CreateQuery("delete from C").ExecuteUpdate();
				t.Commit();
			}
		}

		[Test]
		public async Task PropertyRefLazyLoadAsync()
		{
			using (var s = OpenSession())
			using (var t = s.BeginTransaction())
			{
				var a = await (s.GetAsync<A>(_aWithItemsId));

				Assert.That(NHibernateUtil.IsInitialized(a.CItems), Is.False, "a with items");
				Assert.That(a.CItems, Has.Count.EqualTo(2), "a with items");
				await (t.CommitAsync());
			}

			using (var s = OpenSession())
			using (var t = s.BeginTransaction())
			{
				var entity =
					await (s
						.Query<A>()
						.SingleAsync(a => a.Id != _aWithItemsId));

				Assert.That(NHibernateUtil.IsInitialized(entity.CItems), Is.False, "a without items");
				Assert.That(entity.CItems, Has.Count.EqualTo(0), "a without items");
				await (t.CommitAsync());
			}
		}

		[Test]
		public async Task PropertyRefEagerLoadAsync()
		{
			using (var s = OpenSession())
			using (var t = s.BeginTransaction())
			{
				var entity =
					await (s
						.Query<A>()
						.Where(a => a.Id == _aWithItemsId)
						.FetchMany(a => a.CItems)
						.SingleAsync());

				Assert.That(NHibernateUtil.IsInitialized(entity.CItems), Is.True, "a with items");
				Assert.That(entity.CItems, Has.Count.EqualTo(2), "a with items");
				await (t.CommitAsync());
			}

			using (var s = OpenSession())
			using (var t = s.BeginTransaction())
			{
				var entity =
					await (s
						.Query<A>()
						.Where(a => a.Id != _aWithItemsId)
						.FetchMany(a => a.CItems)
						.SingleAsync());

				Assert.That(NHibernateUtil.IsInitialized(entity.CItems), Is.True, "a without items");
				Assert.That(entity.CItems, Has.Count.EqualTo(0), "a without items");
				await (t.CommitAsync());
			}
		}
	}
}