/source/MongoDB.Tests/IntegrationTests/Inheritance/TestInheritanceWithConcreteBaseClass.cs

using System;

using System.Collections.Generic;

using System.Linq;

using System.Text;

using NUnit.Framework;



namespace MongoDB.IntegrationTests.Inheritance

{

    [TestFixture]

    public class TestInheritanceWithConcreteBaseClass : MongoTestBase

    {

        class Animal

        {

            public Oid Id { get; set; }



            public int Age { get; set; }



            public string Name { get; set; }

        }



        class Bear : Animal

        { }



        abstract class Cat : Animal

        { }



        class Tiger : Cat

        { }



        class Lion : Cat

        { }



        public override string TestCollections

        {

            get { return "Animal"; }

        }



        [SetUp]

        public void TestSetup()

        {

            CleanDB();

        }



        protected override Configuration.MongoConfigurationBuilder GetConfiguration()

        {

            var builder = base.GetConfiguration();

            builder.Mapping(mapping =>

            {

                mapping.DefaultProfile(profile =>

                {

                    profile.SubClassesAre(x => x.IsSubclassOf(typeof(Animal)));

                });



                mapping.Map<Bear>();

                mapping.Map<Cat>();

                mapping.Map<Tiger>();

                mapping.Map<Lion>();

            });



            return builder;

        }



        [Test]

        public void Should_persist_discriminator_using_base_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var docCollection = DB.GetCollection<Document>("Animal");



            var docs = docCollection.FindAll().Sort("Age", IndexOrder.Ascending).Documents.ToList();



            Assert.AreEqual(new[] { "Cat", "Tiger" }, (List<string>)docs[0]["_t"]);

            Assert.IsNull(docs[1]["_t"]);

        }



        [Test]

        public void Should_persist_discriminator_using_inherited_class_collection()

        {

            var animalCollection = DB.GetCollection<Cat>();

            animalCollection.Save(new Lion() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var docCollection = DB.GetCollection<Document>("Animal");



            var docs = docCollection.FindAll().Sort("Age", IndexOrder.Ascending).Documents.ToList();



            Assert.AreEqual(new[] { "Cat", "Tiger" }, (List<string>)docs[0]["_t"]);

            Assert.AreEqual(new[] { "Cat", "Lion" }, (List<string>)docs[1]["_t"]);

        }



        [Test]

        public void Should_fetch_with_base_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var animals = animalCollection.FindAll().Sort("Age", IndexOrder.Ascending).Documents.ToList();



            Assert.AreEqual(2, animals.Count);

            Assert.IsInstanceOfType(typeof(Tiger), animals[0]);

            Assert.AreEqual(19, animals[0].Age);

            Assert.IsInstanceOfType(typeof(Animal), animals[1]);

            Assert.AreEqual(20, animals[1].Age);

        }



        [Test]

        public void Should_fetch_with_base_class_collection_through_linq()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var animals = (from a in animalCollection.Linq()

                           orderby a.Age ascending

                           select a).ToList();



            Assert.AreEqual(2, animals.Count);

            Assert.IsInstanceOfType(typeof(Tiger), animals[0]);

            Assert.AreEqual(19, animals[0].Age);

            Assert.IsInstanceOfType(typeof(Animal), animals[1]);

            Assert.AreEqual(20, animals[1].Age);

        }



        [Test]

        public void Should_fetch_with_inherited_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var catCollection = DB.GetCollection<Cat>();



            var cats = catCollection.FindAll().Sort("Age", IndexOrder.Ascending).Documents.ToList();



            Assert.AreEqual(1, cats.Count);

            Assert.IsInstanceOfType(typeof(Tiger), cats[0]);

            Assert.AreEqual(19, cats[0].Age);

        }



        [Test]

        public void Should_fetch_with_inherited_class_collection_through_linq()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var catCollection = DB.GetCollection<Cat>();



            var animals = (from a in catCollection.Linq()

                           orderby a.Age ascending

                           select a).ToList();



            Assert.AreEqual(1, animals.Count);

            Assert.IsInstanceOfType(typeof(Tiger), animals[0]);

            Assert.AreEqual(19, animals[0].Age);

        }



        [Test]

        public void Should_support_projections_with_base_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20, Name = "Jim" });

            animalCollection.Save(new Tiger() { Age = 19, Name = "Bob" });



            var animals = animalCollection.FindAll().Fields(new { Age = true }).Sort("Age", IndexOrder.Ascending).Documents.ToList();



            Assert.AreEqual(2, animals.Count);

            Assert.IsInstanceOfType(typeof(Tiger), animals[0]);

            Assert.AreEqual(19, animals[0].Age);

            Assert.IsNull(animals[0].Name);

            Assert.IsInstanceOfType(typeof(Animal), animals[1]);

            Assert.AreEqual(20, animals[1].Age);

            Assert.IsNull(animals[1].Name);

        }



        [Test]

        public void Should_support_projections_with_base_class_collections_with_linq()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20, Name = "Jim" });

            animalCollection.Save(new Tiger() { Age = 19, Name = "Bob" });



            var animals = (from a in animalCollection.Linq()

                           orderby a.Age ascending

                           select new { a.Name, a.Age }).ToList();



            Assert.AreEqual(2, animals.Count);

            Assert.AreEqual(19, animals[0].Age);

            Assert.AreEqual("Bob", animals[0].Name);

            Assert.AreEqual(20, animals[1].Age);

            Assert.AreEqual("Jim", animals[1].Name);

        }



        [Test]

        public void Should_support_projections_with_inherited_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20, Name = "Jim" });

            animalCollection.Save(new Tiger() { Age = 19, Name = "Bob" });



            var catCollection = DB.GetCollection<Cat>();



            var cats = catCollection.FindAll().Fields(new { Age = true }).Sort("Age", IndexOrder.Ascending).Documents.ToList();



            Assert.AreEqual(1, cats.Count);

            Assert.IsInstanceOfType(typeof(Tiger), cats[0]);

            Assert.AreEqual(19, cats[0].Age);

            Assert.IsNull(cats[0].Name);

        }



        [Test]

        public void Should_support_projections_with_inherited_class_collections_with_linq()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20, Name = "Jim" });

            animalCollection.Save(new Tiger() { Age = 19, Name = "Bob" });



            var catCollection = DB.GetCollection<Cat>();



            var cats = (from a in catCollection.Linq()

                        orderby a.Age ascending

                        select new { a.Name, a.Age }).ToList();



            Assert.AreEqual(1, cats.Count);

            Assert.AreEqual(19, cats[0].Age);

            Assert.AreEqual("Bob", cats[0].Name);

        }



        [Test]

        public void Should_get_correct_count_with_base_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            Assert.AreEqual(2, animalCollection.Count());

        }



        [Test]

        public void Should_get_correct_count_with_base_class_collection_using_linq()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            Assert.AreEqual(2, animalCollection.Linq().Count());

        }



        [Test]

        public void Should_get_correct_count_with_inherited_class_collection()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var catCollection = DB.GetCollection<Cat>();



            Assert.AreEqual(1, catCollection.Count());

        }



        [Test]

        public void Should_get_correct_count_with_inherited_class_collection_using_linq()

        {

            var animalCollection = DB.GetCollection<Animal>();

            animalCollection.Save(new Animal() { Age = 20 });

            animalCollection.Save(new Tiger() { Age = 19 });



            var catCollection = DB.GetCollection<Cat>();



            Assert.AreEqual(1, catCollection.Linq().Count());

        }

    }

}