#!/usr/bin/env python3
"""Verification test for dir_analyzer.py"""
import subprocess
import json
import sys

# Test JSON mode
result = subprocess.run(['python3', 'dir_analyzer.py', '/etc', '--json'], 
                       capture_output=True, text=True, cwd='/root/troupe-standup/wizard-work/spark')
data = json.loads(result.stdout)

assert 'total_files' in data, "Missing total_files"
assert 'total_size_bytes' in data, "Missing total_size_bytes"
assert data['total_files'] > 0, "Should find files in /etc"
assert data['total_size_bytes'] > 0, "Should have size > 0"

print(f"JSON mode: PASS - {data['total_files']} files, {data['total_size_bytes']} bytes")

# Test text mode
result2 = subprocess.run(['python3', 'dir_analyzer.py', '/tmp'],
                        capture_output=True, text=True, cwd='/root/troupe-standup/wizard-work/spark')
assert 'Total files:' in result2.stdout, "Should show file count"
assert 'Total size:' in result2.stdout, "Should show size"

print("Text mode: PASS")
print("All verification tests passed")