cryptorugmunch
8d25702eca
Squashed from chore/license-relicense. Full message preserved in the original branch commitbb77eb5. See ADR-0002 for the decision rationale. Refs: ADR-0002, commitbb77eb5
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2026 Rug Munch Media LLC
|
|
#
|
|
# Part of Pry — https://git.rugmunch.io/RugMunchMedia/pryscraper
|
|
# Licensed under MIT. See LICENSE.
|
|
"""Tests for scraper templates."""
|
|
|
|
from template_engine import get_template, list_templates
|
|
|
|
|
|
def test_list_templates() -> None:
|
|
templates = list_templates()
|
|
assert len(templates) >= 40 # At least 40 templates
|
|
assert any("amazon" in t["id"] for t in templates)
|
|
assert any("linkedin" in t["id"] for t in templates)
|
|
assert any("github" in t["id"] for t in templates)
|
|
|
|
|
|
def test_get_amazon_template() -> None:
|
|
t = get_template("amazon-product")
|
|
assert t is not None
|
|
assert t["name"] == "Amazon Product"
|
|
assert "schema" in t
|
|
assert len(t["schema"]["fields"]) > 5
|
|
|
|
|
|
def test_get_linkedin_template() -> None:
|
|
t = get_template("linkedin-profile")
|
|
assert t is not None
|
|
assert "profile" in t["name"].lower()
|
|
|
|
|
|
def test_get_nonexistent() -> None:
|
|
t = get_template("nonexistent-site-12345")
|
|
assert t is None
|
|
|
|
|
|
def test_templates_have_categories() -> None:
|
|
templates = list_templates()
|
|
categories = {t["category"] for t in templates}
|
|
assert "ecommerce" in categories
|
|
assert "professional" in categories
|
|
|
|
|
|
def test_templates_have_icons() -> None:
|
|
templates = list_templates()
|
|
for t in templates:
|
|
assert t.get("icon"), f"Template {t['id']} missing icon"
|