connection_test: Handle num connections may be lower than max

ArchiveTeam · Mar 10, 2015 · 21b1e1d · 21b1e1d
1 parent 55f9aa0
commit 21b1e1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wpull/connection_test.py b/wpull/connection_test.py
@@ -210,7 +210,7 @@ def con_fut():
         self.assertEqual(1, len(pool.host_pools))
         connection_pool_entry = list(pool.host_pools.values())[0]
         self.assertIsInstance(connection_pool_entry, HostPool)
-        self.assertEqual(10, connection_pool_entry.count())
+        self.assertGreaterEqual(10, connection_pool_entry.count())
 
     @wpull.testing.async.async_test(timeout=DEFAULT_TIMEOUT)
     def test_over_host_max_limit_cycling(self):
@@ -232,7 +232,7 @@ def con_fut():
         self.assertEqual(1, len(pool.host_pools))
         connection_pool_entry = list(pool.host_pools.values())[0]
         self.assertIsInstance(connection_pool_entry, HostPool)
-        self.assertEqual(10, connection_pool_entry.count())
+        self.assertGreaterEqual(10, connection_pool_entry.count())
 
     @wpull.testing.async.async_test(timeout=DEFAULT_TIMEOUT)
     def test_multiple_hosts(self):