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.CreateNew() .With(c => c.Id = 0) .Build(); Marr.Data.MapRepository.Instance.EnableTraceLogging = false; Db.Insert(series); var covers = Builder.CreateListOfSize(5) .All() .With(c => c.SeriesId = series.Id) .With(c => c.Id = 0) .Build() .ToList(); Db.InsertMany(covers); var loadedSeries = Db.Single(); var sw = Stopwatch.StartNew(); for (int i = 0; i < 10000; i++) { loadedSeries = Db.Single(); 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.CreateNew() .With(c => c.Id = 0) .Build(); Db.Insert(series); var covers = Builder.CreateListOfSize(5) .All() .With(c => c.SeriesId = series.Id) .With(c => c.Id = 0) .Build() .ToList(); Db.InsertMany(covers); var loadedSeries = Db.Single(); loadedSeries.Covers.Value.Should().HaveSameCount(covers); } [Test] public void one_to_one() { var episode = Builder.CreateNew() .With(c => c.Id = 0) .Build(); Db.Insert(episode); var history = Builder.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().Episode.Value; loadedEpisode.Should().NotBeNull(); loadedEpisode.ShouldHave().AllProperties().EqualTo(episode); } [Test] public void embedded_document_as_json() { var quality = new QualityModel { Quality = Quality.Bluray720p, Proper = true }; var history = Builder.CreateNew() .With(c => c.Id = 0) .With(c => c.Quality = quality) .Build(); Db.Insert(history); var loadedQuality = Db.Single().Quality; loadedQuality.Should().Be(quality); } } }