Spaces:
Sleeping
Sleeping
File size: 2,983 Bytes
529b5a7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | """Tests for account data JSON consistency."""
import json
from collections import deque
from pathlib import Path
import pytest
DATA_DIR = Path(__file__).parent.parent / "env" / "data"
ACCOUNT_FILES = [
("easy_account.json", "easy", 7),
("medium_account.json", "med-", 15),
("hard_account.json", "hard", 40),
]
@pytest.fixture(params=ACCOUNT_FILES, ids=lambda x: x[0])
def account(request):
filename, prefix, expected_count = request.param
path = DATA_DIR / filename
with open(path) as f:
data = json.load(f)
return data, prefix, expected_count
def test_resource_count(account):
data, _, expected = account
assert len(data) == expected
def test_required_fields(account):
data, _, _ = account
required = ["resource_id", "resource_type", "name", "environment", "current_config",
"utilization", "monthly_cost", "has_backups", "is_critical",
"dependencies", "usage_pattern"]
for r in data:
for field in required:
assert field in r, f"{r['resource_id']} missing {field}"
def test_resource_id_format(account):
data, prefix, _ = account
for r in data:
assert r["resource_id"].startswith(f"res-{prefix}"), (
f"Bad resource_id: {r['resource_id']} (expected prefix res-{prefix})"
)
def test_no_orphaned_dependencies(account):
data, _, _ = account
ids = {r["resource_id"] for r in data}
for r in data:
for dep in r["dependencies"]:
assert dep in ids, f"{r['resource_id']} depends on non-existent {dep}"
def test_no_circular_dependencies(account):
data, _, _ = account
lookup = {r["resource_id"]: r for r in data}
for r in data:
visited = set()
queue = deque(r["dependencies"])
while queue:
dep_id = queue.popleft()
assert dep_id != r["resource_id"], f"Circular dependency: {r['resource_id']}"
if dep_id in visited:
continue
visited.add(dep_id)
if dep_id in lookup:
queue.extend(lookup[dep_id]["dependencies"])
def test_positive_costs(account):
data, _, _ = account
for r in data:
assert r["monthly_cost"] >= 0, f"{r['resource_id']} has negative cost"
def test_unique_resource_ids(account):
data, _, _ = account
ids = [r["resource_id"] for r in data]
assert len(ids) == len(set(ids)), "Duplicate resource IDs found"
def test_valid_environments(account):
data, _, _ = account
valid = {"prod", "staging", "dev"}
for r in data:
assert r["environment"] in valid, f"{r['resource_id']} has invalid env: {r['environment']}"
def test_valid_resource_types(account):
data, _, _ = account
valid = {"ec2", "rds", "s3", "load_balancer", "kubernetes", "nat_gateway", "elasticsearch", "ebs", "eip"}
for r in data:
assert r["resource_type"] in valid, f"{r['resource_id']} has invalid type: {r['resource_type']}"
|