mirror of
https://github.com/Radarr/Radarr.git
synced 2024-11-04 10:02:40 +01:00
Add Test Fixture for Migration 154
This commit is contained in:
Qstick
parent
5765961c3a
commit
158a06adbf
@ -0,0 +1,375 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using FluentAssertions;
|
||||
using NUnit.Framework;
|
||||
using NzbDrone.Common.Serializer;
|
||||
using NzbDrone.Core.Datastore.Migration;
|
||||
using NzbDrone.Core.Languages;
|
||||
using NzbDrone.Core.Movies;
|
||||
using NzbDrone.Core.Qualities;
|
||||
using NzbDrone.Core.Test.Framework;
|
||||
|
||||
namespace NzbDrone.Core.Test.Datastore.Migration
|
||||
{
|
||||
[TestFixture]
|
||||
public class add_language_to_files_history_blacklistFixture : MigrationTest<add_language_to_files_history_blacklist>
|
||||
{
|
||||
private void AddDefaultProfile(add_language_to_files_history_blacklist m, string name, Language language)
|
||||
{
|
||||
var allowed = new Quality[] { Quality.WEBDL720p };
|
||||
|
||||
var items = Quality.DefaultQualityDefinitions
|
||||
.OrderBy(v => v.Weight)
|
||||
.Select(v => new { Quality = (int)v.Quality, Allowed = allowed.Contains(v.Quality) })
|
||||
.ToList();
|
||||
|
||||
var profile = new { Id = 1, Name = name, Cutoff = (int)Quality.WEBDL720p, Items = items.ToJson(), Language = (int)language };
|
||||
|
||||
var movie = new
|
||||
{
|
||||
Id = 1,
|
||||
Monitored = true,
|
||||
Title = "My Movie",
|
||||
CleanTitle = "mytitle",
|
||||
Status = MovieStatusType.Announced,
|
||||
MinimumAvailability = MovieStatusType.Announced,
|
||||
Images = new[] { new { CoverType = "Poster" } }.ToJson(),
|
||||
HasPreDBEntry = false,
|
||||
PathState = 1,
|
||||
Runtime = 90,
|
||||
ProfileId = 1,
|
||||
MovieFileId = 1,
|
||||
Path = "/Some/Path",
|
||||
TitleSlug = "123456",
|
||||
TmdbId = 123456
|
||||
};
|
||||
|
||||
m.Insert.IntoTable("Profiles").Row(profile);
|
||||
m.Insert.IntoTable("Movies").Row(movie);
|
||||
}
|
||||
|
||||
private void AddMovieFile(add_language_to_files_history_blacklist m, string sceneName, string mediaInfoLanugaes)
|
||||
{
|
||||
m.Insert.IntoTable("MovieFiles").Row(new
|
||||
{
|
||||
MovieId = 1,
|
||||
Quality = new
|
||||
{
|
||||
Quality = 6
|
||||
}.ToJson(),
|
||||
Size = 997478103,
|
||||
DateAdded = DateTime.Now,
|
||||
SceneName = sceneName,
|
||||
MediaInfo = new
|
||||
{
|
||||
AudioLanguages = mediaInfoLanugaes
|
||||
}.ToJson(),
|
||||
RelativePath = "Never Say Never Again.1983.Bluray-720p.mp4",
|
||||
});
|
||||
}
|
||||
|
||||
private void AddHistory(add_language_to_files_history_blacklist m, string sourceTitle)
|
||||
{
|
||||
m.Insert.IntoTable("History").Row(new
|
||||
{
|
||||
MovieId = 1,
|
||||
Quality = new
|
||||
{
|
||||
Quality = 6
|
||||
}.ToJson(),
|
||||
EventType = 1,
|
||||
Date = DateTime.Now,
|
||||
SourceTitle = sourceTitle,
|
||||
Data = new
|
||||
{
|
||||
Indexer = "My Indexer"
|
||||
}.ToJson()
|
||||
});
|
||||
}
|
||||
|
||||
private void AddBlacklist(add_language_to_files_history_blacklist m, string sourceTitle)
|
||||
{
|
||||
m.Insert.IntoTable("Blacklist").Row(new
|
||||
{
|
||||
MovieId = 1,
|
||||
Quality = new
|
||||
{
|
||||
Quality = 6
|
||||
}.ToJson(),
|
||||
Date = DateTime.Now,
|
||||
SourceTitle = sourceTitle,
|
||||
Size = 997478103
|
||||
});
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_add_languages_from_media_info_if_available()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "My.Movie.2018.German.BluRay-Radarr", "Japanese");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Japanese);
|
||||
items.First().Languages.Should().NotContain((int)Language.English);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_add_languages_from_media_info_with_multiple_language()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "My.Movie.2018.German.BluRay-Radarr", "Japanese / French");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(2);
|
||||
items.First().Languages.Should().Contain((int)Language.Japanese);
|
||||
items.First().Languages.Should().Contain((int)Language.French);
|
||||
items.First().Languages.Should().NotContain((int)Language.English);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_fallback_to_scenename_if_no_mediainfo_languages()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "My.Movie.2018.German.BluRay-Radarr", "");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.German);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_fallback_to_scenename_if_mediainfo_language_invalid()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "My.Movie.2018.German.BluRay-Radarr", "English (USA)");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.German);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.English);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_fallback_to_profile_if_no_mediainfo_no_scene_name()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "", "");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_handle_if_null_mediainfo_and_null_scenename()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, null, null);
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_fallback_to_profile_if_unknown_language_from_scene_name()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "My.Movie.2018.BluRay-Radarr", "");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_use_english_if_fallback_to_profile_and_profile_is_any()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Any);
|
||||
AddMovieFile(c, "", "");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM MovieFiles");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.English);
|
||||
items.First().Languages.Should().NotContain((int)Language.Any);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_history_languages_from_sourceTitle()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddHistory(c, "My.Movie.2018.Italian.BluRay-Radarr");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM History");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Italian);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_history_languages_from_profile_if_no_sourceTitle()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddHistory(c, "");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM History");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_history_languages_from_profile_if_unknown_sourceTitle()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddHistory(c, "Man on Fire");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM History");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_history_languages_from_moviefile_release_mapping_with_mediainfo()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddMovieFile(c, "My.Movie.2018.Italian.BluRay-Radarr", "Italian / French / German");
|
||||
AddHistory(c, "My.Movie.2018.Italian.BluRay-Radarr");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM History");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(3);
|
||||
items.First().Languages.Should().Contain((int)Language.Italian);
|
||||
items.First().Languages.Should().Contain((int)Language.French);
|
||||
items.First().Languages.Should().Contain((int)Language.German);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_blacklist_languages_from_sourceTitle()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddBlacklist(c, "My.Movie.2018.Italian.BluRay-Radarr");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM Blacklist");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Italian);
|
||||
items.First().Languages.Should().NotContain((int)Language.Dutch);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_blacklist_languages_from_profile_if_no_sourceTitle()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddBlacklist(c, "");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM Blacklist");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_assign_blacklist_languages_from_profile_if_unknown_sourceTitle()
|
||||
{
|
||||
var db = WithMigrationTestDb(c =>
|
||||
{
|
||||
AddDefaultProfile(c, "My Custom Profile", Language.Dutch);
|
||||
AddBlacklist(c, "Man on Fire");
|
||||
});
|
||||
|
||||
var items = db.Query<ModelWithLanguages154>("SELECT Id, Languages FROM Blacklist");
|
||||
|
||||
items.Should().HaveCount(1);
|
||||
items.First().Languages.Count.Should().Be(1);
|
||||
items.First().Languages.Should().Contain((int)Language.Dutch);
|
||||
items.First().Languages.Should().NotContain((int)Language.Unknown);
|
||||
}
|
||||
}
|
||||
|
||||
public class ModelWithLanguages154
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public List<int> Languages { get; set; }
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user