You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
3.9 KiB
122 lines
3.9 KiB
5 years ago
|
using FizzWare.NBuilder;
|
||
|
using FluentAssertions;
|
||
|
using NUnit.Framework;
|
||
|
using NzbDrone.Core.Extras.Subtitles;
|
||
|
using NzbDrone.Core.Languages;
|
||
|
using NzbDrone.Core.MediaFiles;
|
||
|
using NzbDrone.Core.Movies;
|
||
|
using NzbDrone.Core.Test.Framework;
|
||
|
|
||
|
namespace NzbDrone.Core.Test.MediaFiles
|
||
|
{
|
||
|
[TestFixture]
|
||
|
public class SubtitleFileRepositoryFixture : DbTest<SubtitleFileRepository, SubtitleFile>
|
||
|
{
|
||
|
private Movie _movie;
|
||
|
private MovieFile _movieFile1;
|
||
|
private MovieFile _movieFile2;
|
||
|
|
||
|
[SetUp]
|
||
|
public void Setup()
|
||
|
{
|
||
|
_movie = Builder<Movie>.CreateNew()
|
||
|
.With(s => s.Id = 7)
|
||
|
.Build();
|
||
|
|
||
|
_movieFile1 = Builder<MovieFile>.CreateNew()
|
||
|
.With(s => s.Id = 10)
|
||
|
.With(s => s.MovieId = _movie.Id)
|
||
|
.Build();
|
||
|
|
||
|
_movieFile2 = Builder<MovieFile>.CreateNew()
|
||
|
.With(s => s.Id = 11)
|
||
|
.With(s => s.MovieId = _movie.Id)
|
||
|
.Build();
|
||
|
}
|
||
|
|
||
|
[Test]
|
||
|
public void should_delete_files_by_movieId()
|
||
|
{
|
||
|
var files = Builder<SubtitleFile>.CreateListOfSize(5)
|
||
|
.All()
|
||
|
.With(c => c.Id = 0)
|
||
|
.With(c => c.MovieId = _movie.Id)
|
||
|
.With(c => c.MovieFileId = 11)
|
||
|
.With(c => c.Language = Language.English)
|
||
|
.BuildListOfNew();
|
||
|
|
||
|
Db.InsertMany(files);
|
||
|
|
||
|
Subject.DeleteForMovie(_movie.Id);
|
||
|
|
||
|
var remainingFiles = Subject.GetFilesByMovie(_movie.Id);
|
||
|
|
||
|
remainingFiles.Should().HaveCount(0);
|
||
|
}
|
||
|
|
||
|
[Test]
|
||
|
public void should_delete_files_by_movieFileId()
|
||
|
{
|
||
|
var files = Builder<SubtitleFile>.CreateListOfSize(5)
|
||
|
.All()
|
||
|
.With(c => c.Id = 0)
|
||
|
.With(c => c.MovieId = _movie.Id)
|
||
|
.With(c => c.MovieFileId = _movieFile2.Id)
|
||
|
.With(c => c.Language = Language.English)
|
||
|
.Random(2)
|
||
|
.With(c => c.MovieFileId = _movieFile1.Id)
|
||
|
.BuildListOfNew();
|
||
|
|
||
|
Db.InsertMany(files);
|
||
|
|
||
|
Subject.DeleteForMovieFile(_movieFile2.Id);
|
||
|
|
||
|
var remainingFiles = Subject.GetFilesByMovie(_movie.Id);
|
||
|
|
||
|
remainingFiles.Should().HaveCount(2);
|
||
|
}
|
||
|
|
||
|
[Test]
|
||
|
public void should_get_files_by_movieFileId()
|
||
|
{
|
||
|
var files = Builder<SubtitleFile>.CreateListOfSize(5)
|
||
|
.All()
|
||
|
.With(c => c.Id = 0)
|
||
|
.With(c => c.MovieId = _movie.Id)
|
||
|
.With(c => c.MovieFileId = _movieFile2.Id)
|
||
|
.With(c => c.Language = Language.English)
|
||
|
.Random(2)
|
||
|
.With(c => c.MovieFileId = _movieFile1.Id)
|
||
|
.BuildListOfNew();
|
||
|
|
||
|
Db.InsertMany(files);
|
||
|
|
||
|
var remainingFiles = Subject.GetFilesByMovieFile(_movieFile2.Id);
|
||
|
|
||
|
remainingFiles.Should().HaveCount(3);
|
||
|
remainingFiles.Should().OnlyContain(c => c.MovieFileId == _movieFile2.Id);
|
||
|
}
|
||
|
|
||
|
[Test]
|
||
|
public void should_get_files_by_movieId()
|
||
|
{
|
||
|
var files = Builder<SubtitleFile>.CreateListOfSize(5)
|
||
|
.All()
|
||
|
.With(c => c.Id = 0)
|
||
|
.With(c => c.MovieId = _movie.Id)
|
||
|
.With(c => c.MovieFileId = _movieFile2.Id)
|
||
|
.With(c => c.Language = Language.English)
|
||
|
.Random(2)
|
||
|
.With(c => c.MovieFileId = _movieFile1.Id)
|
||
|
.BuildListOfNew();
|
||
|
|
||
|
Db.InsertMany(files);
|
||
|
|
||
|
var remainingFiles = Subject.GetFilesByMovie(_movie.Id);
|
||
|
|
||
|
remainingFiles.Should().HaveCount(5);
|
||
|
remainingFiles.Should().OnlyContain(c => c.MovieId == _movie.Id);
|
||
|
}
|
||
|
}
|
||
|
}
|