VersionTests.cs

using NHibernate.Mapping.ByCode;
using NUnit.Framework;

namespace NHibernate.Test.MappingByCode.ExpliticMappingTests
{
	[TestFixture]
	public class VersionTests
	{
		private class MyClass
		{
			public int Id { get; set; }
			public int Version { get; set; }
		}
		private class MyRoot: MyClass
		{
		}

		[Test]
		public void WhenPropertyUsedAsVersionThenRegister()
		{
			var inspector = new ExplicitlyDeclaredModel();
			var mapper = new ModelMapper(inspector);
			mapper.Class<MyClass>(
				map =>
				{
					map.Id(x => x.Id, idmap => { });
					map.Version(x => x.Version, vm => { });
				});

			Assert.That(inspector.IsVersion(For<MyClass>.Property(x => x.Version)), Is.True);
		}

		[Test]
		public void WhenPropertyVersionFromBaseEntityThenFindItAsVersion()
		{
			var inspector = new ExplicitlyDeclaredModel();
			var mapper = new ModelMapper(inspector);
			mapper.Class<MyClass>(
				map =>
				{
					map.Id(x => x.Id, idmap => { });
					map.Version(x => x.Version, vm => { });
				});

			Assert.That(inspector.IsVersion(For<MyRoot>.Property(x => x.Version)), Is.True);
		}
	}
}