69 lines
2.0 KiB
C#
69 lines
2.0 KiB
C#
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.MediaFiles;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Core.Test.MediaFileTests
|
|
{
|
|
[TestFixture]
|
|
public class MediaFileRepositoryFixture : DbTest<MediaFileRepository, EpisodeFile>
|
|
{
|
|
[Test]
|
|
public void get_files_by_series()
|
|
{
|
|
var files = Builder<EpisodeFile>.CreateListOfSize(10)
|
|
.All()
|
|
.With(c => c.Id = 0)
|
|
.With(c => c.Quality =new QualityModel(Quality.Bluray720p))
|
|
.Random(4)
|
|
.With(s => s.SeriesId = 12)
|
|
.BuildListOfNew();
|
|
|
|
|
|
Db.InsertMany(files);
|
|
|
|
var seriesFiles = Subject.GetFilesBySeries(12);
|
|
|
|
seriesFiles.Should().HaveCount(4);
|
|
seriesFiles.Should().OnlyContain(c => c.SeriesId == 12);
|
|
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void GetFileByPath_should_return_null_if_file_does_not_exist_in_database()
|
|
{
|
|
Subject.GetFileByPath(@"C:\Test\EpisodeFile.avi").Should().BeNull();
|
|
}
|
|
|
|
[Test]
|
|
public void exists_should_return_false_if_file_doesnt_exist()
|
|
{
|
|
Subject.Exists(@"C:\Test\EpisodeFile.avi").Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void GetFileByPath_should_return_EpisodeFile_if_file_exists_in_database()
|
|
{
|
|
var path = @"C:\Test\EpisodeFile.avi".AsOsAgnostic();
|
|
|
|
var episodeFile = Builder<EpisodeFile>.CreateNew()
|
|
.With(f => f.Id = 0)
|
|
.With(f => f.Path = path.CleanFilePath())
|
|
.Build();
|
|
|
|
Subject.Insert(episodeFile);
|
|
|
|
var file = Subject.GetFileByPath(path);
|
|
|
|
//Resolve
|
|
file.Should().NotBeNull();
|
|
file.Path.Should().Be(path.CleanFilePath());
|
|
}
|
|
}
|
|
} |