SecondLevelCacheTest.cs

using System.Data.Common;
using System.Collections;
using NHibernate.Cache;
using NHibernate.Cfg;
using NHibernate.Engine;
using NUnit.Framework;

namespace NHibernate.Test.SecondLevelCacheTests
{
	using Criterion;

	[TestFixture]
	public class SecondLevelCacheTest : TestCase
	{
		protected override string MappingsAssembly
		{
			get { return "NHibernate.Test"; }
		}

		protected override string[] Mappings
		{
			get { return new string[] { "SecondLevelCacheTest.Item.hbm.xml" }; }
		}

		protected override void Configure(Configuration configuration)
		{
			base.Configure(configuration);
			configuration.Properties[Environment.CacheProvider] = typeof(HashtableCacheProvider).AssemblyQualifiedName;
			configuration.Properties[Environment.UseQueryCache] = "true";
		}

		protected override void OnSetUp()
		{
			// Clear cache at each test.
			RebuildSessionFactory();
			using (ISession session = OpenSession())
			using (ITransaction tx = session.BeginTransaction())
			{
				Item item = new Item();
				item.Id = 1;
				session.Save(item);
				for (int i = 0; i < 4; i++)
				{
					Item child = new Item();
					child.Id = i + 2;
					child.Parent = item;
					session.Save(child);
					item.Children.Add(child);
				}

				for (int i = 0; i < 5; i++)
				{
					AnotherItem obj = new AnotherItem("Item #" + i);
					obj.Id = i + 1;
					session.Save(obj);
				}

				tx.Commit();
			}

			Sfi.Evict(typeof(Item));
			Sfi.EvictCollection(typeof(Item).FullName + ".Children");
		}

		protected override void OnTearDown()
		{
			using (ISession session = OpenSession())
			{
				session.Delete("from Item"); //cleaning up
				session.Delete("from AnotherItem"); //cleaning up
				session.Flush();
			}
		}

		[Test]
		public void CachedQueriesHandlesEntitiesParametersCorrectly()
		{
			using (ISession session = OpenSession())
			{
				Item one = (Item)session.Load(typeof(Item), 1);
				IList results = session.CreateQuery("from Item item where item.Parent = :parent")
					.SetEntity("parent", one)
					.SetCacheable(true).List();
				Assert.AreEqual(4, results.Count);
				foreach (Item item in results)
				{
					Assert.AreEqual(1, item.Parent.Id);
				}
			}

			using (ISession session = OpenSession())
			{
				Item two = (Item)session.Load(typeof(Item), 2);
				IList results = session.CreateQuery("from Item item where item.Parent = :parent")
					.SetEntity("parent", two)
					.SetCacheable(true).List();
				Assert.AreEqual(0, results.Count);
			}
		}

		[Test]
		public void DeleteItemFromCollectionThatIsInTheSecondLevelCache()
		{
			using (ISession session = OpenSession())
			{
				Item item = (Item)session.Load(typeof(Item), 1);
				Assert.IsTrue(item.Children.Count == 4); // just force it into the second level cache here
			}
			int childId = -1;
			using (ISession session = OpenSession())
			{
				Item item = (Item)session.Load(typeof(Item), 1);
				Item child = (Item)item.Children[0];
				childId = child.Id;
				session.Delete(child);
				item.Children.Remove(child);
				session.Flush();
			}

			using (ISession session = OpenSession())
			{
				Item item = (Item)session.Load(typeof(Item), 1);
				Assert.AreEqual(3, item.Children.Count);
				foreach (Item child in item.Children)
				{
					NHibernateUtil.Initialize(child);
					Assert.IsFalse(child.Id == childId);
				}
			}
		}

		[Test]
		public void InsertItemToCollectionOnTheSecondLevelCache()
		{
			using (ISession session = OpenSession())
			{
				Item item = (Item)session.Load(typeof(Item), 1);
				Item child = new Item();
				child.Id = 6;
				item.Children.Add(child);
				session.Save(child);
				session.Flush();
			}

			using (ISession session = OpenSession())
			{
				Item item = (Item)session.Load(typeof(Item), 1);
				int count = item.Children.Count;
				Assert.AreEqual(5, count);
			}
		}

		[Test]
		public void SecondLevelCacheWithCriteriaQueries()
		{
			using (ISession session = OpenSession())
			{
				IList list = session.CreateCriteria(typeof(AnotherItem))
					.Add(Expression.Gt("Id", 2))
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);

				using (var cmd = session.Connection.CreateCommand())
				{
					cmd.CommandText = "DELETE FROM AnotherItem";
					cmd.ExecuteNonQuery();
				}
			}

			using (ISession session = OpenSession())
			{
				//should bring from cache
				IList list = session.CreateCriteria(typeof(AnotherItem))
					.Add(Expression.Gt("Id", 2))
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);
			}
		}

		[Test]
		public void SecondLevelCacheWithCriteriaQueriesForItemWithCollections()
		{
			using (ISession session = OpenSession())
			{
				IList list = session.CreateCriteria(typeof(Item))
					.Add(Expression.Gt("Id", 2))
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);

				using (var cmd = session.Connection.CreateCommand())
				{
					cmd.CommandText = "DELETE FROM Item";
					cmd.ExecuteNonQuery();
				}
			}

			using (ISession session = OpenSession())
			{
				//should bring from cache
				IList list = session.CreateCriteria(typeof(Item))
					.Add(Expression.Gt("Id", 2))
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);
			}
		}

		[Test]
		public void SecondLevelCacheWithHqlQueriesForItemWithCollections()
		{
			using (ISession session = OpenSession())
			{
				IList list = session.CreateQuery("from Item i where i.Id > 2")
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);

				using (var cmd = session.Connection.CreateCommand())
				{
					cmd.CommandText = "DELETE FROM Item";
					cmd.ExecuteNonQuery();
				}
			}

			using (ISession session = OpenSession())
			{
				//should bring from cache
				IList list = session.CreateQuery("from Item i where i.Id > 2")
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);
			}
		}

		[Test]
		public void SecondLevelCacheWithHqlQueries()
		{
			using (ISession session = OpenSession())
			{
				IList list = session.CreateQuery("from AnotherItem i where i.Id > 2")
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);

				using (var cmd = session.Connection.CreateCommand())
				{
					cmd.CommandText = "DELETE FROM AnotherItem";
					cmd.ExecuteNonQuery();
				}
			}

			using (ISession session = OpenSession())
			{
				//should bring from cache
				IList list = session.CreateQuery("from AnotherItem i where i.Id > 2")
					.SetCacheable(true)
					.List();
				Assert.AreEqual(3, list.Count);
			}
		}
	}
}