using System.Linq; using FizzWare.NBuilder; using FluentAssertions; using NUnit.Framework; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.Qualities; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Movies; using System.Collections.Generic; using NzbDrone.Core.Languages; namespace NzbDrone.Core.Test.Datastore { [TestFixture] public class DatabaseRelationshipFixture : DbTest { [SetUp] public void Setup() { // This is kinda hacky here, since we are kinda testing if the QualityDef converter works as well. } [Ignore("MovieFile isnt lazy loaded anymore so this will fail.")] [Test] //TODO: Update this! public void one_to_one() { var episodeFile = Builder.CreateNew() .With(c => c.Quality = new QualityModel()) .BuildNew(); Db.Insert(episodeFile); var episode = Builder.CreateNew() .With(c => c.MovieFileId = episodeFile.Id) .BuildNew(); Db.Insert(episode); var loadedEpisode = Db.Single(); var loadedEpisodeFile = loadedEpisode.MovieFile; loadedEpisodeFile.Should().NotBeNull(); loadedEpisodeFile.ShouldBeEquivalentTo(episodeFile, options => options .IncludingAllRuntimeProperties() .Excluding(c => c.DateAdded) .Excluding(c => c.Path) .Excluding(c => c.Movie)); } [Test] public void one_to_one_should_not_query_db_if_foreign_key_is_zero() { var episode = Builder.CreateNew() .With(c => c.MovieFileId = 0) .BuildNew(); Db.Insert(episode); Db.Single().MovieFile.Should().BeNull(); } [Test] public void embedded_document_as_json() { var quality = new QualityModel { Quality = Quality.Bluray720p, Revision = new Revision(version: 2 )}; var languages = new List { Language.English }; var history = Builder.CreateNew() .With(c => c.Id = 0) .With(c => c.Quality = quality) .With(c => c.Languages = languages) .Build(); Db.Insert(history); var loadedQuality = Db.Single().Quality; loadedQuality.Should().Be(quality); } [Test] public void embedded_list_of_document_with_json() { var languages = new List { Language.English }; var history = Builder.CreateListOfSize(2) .All().With(c => c.Id = 0) .With(c => c.Languages = languages) .Build().ToList(); history[0].Quality = new QualityModel { Quality = Quality.HDTV1080p, Revision = new Revision(version: 2)}; history[1].Quality = new QualityModel { Quality = Quality.Bluray720p, Revision = new Revision(version: 2)}; Db.InsertMany(history); var returnedHistory = Db.All(); returnedHistory[0].Quality.Quality.Should().Be(Quality.HDTV1080p); } } }