Sonarr/NzbDrone.Core.Test/Datastore/DatabaseRelationshipFixture.cs

using System;
using System.Diagnostics;
using System.Linq;
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Qualities;
using NzbDrone.Core.Test.Framework;
using NzbDrone.Core.Tv;

namespace NzbDrone.Core.Test.Datastore
{
    [TestFixture]
    public class DatabaseRelationshipFixture : DbTest
    {
        /*        [Test]
                [Explicit]
                public void benchmark()
                {
                    var series = Builder<Series>.CreateNew()
                                                .With(c => c.Id = 0)
                                                .Build();

                    Marr.Data.MapRepository.Instance.EnableTraceLogging = false;

                    Db.Insert(series);

                    var covers = Builder<MediaCover.MediaCover>.CreateListOfSize(5)
                        .All()
                        .With(c => c.SeriesId = series.Id)
                        .With(c => c.Id = 0)
                        .Build()
                        .ToList();

                    Db.InsertMany(covers);

                    var loadedSeries = Db.Single<Series>();

                    var sw = Stopwatch.StartNew();
                    for (int i = 0; i < 10000; i++)
                    {
                        loadedSeries = Db.Single<Series>();
                        var list = loadedSeries.Covers.Value;
                    }

                    sw.Stop();

                    Console.WriteLine(sw.Elapsed);

                    loadedSeries.Covers.Value.Should().HaveSameCount(covers);
                }

                [Test]
                public void one_to_many()
                {
                    var series = Builder<Series>.CreateNew()
                                                .With(c => c.Id = 0)
                                                .Build();

                    Db.Insert(series);

                    var covers = Builder<MediaCover.MediaCover>.CreateListOfSize(5)
                        .All()
                        .With(c => c.SeriesId = series.Id)
                        .With(c => c.Id = 0)
                        .Build()
                        .ToList();

                    Db.InsertMany(covers);

                    var loadedSeries = Db.Single<Series>();
                    loadedSeries.Covers.Value.Should().HaveSameCount(covers);
                }*/

        [Test]
        public void one_to_one()
        {
            var episode = Builder<Episode>.CreateNew()
                                                       .With(c => c.Id = 0)
                                                       .Build();

            Db.Insert(episode);


            var history = Builder<History.History>.CreateNew()
                            .With(c => c.Id = 0)
                            .With(c => c.EpisodeId = episode.Id)
                            .With(c => c.Quality = new QualityModel())
                            .Build();

            Db.Insert(history);

            var loadedEpisode = Db.Single<History.History>().Episode.Value;

            loadedEpisode.Should().NotBeNull();
            loadedEpisode.ShouldHave().AllProperties().But(c => c.SeriesTitle).EqualTo(episode);
        }


        [Test]
        public void embedded_document_as_json()
        {
            var quality = new QualityModel { Quality = Quality.Bluray720p, Proper = true };

            var history = Builder<History.History>.CreateNew()
                            .With(c => c.Id = 0)
                            .With(c => c.Quality = quality)
                            .Build();

            Db.Insert(history);

            var loadedQuality = Db.Single<History.History>().Quality;
            loadedQuality.Should().Be(quality);
        }

        [Test]
        public void embedded_list_of_document_with_json()
        {
            var quality = new QualityModel { Quality = Quality.Bluray720p, Proper = true };

            var history = Builder<History.History>.CreateListOfSize(2)
                            .All().With(c => c.Id = 0)
                            .Build().ToList();

            history[0].Quality = new QualityModel(Quality.HDTV1080p, true);
            history[1].Quality = new QualityModel(Quality.Bluray720p, true);


            Db.InsertMany(history);

            var returnedHistory = Db.All<History.History>();

            returnedHistory[0].Quality.Quality.Should().Be(Quality.HDTV1080p);
        }
    }
}