backend/tests: Add tests for the anchira scraper

1 files changed, 107 insertions, 0 deletions

diff --git a/tests/plugins/scrapers/test_anchira.py b/tests/plugins/scrapers/test_anchira.py
new file mode 100644
index 0000000..0554957
--- /dev/null
+++ b/tests/plugins/scrapers/test_anchira.py
@@ -0,0 +1,107 @@
+import os
+from datetime import date
+from zipfile import ZipFile
+
+import pytest
+
+import hircine.enums as enums
+from hircine.plugins.scrapers.anchira import AnchiraYamlScraper
+from hircine.scraper.types import (
+    URL,
+    Artist,
+    Censorship,
+    Circle,
+    Date,
+    Direction,
+    Language,
+    Rating,
+    Tag,
+    Title,
+    World,
+)
+
+
+@pytest.fixture
+def archive_file(tmpdir):
+    file = os.path.join(tmpdir, "archive.zip")
+
+    data = """
+---
+Source: https://anchira.to/g/1/1
+URL: https://example.com
+Title: Example Title
+Artist:
+- Example
+Circle:
+- Example
+Parody:
+- Original Work
+- Example
+Magazine: []
+Tags:
+- Unlimited
+- Book
+Released: 1574394240
+Pages: 102
+...
+    """
+
+    with ZipFile(file, "x") as ziph:
+        ziph.writestr("info.yaml", data)
+
+    yield file
+
+
+def test_does_scrape(monkeypatch, archive_file, gen_comic):
+    comic = next(gen_comic)
+    comic.archive.path = archive_file
+
+    scraper = AnchiraYamlScraper(comic)
+
+    assert scraper.is_available
+    assert scraper.source == AnchiraYamlScraper.source
+    assert scraper.name == "anchira.to info.yaml"
+
+    assert set(scraper.collect()) == set(
+        [
+            Artist(name="Example"),
+            Circle(name="Example"),
+            Date(value=date(2019, 11, 22)),
+            Direction(value=enums.Direction.RIGHT_TO_LEFT),
+            Language(value=enums.Language.EN),
+            Tag(namespace="none", tag="Book"),
+            Title(value="Example Title"),
+            URL(value="https://example.com"),
+            World(name="Example"),
+        ]
+    )
+
+
+def test_does_not_scrape_on_error(tmpdir, gen_comic):
+    comic = next(gen_comic)
+    comic.archive.path = os.path.join(tmpdir, "nonexistent.zip")
+
+    scraper = AnchiraYamlScraper(comic)
+
+    assert scraper.data == {}
+    assert not scraper.is_available
+
+
+@pytest.mark.parametrize(
+    "tag, parsed",
+    [
+        ("Hentai", Rating(value=enums.Rating.EXPLICIT)),
+        ("Non-H", Rating(value=enums.Rating.QUESTIONABLE)),
+        ("Ecchi", Rating(value=enums.Rating.QUESTIONABLE)),
+        ("Uncensored", Censorship(value=enums.Censorship.NONE)),
+    ],
+    ids=[
+        "hentai",
+        "non-h",
+        "ecchi",
+        "uncensored",
+    ],
+)
+def test_parses_tags(tag, parsed):
+    scraper = AnchiraYamlScraper(None)
+    assert scraper.parse_tag(tag) == parsed