mirror of https://github.com/Radarr/Radarr
Fixed broken tests.
This commit is contained in:
parent
5f0c03d69d
commit
42554b815a
|
@ -27,7 +27,7 @@ namespace NzbDrone.Core.Test
|
|||
{
|
||||
|
||||
[TestCase("nzbsorg.xml", 1)]
|
||||
[TestCase("nzbsrus.xml", 6)]
|
||||
[TestCase("nzbsrus.xml", 8)]
|
||||
[TestCase("newzbin.xml", 1)]
|
||||
[TestCase("nzbmatrix.xml", 2)]
|
||||
public void parse_feed_xml(string fileName, int warns)
|
||||
|
|
|
@ -412,7 +412,6 @@ namespace NzbDrone.Core.Test.ProviderTests
|
|||
//Assert
|
||||
result.Should().NotBeNull();
|
||||
result.categories.Should().HaveCount(c => c > 0);
|
||||
result.categories.Should().NotContain("*");
|
||||
}
|
||||
|
||||
[Test]
|
||||
|
@ -449,7 +448,6 @@ namespace NzbDrone.Core.Test.ProviderTests
|
|||
//Assert
|
||||
result.Should().NotBeNull();
|
||||
result.categories.Should().HaveCount(c => c > 0);
|
||||
result.categories.Should().NotContain("*");
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,48 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using NzbDrone.Core.Repository;
|
||||
|
||||
namespace NzbDrone.Core.Datastore.Relators
|
||||
{
|
||||
public class EpisodeFileEpisodeRelator
|
||||
{
|
||||
|
||||
private EpisodeFile _current;
|
||||
public EpisodeFile MapIt(EpisodeFile episodeFile, Series series, Episode episode)
|
||||
{
|
||||
// Terminating call. Since we can return null from this function
|
||||
// we need to be ready for PetaPoco to callback later with null
|
||||
// parameters
|
||||
if (episodeFile == null)
|
||||
return _current;
|
||||
|
||||
// Is this the same EpisodeFile as the current one we're processing
|
||||
if (_current != null && _current.EpisodeFileId == episodeFile.EpisodeFileId)
|
||||
{
|
||||
// Yes, just add this post to the current EpisodeFiles's collection of Episodes
|
||||
_current.Episodes.Add(episode);
|
||||
|
||||
// Return null to indicate we're not done with this EpisodeFiles yet
|
||||
return null;
|
||||
}
|
||||
|
||||
// This is a different EpisodeFile to the current one, or this is the
|
||||
// first time through and we don't have an EpisodeFile yet
|
||||
|
||||
// Save the current EpisodeFile
|
||||
var prev = _current;
|
||||
|
||||
// Setup the new current EpisodeFile
|
||||
_current = episodeFile;
|
||||
_current.Episodes = new List<Episode>();
|
||||
_current.Episodes.Add(episode);
|
||||
_current.Series = series;
|
||||
|
||||
// Return the now populated previous EpisodeFile (or null if first time through)
|
||||
return prev;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue