From ca65deb369692e856088c58a2267016a529efced Mon Sep 17 00:00:00 2001
From: Bastian Kleineidam <bastian.kleineidam@web.de>
Date: Thu, 28 Feb 2013 21:08:43 +0100
Subject: [PATCH] Limit connections to hosts for tests.

---
 tests/test_comics.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/tests/test_comics.py b/tests/test_comics.py
index 7102b9e2f..58cd14c16 100644
--- a/tests/test_comics.py
+++ b/tests/test_comics.py
@@ -5,11 +5,30 @@ import tempfile
 import shutil
 import re
 import os
+import multiprocessing
+import urlparse
 from itertools import islice
 from unittest import TestCase
 from dosagelib import scraper
 
 
+def get_host(url):
+    """Get host part of URL."""
+    return urlparse.urlsplit(url)[1].lower()
+
+
+# Dictionary with per-host locks.
+_locks = {}
+# Allowed number of connections per host
+MaxConnections = 4
+
+def get_lock(host):
+    """Get bounded semphore for given host."""
+    if host not in _locks:
+        _locks[host] = multiprocessing.BoundedSemaphore(MaxConnections)
+    return _locks[host]
+
+
 class _ComicTester(TestCase):
     """Basic comic test class."""
     scraperclass=None
@@ -33,6 +52,12 @@ class _ComicTester(TestCase):
         # at least 5 strips from the start, and find strip images
         # on at least 4 pages.
         scraperobj = self.scraperclass()
+        # Limit number of connections to one host.
+        host = get_host(scraperobj.url)
+        with get_lock(host):
+            self._test_comic(scraperobj)
+
+    def _test_comic(self, scraperobj):
         num = 0
         max_strips = 5
         for strip in islice(scraperobj.getAllStrips(), 0, max_strips):