NH-2823

Author: rjperes

src/NHibernate.Test/MappingByCode/ExplicitMappingTests/OptimisticLockModeTests.cs

@@ -0,0 +1,33 @@
+using NHibernate.Cfg.MappingSchema;
+using NHibernate.Mapping.ByCode;
+using NUnit.Framework;
+using SharpTestsEx;
+
+namespace NHibernate.Test.MappingByCode.ExplicitMappingTests
+{
+	[TestFixture]
+	public class OptimisticLockModeTests
+	{
+		private class MyClass
+		{
+			public int Id { get; set; }
+			public int Version { get; set; }
+		}
+
+		[Test]
+		public void OptimisticLockModeTest()
+		{
+			//NH-2823
+			var mapper = new ModelMapper();
+			mapper.Class<MyClass>(
+				map =>
+				{
+					map.Id(x => x.Id, idmap => { });
+					map.OptimisticLock(OptimisticLockMode.Dirty);
+				});
+
+			var mappings = mapper.CompileMappingForAllExplicitlyAddedEntities();
+			Assert.AreEqual(mappings.RootClasses[0].optimisticlock, HbmOptimisticLockMode.Dirty);
+		}
+	}
+}

src/NHibernate.Test/NHibernate.Test.csproj

@@ -588,6 +588,7 @@
     <Compile Include="MappingByCode\ExplicitMappingTests\PoidTests.cs" />
     <Compile Include="MappingByCode\ExplicitMappingTests\RootClassPropertiesSplitsTests.cs" />
     <Compile Include="MappingByCode\ExplicitMappingTests\SubclassPropertiesSplitsTests.cs" />
+    <Compile Include="MappingByCode\ExplicitMappingTests\OptimisticLockModeTests.cs" />
     <Compile Include="MappingByCode\ExplicitMappingTests\VersionTests.cs" />
     <Compile Include="MappingByCode\For.cs" />
     <Compile Include="MappingByCode\GeneratorTests.cs" />

src/NHibernate/Mapping/ByCode/IClassMapper.cs

@@ -27,6 +27,7 @@ public interface IClassAttributesMapper : IEntityAttributesMapper, IEntitySqlsMa
 		void Where(string whereClause);
 		void SchemaAction(SchemaAction action);
 		void Polymorphism(PolymorphismType type);
+		void OptimisticLock(OptimisticLockMode mode);
 	}
 
 	public interface IClassMapper : IClassAttributesMapper, IPropertyContainerMapper
@@ -67,6 +68,7 @@ public interface IClassAttributesMapper<TEntity> : IEntityAttributesMapper, IEnt
 		void Where(string whereClause);
 		void SchemaAction(SchemaAction action);
 		void Polymorphism(PolymorphismType type);
+		void OptimisticLock(OptimisticLockMode mode);
 	}
 
 	public interface IClassMapper<TEntity> : IClassAttributesMapper<TEntity>, IPropertyContainerMapper<TEntity> where TEntity : class

src/NHibernate/Mapping/ByCode/Impl/ClassMapper.cs

@@ -59,6 +59,10 @@ public Dictionary<string, IJoinMapper> JoinMappers
 		}
 
 		#region Implementation of IClassMapper
+		public void OptimisticLock(OptimisticLockMode mode)
+		{
+			classMapping.optimisticlock = (HbmOptimisticLockMode)Enum.Parse(typeof(OptimisticLockMode), mode.ToString());
+		}
 
 		public void Id(Action<IIdMapper> mapper)
 		{

src/NHibernate/Mapping/ByCode/Impl/CustomizersImpl/ClassCustomizer.cs

@@ -30,6 +30,11 @@ private Dictionary<string, IJoinMapper<TEntity>> JoinCustomizers
 
 		#region Implementation of IClassAttributesMapper<TEntity>
 
+		public void OptimisticLock(OptimisticLockMode mode)
+		{
+			CustomizersHolder.AddCustomizer(typeof(TEntity), (IClassMapper m) => m.OptimisticLock(mode));
+		}
+
 		public void Id<TProperty>(Expression<Func<TEntity, TProperty>> idProperty)
 		{
 			Id(idProperty, x => { });

src/NHibernate/Mapping/ByCode/OptimisticLockMode.cs

@@ -0,0 +1,18 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+namespace NHibernate.Mapping.ByCode
+{
+	public enum OptimisticLockMode
+	{
+		None,
+
+		Version,
+
+		Dirty,
+
+		All,
+	}
+}

src/NHibernate/NHibernate.csproj

@@ -353,6 +353,7 @@
     <Compile Include="Mapping\ByCode\Impl\CustomizersImpl\ManyToAnyCustomizer.cs" />
     <Compile Include="Mapping\ByCode\Impl\DynamicComponentMapper.cs" />
     <Compile Include="Mapping\ByCode\Impl\ManyToAnyMapper.cs" />
+    <Compile Include="Mapping\ByCode\OptimisticLockMode.cs" />
     <Compile Include="Mapping\ByCode\Import.cs" />
     <Compile Include="Mapping\ByCode\PolymorphismType.cs" />
     <Compile Include="Mapping\ByCode\PropertyGeneration.cs" />