mirror of
https://github.com/lidarr/Lidarr
synced 2025-03-15 16:26:34 +00:00
338 lines
15 KiB
C#
338 lines
15 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using FizzWare.NBuilder;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.DataAugmentation.Scene;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Test.Common;
|
|
using FluentAssertions;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.DataAugmentation;
|
|
|
|
namespace NzbDrone.Core.Test.DataAugmentation.Scene
|
|
{
|
|
[TestFixture]
|
|
|
|
public class SceneMappingServiceFixture : CoreTest<SceneMappingService>
|
|
{
|
|
private List<SceneMapping> _fakeMappings;
|
|
|
|
private Mock<ISceneMappingProvider> _provider1;
|
|
private Mock<ISceneMappingProvider> _provider2;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_fakeMappings = Builder<SceneMapping>.CreateListOfSize(5).BuildListOfNew();
|
|
|
|
_fakeMappings[0].SearchTerm = "Words";
|
|
_fakeMappings[1].SearchTerm = "That";
|
|
_fakeMappings[2].SearchTerm = "Can";
|
|
_fakeMappings[3].SearchTerm = "Be";
|
|
_fakeMappings[4].SearchTerm = "Cleaned";
|
|
|
|
_fakeMappings[0].ParseTerm = "Words";
|
|
_fakeMappings[1].ParseTerm = "That";
|
|
_fakeMappings[2].ParseTerm = "Can";
|
|
_fakeMappings[3].ParseTerm = "Be";
|
|
_fakeMappings[4].ParseTerm = "Cleaned";
|
|
|
|
_provider1 = new Mock<ISceneMappingProvider>();
|
|
_provider1.Setup(s => s.GetSceneMappings()).Returns(_fakeMappings);
|
|
|
|
_provider2 = new Mock<ISceneMappingProvider>();
|
|
_provider2.Setup(s => s.GetSceneMappings()).Returns(_fakeMappings);
|
|
}
|
|
|
|
private void GivenProviders(IEnumerable<Mock<ISceneMappingProvider>> providers)
|
|
{
|
|
Mocker.SetConstant<IEnumerable<ISceneMappingProvider>>(providers.Select(s => s.Object));
|
|
}
|
|
|
|
[Test]
|
|
public void should_purge_existing_mapping_and_add_new_ones()
|
|
{
|
|
GivenProviders(new [] { _provider1 });
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(_fakeMappings);
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
AssertMappingUpdated();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_delete_if_fetch_fails()
|
|
{
|
|
GivenProviders(new[] { _provider1 });
|
|
|
|
_provider1.Setup(c => c.GetSceneMappings()).Throws(new WebException());
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
AssertNoUpdate();
|
|
|
|
ExceptionVerification.ExpectedErrors(1);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_delete_if_fetch_returns_empty_list()
|
|
{
|
|
GivenProviders(new[] { _provider1 });
|
|
|
|
_provider1.Setup(c => c.GetSceneMappings()).Returns(new List<SceneMapping>());
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
AssertNoUpdate();
|
|
|
|
ExceptionVerification.ExpectedWarns(1);
|
|
}
|
|
|
|
[Test]
|
|
public void should_get_mappings_for_all_providers()
|
|
{
|
|
GivenProviders(new[] { _provider1, _provider2 });
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(_fakeMappings);
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
_provider1.Verify(c => c.GetSceneMappings(), Times.Once());
|
|
_provider2.Verify(c => c.GetSceneMappings(), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_refresh_cache_if_cache_is_empty_when_looking_for_tvdb_id()
|
|
{
|
|
Subject.FindTvdbId("title");
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>()
|
|
.Verify(v => v.All(), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_refresh_cache_if_cache_is_not_empty_when_looking_for_tvdb_id()
|
|
{
|
|
GivenProviders(new[] { _provider1 });
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>()
|
|
.Setup(s => s.All())
|
|
.Returns(Builder<SceneMapping>.CreateListOfSize(1).Build());
|
|
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>()
|
|
.Verify(v => v.All(), Times.Once());
|
|
|
|
Subject.FindTvdbId("title");
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>()
|
|
.Verify(v => v.All(), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_add_mapping_with_blank_title()
|
|
{
|
|
GivenProviders(new[] { _provider1 });
|
|
|
|
var fakeMappings = Builder<SceneMapping>.CreateListOfSize(2)
|
|
.TheLast(1)
|
|
.With(m => m.Title = null)
|
|
.Build()
|
|
.ToList();
|
|
|
|
_provider1.Setup(s => s.GetSceneMappings()).Returns(fakeMappings);
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(_fakeMappings);
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Verify(c => c.InsertMany(It.Is<IList<SceneMapping>>(m => !m.Any(s => s.Title.IsNullOrWhiteSpace()))), Times.Once());
|
|
ExceptionVerification.ExpectedWarns(1);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_add_mapping_with_blank_search_title()
|
|
{
|
|
GivenProviders(new[] { _provider1 });
|
|
|
|
var fakeMappings = Builder<SceneMapping>.CreateListOfSize(2)
|
|
.TheLast(1)
|
|
.With(m => m.SearchTerm = null)
|
|
.Build()
|
|
.ToList();
|
|
|
|
_provider1.Setup(s => s.GetSceneMappings()).Returns(fakeMappings);
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(_fakeMappings);
|
|
|
|
Subject.Execute(new UpdateSceneMappingCommand());
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Verify(c => c.InsertMany(It.Is<IList<SceneMapping>>(m => !m.Any(s => s. SearchTerm.IsNullOrWhiteSpace()))), Times.Once());
|
|
ExceptionVerification.ExpectedWarns(1);
|
|
}
|
|
|
|
|
|
[TestCase("Working!!", "Working!!", 1)]
|
|
[TestCase("Working`!!", "Working`!!", 2)]
|
|
[TestCase("Working!!!", "Working!!!", 3)]
|
|
[TestCase("Working!!!!", "Working!!!", 3)]
|
|
[TestCase("Working !!", "Working!!", 1)]
|
|
public void should_return_single_match(string parseTitle, string title, int expectedSeasonNumber)
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Working!!", ParseTerm = "working", SearchTerm = "Working!!", TvdbId = 100, SceneSeasonNumber = 1 },
|
|
new SceneMapping { Title = "Working`!!", ParseTerm = "working", SearchTerm = "Working`!!", TvdbId = 100, SceneSeasonNumber = 2 },
|
|
new SceneMapping { Title = "Working!!!", ParseTerm = "working", SearchTerm = "Working!!!", TvdbId = 100, SceneSeasonNumber = 3 },
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
var tvdbId = Subject.FindTvdbId(parseTitle);
|
|
var seasonNumber = Subject.GetSceneSeasonNumber(parseTitle);
|
|
|
|
tvdbId.Should().Be(100);
|
|
seasonNumber.Should().Be(expectedSeasonNumber);
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_alternate_title_for_global_season()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu 1", ParseTerm = "fudanshikoukouseikatsu1", SearchTerm = "Fudanshi Koukou Seikatsu 1", TvdbId = 100, SeasonNumber = null, SceneSeasonNumber = null },
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu 2", ParseTerm = "fudanshikoukouseikatsu2", SearchTerm = "Fudanshi Koukou Seikatsu 2", TvdbId = 100, SeasonNumber = -1, SceneSeasonNumber = null },
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu 3", ParseTerm = "fudanshikoukouseikatsu3", SearchTerm = "Fudanshi Koukou Seikatsu 3", TvdbId = 100, SeasonNumber = null, SceneSeasonNumber = -1 },
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu 4", ParseTerm = "fudanshikoukouseikatsu4", SearchTerm = "Fudanshi Koukou Seikatsu 4", TvdbId = 100, SeasonNumber = -1, SceneSeasonNumber = -1 },
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
var names = Subject.GetSceneNames(100, new List<int> { 10 }, new List<int> { 10 });
|
|
names.Should().HaveCount(4);
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_alternate_title_for_season()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu", ParseTerm = "fudanshikoukouseikatsu", SearchTerm = "Fudanshi Koukou Seikatsu", TvdbId = 100, SeasonNumber = 1, SceneSeasonNumber = null }
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
var names = Subject.GetSceneNames(100, new List<int> { 1 }, new List<int> { 10 });
|
|
names.Should().HaveCount(1);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_return_alternate_title_for_season()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu", ParseTerm = "fudanshikoukouseikatsu", SearchTerm = "Fudanshi Koukou Seikatsu", TvdbId = 100, SeasonNumber = 1, SceneSeasonNumber = null }
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
var names = Subject.GetSceneNames(100, new List<int> { 2 }, new List<int> { 10 });
|
|
names.Should().BeEmpty();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_alternate_title_for_sceneseason()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu", ParseTerm = "fudanshikoukouseikatsu", SearchTerm = "Fudanshi Koukou Seikatsu", TvdbId = 100, SeasonNumber = null, SceneSeasonNumber = 1 }
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
var names = Subject.GetSceneNames(100, new List<int> { 10 }, new List<int> { 1 });
|
|
names.Should().HaveCount(1);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_return_alternate_title_for_sceneseason()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu", ParseTerm = "fudanshikoukouseikatsu", SearchTerm = "Fudanshi Koukou Seikatsu", TvdbId = 100, SeasonNumber = null, SceneSeasonNumber = 1 }
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
var names = Subject.GetSceneNames(100, new List<int> { 10 }, new List<int> { 2 });
|
|
names.Should().BeEmpty();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_alternate_title_for_fairy_tail()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fairy Tail S2", ParseTerm = "fairytails2", SearchTerm = "Fairy Tail S2", TvdbId = 100, SeasonNumber = null, SceneSeasonNumber = 2 }
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
Subject.GetSceneNames(100, new List<int> { 4 }, new List<int> { 20 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 5 }, new List<int> { 20 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 6 }, new List<int> { 20 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 7 }, new List<int> { 20 }).Should().BeEmpty();
|
|
|
|
Subject.GetSceneNames(100, new List<int> { 20 }, new List<int> { 1 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 20 }, new List<int> { 2 }).Should().NotBeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 20 }, new List<int> { 3 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 20 }, new List<int> { 4 }).Should().BeEmpty();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_alternate_title_for_fudanshi()
|
|
{
|
|
var mappings = new List<SceneMapping>
|
|
{
|
|
new SceneMapping { Title = "Fudanshi Koukou Seikatsu", ParseTerm = "fudanshikoukouseikatsu", SearchTerm = "Fudanshi Koukou Seikatsu", TvdbId = 100, SeasonNumber = null, SceneSeasonNumber = 1 }
|
|
};
|
|
|
|
Mocker.GetMock<ISceneMappingRepository>().Setup(c => c.All()).Returns(mappings);
|
|
|
|
Subject.GetSceneNames(100, new List<int> { 1 }, new List<int> { 20 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 2 }, new List<int> { 20 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 3 }, new List<int> { 20 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 4 }, new List<int> { 20 }).Should().BeEmpty();
|
|
|
|
Subject.GetSceneNames(100, new List<int> { 1 }, new List<int> { 1 }).Should().NotBeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 2 }, new List<int> { 2 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 3 }, new List<int> { 3 }).Should().BeEmpty();
|
|
Subject.GetSceneNames(100, new List<int> { 4 }, new List<int> { 4 }).Should().BeEmpty();
|
|
}
|
|
|
|
private void AssertNoUpdate()
|
|
{
|
|
_provider1.Verify(c => c.GetSceneMappings(), Times.Once());
|
|
Mocker.GetMock<ISceneMappingRepository>().Verify(c => c.Clear(It.IsAny<string>()), Times.Never());
|
|
Mocker.GetMock<ISceneMappingRepository>().Verify(c => c.InsertMany(_fakeMappings), Times.Never());
|
|
}
|
|
|
|
private void AssertMappingUpdated()
|
|
{
|
|
_provider1.Verify(c => c.GetSceneMappings(), Times.Once());
|
|
Mocker.GetMock<ISceneMappingRepository>().Verify(c => c.Clear(It.IsAny<string>()), Times.Once());
|
|
Mocker.GetMock<ISceneMappingRepository>().Verify(c => c.InsertMany(_fakeMappings), Times.Once());
|
|
|
|
foreach (var sceneMapping in _fakeMappings)
|
|
{
|
|
Subject.GetSceneNames(sceneMapping.TvdbId, _fakeMappings.Select(m => m.SeasonNumber.Value).Distinct().ToList(), new List<int>()).Should().Contain(sceneMapping.SearchTerm);
|
|
Subject.FindTvdbId(sceneMapping.ParseTerm).Should().Be(sceneMapping.TvdbId);
|
|
}
|
|
}
|
|
}
|
|
}
|