Update ComponentAsIdTest.cs

Author: ogorodnikovInfopulse

Src/NHibernate.Envers.Tests/NetSpecific/Integration/ComponentAsId/ComponentAsIdTest.cs

@@ -1,4 +1,5 @@
 using NUnit.Framework;
+using NHibernate.Envers.Query;
 
 namespace NHibernate.Envers.Tests.NetSpecific.Integration.ComponentAsId
 {
@@ -43,6 +44,44 @@ public void ComponentAsIdTestMethod()
 			});
 		}
 
+		[Test]
+		public void ComponentAsIdGetAudit()
+		{
+			var ent1 = new Entity1
+			{
+				Id = 1
+			};
+
+			Save(ent1);
+
+			var ent2 = new Entity2()
+			{
+				Id = 1
+			};
+
+			Save(ent2);
+
+			var udf = new SomeEntUDF
+			{
+				Id = new ComponentAsId
+				{
+					Key1 = ent1,
+					Key2 = ent2
+				},
+			};
+
+			Save(udf);
+
+
+			var history = Session.Auditer().CreateQuery()
+				.ForRevisionsOfEntity(typeof(SomeEntUDF), false, true)
+				.Add(AuditEntity.Property("Id.Key1.Id").Eq(ent1.Id))
+				.GetResultList();
+
+			Assert.AreEqual(1, history.Count);
+
+		}
+
 		void Save(object o)
 		{
 			using (var tran = Session.BeginTransaction())