diff --git a/tests/FilterLists.Agent.Tests/UriExtensionsTests.cs b/tests/FilterLists.Agent.Tests/UriExtensionsTests.cs index 86e66c382..b475e9d1c 100644 --- a/tests/FilterLists.Agent.Tests/UriExtensionsTests.cs +++ b/tests/FilterLists.Agent.Tests/UriExtensionsTests.cs @@ -1,4 +1,5 @@ using System; +using System.Collections.Generic; using FilterLists.Agent.Extensions; using Xunit; @@ -14,5 +15,31 @@ public void IsValidUrl_WithValidUrl_ReturnsTrue() var sut = uri.IsValidUrl(); Assert.True(sut); } + + [Fact] + public void DistributeByHost_HasOverweightHost_DistributesEvenly() + { + var uris = new List + { + new Uri("https://www.google.com/"), + new Uri("https://www.google.com/"), + new Uri("https://www.google.com/"), + new Uri("https://www.facebook.com/"), + new Uri("https://www.facebook.com/") + }; + + var sut = uris.DistributeByHost(); + + var expected = new List + { + new Uri("https://www.google.com/"), + new Uri("https://www.facebook.com/"), + new Uri("https://www.google.com/"), + new Uri("https://www.facebook.com/"), + new Uri("https://www.google.com/") + }; + + Assert.Equal(expected, sut); + } } } \ No newline at end of file