cryptorugmunch
0ecc250349
The duplicate file names (advanced, agency, auth, compliance, costing, etc.) in both root and routers/ made imports ambiguous. Renamed root modules to pry_<name>.py so: from quality import X -> from pry_quality import X from routers.quality import router (unchanged) Also: - Renamed x402.py to pry_x402/ package directory - Fixed 21+ bare imports across api.py, deps.py, routers/, tests/, llm_providers/ Tests: 593 passed, 1 skipped (test_ready_returns_200 fails pre-existing because Ollama is unreachable from test env, unrelated to this refactor). Audit item 8.
83 lines
2.3 KiB
Python
83 lines
2.3 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 human-in-the-loop review workflow."""
|
|
|
|
from pry_review import (
|
|
REVIEW_STATUS,
|
|
approve_review,
|
|
get_review_queue,
|
|
reject_review,
|
|
submit_for_review,
|
|
)
|
|
|
|
|
|
def test_review_status_defined() -> None:
|
|
assert "pending" in REVIEW_STATUS
|
|
assert "approved" in REVIEW_STATUS
|
|
assert "rejected" in REVIEW_STATUS
|
|
|
|
|
|
def test_submit_review() -> None:
|
|
import asyncio
|
|
|
|
result = asyncio.run(
|
|
submit_for_review(
|
|
data={"price": 999.99, "name": "Test"},
|
|
extraction_url="https://example.com",
|
|
confidence_score=0.5,
|
|
flagged_fields=[{"field": "price", "issue": "Suspicious value"}],
|
|
)
|
|
)
|
|
assert "id" in result
|
|
assert result["status"] == "pending"
|
|
assert result["confidence_score"] == 0.5
|
|
assert len(result["flagged_fields"]) == 1
|
|
|
|
|
|
def test_submit_and_approve() -> None:
|
|
import asyncio
|
|
|
|
result = asyncio.run(
|
|
submit_for_review(
|
|
data={"name": "Test"},
|
|
extraction_url="https://example.com",
|
|
)
|
|
)
|
|
review_id = result["id"]
|
|
approved = asyncio.run(approve_review(review_id, reviewer="test_user", notes="Looks good"))
|
|
assert approved["status"] == "approved"
|
|
assert approved["reviewed_by"] == "test_user"
|
|
|
|
|
|
def test_submit_and_reject() -> None:
|
|
import asyncio
|
|
|
|
result = asyncio.run(
|
|
submit_for_review(
|
|
data={"name": "Test"},
|
|
extraction_url="https://example.com",
|
|
)
|
|
)
|
|
review_id = result["id"]
|
|
rejected = asyncio.run(reject_review(review_id, reviewer="test_user", notes="Bad data"))
|
|
assert rejected["status"] == "rejected"
|
|
assert rejected["review_notes"] == "Bad data"
|
|
|
|
|
|
def test_get_review_queue() -> None:
|
|
import asyncio
|
|
|
|
asyncio.run(submit_for_review(data={"a": 1}, extraction_url="https://example.com"))
|
|
queue = get_review_queue()
|
|
assert len(queue) >= 1
|
|
assert "data" not in queue[0] # Summary should not include full payload
|
|
|
|
|
|
def test_approve_nonexistent() -> None:
|
|
import asyncio
|
|
|
|
result = asyncio.run(approve_review("nonexistent-id"))
|
|
assert "error" in result
|