From 7a9191cf36585868feb5a525ae93f1402579d623 Mon Sep 17 00:00:00 2001
From: Joakim Soderlund <joakim.soderlund@gmail.com>
Date: Sun, 24 Mar 2019 14:04:53 +0100
Subject: [PATCH] Improve performance of Fimfarchive fetcher tests

---
 tests/fetchers/test_fimfarchive.py | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/tests/fetchers/test_fimfarchive.py b/tests/fetchers/test_fimfarchive.py
index be67b27..2dc3320 100644
--- a/tests/fetchers/test_fimfarchive.py
+++ b/tests/fetchers/test_fimfarchive.py
@@ -25,13 +25,14 @@ Fimfarchive fetcher tests.
 import json
 from io import BytesIO
 from typing import Any, Dict, List
+from unittest.mock import patch, MagicMock
 from zipfile import ZipFile
 
 import arrow
 import pytest
 
 from fimfarchive.exceptions import InvalidStoryError, StorySourceError
-from fimfarchive.fetchers import Fetcher, FimfarchiveFetcher
+from fimfarchive.fetchers import fimfarchive, Fetcher, FimfarchiveFetcher
 from fimfarchive.stories import Story
 from fimfarchive.utils import JayWalker
 
@@ -171,6 +172,16 @@ def serialize(obj: Dict) -> str:
     return output
 
 
+class PoolMock(MagicMock):
+    """
+    Mocks the multiprocessing pool.
+    """
+
+    @staticmethod
+    def imap(func, iterable, chunksize):
+        yield from map(func, iterable)
+
+
 class TestFimfarchiveFetcher:
     """
     FimfarchiveFetcher tests.
@@ -196,8 +207,16 @@ class TestFimfarchiveFetcher:
 
         return stream
 
+    @pytest.fixture
+    def pool(self):
+        """
+        Yields a multiprocessing pool mock.
+        """
+        with patch.object(fimfarchive, 'Pool', PoolMock) as mock:
+            yield mock
+
     @pytest.fixture()
-    def fetcher(self, archive):
+    def fetcher(self, archive, pool):
         """
         Returns the fetcher instance to test.
         """