From d1d654ebac2d51e3841675faeb56480e440f622f Mon Sep 17 00:00:00 2001
From: Wolfgang Müller
Date: Tue, 5 Mar 2024 18:08:09 +0100
Subject: Initial commit

---
 tests/scrapers/test_scraper.py | 55 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 tests/scrapers/test_scraper.py

(limited to 'tests/scrapers/test_scraper.py')

diff --git a/tests/scrapers/test_scraper.py b/tests/scrapers/test_scraper.py
new file mode 100644
index 0000000..6f6f29d
--- /dev/null
+++ b/tests/scrapers/test_scraper.py
@@ -0,0 +1,55 @@
+from hircine.scraper import Scraper, ScrapeWarning
+
+
+class MockScraper(Scraper):
+    is_available = True
+
+    def scrape(self):
+        yield lambda: "foo"
+        yield "bar"
+
+
+class WarningScraper(Scraper):
+    is_available = True
+
+    def warn(self, str):
+        raise ScrapeWarning("Invalid input")
+
+    def scrape(self):
+        yield lambda: "foo"
+        yield lambda: self.warn("bar")
+        yield "baz"
+
+
+class ParserlessScraper(Scraper):
+    is_available = True
+
+    def scrape(self):
+        yield "literal"
+
+
+def test_scraper_collects():
+    generator = MockScraper(None).collect()
+
+    assert set(generator) == set(["foo", "bar"])
+
+
+def test_scraper_collects_with_transformer():
+    generator = MockScraper(None).collect([lambda gen, info: map(str.upper, gen)])
+
+    assert set(generator) == set(["FOO", "BAR"])
+
+
+def test_scraper_collects_warnings():
+    scraper = WarningScraper(None)
+    generator = scraper.collect()
+
+    assert set(generator) == set(["foo", "baz"])
+    assert scraper.get_warnings() == ["Invalid input"]
+
+
+def test_scraper_collects_literal():
+    scraper = ParserlessScraper(None)
+    generator = scraper.collect()
+
+    assert set(generator) == set(["literal"])
-- 
cgit v1.2.3-2-gb3c3