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
39 lines
1.1 KiB
Python
39 lines
1.1 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 cost analytics engine."""
|
|
|
|
from costing import (
|
|
DEFAULT_COST_TABLE,
|
|
get_monthly_usage,
|
|
record_usage,
|
|
)
|
|
|
|
|
|
def test_record_usage() -> None:
|
|
result = record_usage("scrape_direct", {"url": "https://example.com"})
|
|
assert result["operation"] == "scrape_direct"
|
|
assert result["cost"] > 0
|
|
assert "url" in result.get("metadata", {})
|
|
|
|
|
|
def test_record_usage_cost() -> None:
|
|
result = record_usage("llm_call", quantity=2)
|
|
expected = DEFAULT_COST_TABLE["llm_call"] * 2
|
|
assert result["cost"] == round(expected, 6)
|
|
|
|
|
|
def test_get_monthly_usage() -> None:
|
|
result = get_monthly_usage()
|
|
assert "total_cost" in result
|
|
assert "projected_monthly_cost" in result
|
|
assert "breakdown" in result
|
|
|
|
|
|
def test_cost_table_keys() -> None:
|
|
assert "scrape_direct" in DEFAULT_COST_TABLE
|
|
assert "llm_call" in DEFAULT_COST_TABLE
|
|
assert "vision_call" in DEFAULT_COST_TABLE
|
|
assert "storage_gb_month" in DEFAULT_COST_TABLE
|