From bd2a58e3ee05c7323ae004c17ad5a99641ec975d Mon Sep 17 00:00:00 2001
From: Joakim Soderlund <joakim.soderlund@gmail.com>
Date: Sat, 10 Oct 2015 01:46:17 +0200
Subject: [PATCH] Add tests for Fimfarchive fetcher

---
 tests/test_fetchers.py | 59 +++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 58 insertions(+), 1 deletion(-)

diff --git a/tests/test_fetchers.py b/tests/test_fetchers.py
index 1ebc8b8..ddfb07c 100644
--- a/tests/test_fetchers.py
+++ b/tests/test_fetchers.py
@@ -25,7 +25,7 @@ Fetcher tests.
 import pytest
 
 from fimfarchive.exceptions import InvalidStoryError, StorySourceError
-from fimfarchive.fetchers import FimfictionFetcher
+from fimfarchive.fetchers import FimfarchiveFetcher, FimfictionFetcher
 
 
 VALID_STORY_PK = 9
@@ -33,6 +33,8 @@ INVALID_STORY_PK = 7
 EMPTY_STORY_PK = 8
 PROTECTED_STORY_PK = 208799
 
+FIMFARCHIVE_PATH = 'fimfarchive-20150901.zip'
+
 
 class TestFimfictionFetcher:
     """
@@ -138,3 +140,58 @@ class TestFimfictionFetcher:
         """
         with pytest.raises(InvalidStoryError):
             self.fetcher.fetch_data(PROTECTED_STORY_PK)
+
+
+class TestFimfarchiveFetcher:
+    """
+    FimfarchiveFetcher tests.
+    """
+    fetcher = None
+
+    @classmethod
+    def setup_class(cls):
+        cls.fetcher = FimfarchiveFetcher(FIMFARCHIVE_PATH)
+
+    @classmethod
+    def teardown_class(cls):
+        cls.fetcher.close()
+        cls.fetcher = None
+
+    def test_with_statment(self):
+        """
+        Tests fetcher can be used in with statements.
+        """
+        with FimfarchiveFetcher(FIMFARCHIVE_PATH) as fetcher:
+            fetcher.lookup(VALID_STORY_PK)
+
+        with pytest.raises(StorySourceError):
+            fetcher.lookup(VALID_STORY_PK)
+
+    def test_fetch_meta_for_valid_story(self):
+        """
+        Tests meta is returned if story is valid
+        """
+        meta = self.fetcher.fetch_meta(VALID_STORY_PK)
+        assert meta['id'] == VALID_STORY_PK
+        assert meta['words'] != 0
+
+    def test_fetch_meta_for_invalid_story(self):
+        """
+        Tests `InvalidStoryError` is raised if story is invalid.
+        """
+        with pytest.raises(InvalidStoryError):
+            self.fetcher.fetch_meta(INVALID_STORY_PK)
+
+    def test_fetch_data_for_valid_story(self):
+        """
+        Tests data is returned if story is valid.
+        """
+        data = self.fetcher.fetch_data(VALID_STORY_PK)
+        assert len(data) != 0
+
+    def test_fetch_data_for_invalid_story(self):
+        """
+        Tests `InvalidStoryError` is raised if story is invalid.
+        """
+        with pytest.raises(InvalidStoryError):
+            self.fetcher.fetch_data(INVALID_STORY_PK)