* Fix inherited classes go to same collection as baseclass (https://mongorepository.codeplex.com/discussions/572382)

Author: RobThree

MongoRepository/dev/MongoRepository/Util.cs

@@ -153,7 +153,7 @@ private static string GetCollectionNameFromType(Type entitytype)
             }
             else
             {
-                if (entitytype.Equals(typeof(Entity)))
+                if (typeof(Entity).IsAssignableFrom(entitytype))
                 {
                     // No attribute found, get the basetype
                     while (!entitytype.BaseType.Equals(typeof(Entity)))

MongoRepository/dev/MongoRepositoryTests/RepoTests.cs

@@ -338,13 +338,45 @@ public abstract class BaseA : BaseItem
 
         public class SpecialA : BaseA
         { }
-        
+
         [TestMethod]
         public void Discussion433878()
         {
             var specialRepository = new MongoRepository<SpecialA>();
         }
         #endregion
 
+        #region Reproduce issue: https://mongorepository.codeplex.com/discussions/572382
+        public abstract class ClassA : Entity
+        {
+            public string Prop1 { get; set; }
+        }
+
+        public class ClassB : ClassA
+        {
+            public string Prop2 { get; set; }
+        }
+
+        public class ClassC : ClassA
+        {
+            public string Prop3 { get; set; }
+        }
+
+        [TestMethod]
+        public void Discussion572382()
+        {
+            var repo = new MongoRepository<ClassA>() { 
+                new ClassB() { Prop1 = "A", Prop2 = "B" } ,
+                new ClassC() { Prop1 = "A", Prop3 = "C" }
+            };
+
+            Assert.AreEqual(2, repo.Count());
+
+            Assert.AreEqual(2, repo.OfType<ClassA>().Count());
+            Assert.AreEqual(1, repo.OfType<ClassB>().Count());
+            Assert.AreEqual(1, repo.OfType<ClassC>().Count());
+        }
+        #endregion
+
     }
 }