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
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 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
|