mirror of https://github.com/Radarr/Radarr
73 lines
2.3 KiB
C#
73 lines
2.3 KiB
C#
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.History;
|
|
using NzbDrone.Core.Languages;
|
|
using NzbDrone.Core.Movies;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.HistoryTests
|
|
{
|
|
[TestFixture]
|
|
public class HistoryRepositoryFixture : DbTest<HistoryRepository, MovieHistory>
|
|
{
|
|
private Movie _movie1;
|
|
private Movie _movie2;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_movie1 = Builder<Movie>.CreateNew()
|
|
.With(s => s.Id = 7)
|
|
.Build();
|
|
|
|
_movie2 = Builder<Movie>.CreateNew()
|
|
.With(s => s.Id = 8)
|
|
.Build();
|
|
}
|
|
|
|
[Test]
|
|
public void should_read_write_dictionary()
|
|
{
|
|
var history = Builder<MovieHistory>.CreateNew()
|
|
.With(c => c.Quality = new QualityModel())
|
|
.With(c => c.Languages = new List<Language>())
|
|
.BuildNew();
|
|
|
|
history.Data.Add("key1", "value1");
|
|
history.Data.Add("key2", "value2");
|
|
|
|
Subject.Insert(history);
|
|
|
|
StoredModel.Data.Should().HaveCount(2);
|
|
}
|
|
|
|
[Test]
|
|
public void should_get_download_history()
|
|
{
|
|
var historyBluray = Builder<MovieHistory>.CreateNew()
|
|
.With(c => c.Quality = new QualityModel(Quality.Bluray1080p))
|
|
.With(c => c.Languages = new List<Language> { Language.English })
|
|
.With(c => c.MovieId = 12)
|
|
.With(c => c.EventType = MovieHistoryEventType.Grabbed)
|
|
.BuildNew();
|
|
|
|
var historyDvd = Builder<MovieHistory>.CreateNew()
|
|
.With(c => c.Quality = new QualityModel(Quality.DVD))
|
|
.With(c => c.Languages = new List<Language> { Language.English })
|
|
.With(c => c.MovieId = 12)
|
|
.With(c => c.EventType = MovieHistoryEventType.Grabbed)
|
|
.BuildNew();
|
|
|
|
Subject.Insert(historyBluray);
|
|
Subject.Insert(historyDvd);
|
|
|
|
var downloadHistory = Subject.FindDownloadHistory(12, new QualityModel(Quality.Bluray1080p));
|
|
|
|
downloadHistory.Should().HaveCount(1);
|
|
}
|
|
}
|
|
}
|