diff --git a/NzbDrone.Core.Test/FluentTest.cs b/NzbDrone.Core.Test/FluentTest.cs index 26d88c907..a6e7be4cc 100644 --- a/NzbDrone.Core.Test/FluentTest.cs +++ b/NzbDrone.Core.Test/FluentTest.cs @@ -80,5 +80,89 @@ namespace NzbDrone.Core.Test Console.WriteLine(dateTime.DayOfWeek); dateTime.ToBestDateString().Should().Be(dateTime.ToShortDateString()); } + + [Test] + public void ParentUriString_should_return_self_if_already_parent() + { + //Setup + var url = "http://www.nzbdrone.com"; + var uri = new Uri(url); + + //Act + var result = uri.ParentUriString(); + + //Resolve + result.Should().Be(url); + } + + [Test] + public void ParentUriString_should_return_parent_url_when_path_is_passed() + { + //Setup + var url = "http://www.nzbdrone.com/test/"; + var uri = new Uri(url); + + //Act + var result = uri.ParentUriString(); + + //Resolve + result.Should().Be("http://www.nzbdrone.com"); + } + + [Test] + public void ParentUriString_should_return_parent_url_when_multiple_paths_are_passed() + { + //Setup + var url = "http://www.nzbdrone.com/test/test2"; + var uri = new Uri(url); + + //Act + var result = uri.ParentUriString(); + + //Resolve + result.Should().Be("http://www.nzbdrone.com"); + } + + [Test] + public void ParentUriString_should_return_parent_url_when_url_with_query_string_is_passed() + { + //Setup + var url = "http://www.nzbdrone.com/test.aspx?test=10"; + var uri = new Uri(url); + + //Act + var result = uri.ParentUriString(); + + //Resolve + result.Should().Be("http://www.nzbdrone.com"); + } + + [Test] + public void ParentUriString_should_return_parent_url_when_url_with_path_and_query_strings_is_passed() + { + //Setup + var url = "http://www.nzbdrone.com/tester/test.aspx?test=10"; + var uri = new Uri(url); + + //Act + var result = uri.ParentUriString(); + + //Resolve + result.Should().Be("http://www.nzbdrone.com"); + } + + [Test] + public void ParentUriString_should_return_parent_url_when_url_with_query_strings_is_passed() + { + //Setup + var url = "http://www.nzbdrone.com/test.aspx?test=10&test2=5"; + var uri = new Uri(url); + + //Act + var result = uri.ParentUriString(); + + //Resolve + result.Should().Be("http://www.nzbdrone.com"); + } } } diff --git a/NzbDrone.Core/Fluent.cs b/NzbDrone.Core/Fluent.cs index 9d9cabdf6..fe611d249 100644 --- a/NzbDrone.Core/Fluent.cs +++ b/NzbDrone.Core/Fluent.cs @@ -49,7 +49,7 @@ namespace NzbDrone.Core public static string ParentUriString(this Uri uri) { - return uri.AbsoluteUri.Remove(uri.AbsoluteUri.Length - String.Join("", uri.Segments).Length); + return uri.AbsoluteUri.Remove(uri.AbsoluteUri.Length - String.Join("", uri.Segments).Length - uri.Query.Length); } } }