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
