111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test suite for habits.json schema validation
|
|
"""
|
|
import json
|
|
import os
|
|
from datetime import datetime
|
|
|
|
|
|
def test_habits_file_exists():
|
|
"""Test that habits.json file exists"""
|
|
assert os.path.exists('dashboard/habits.json'), "habits.json should exist in dashboard/"
|
|
print("✓ habits.json file exists")
|
|
|
|
|
|
def test_valid_json():
|
|
"""Test that habits.json is valid JSON"""
|
|
try:
|
|
with open('dashboard/habits.json', 'r') as f:
|
|
data = json.load(f)
|
|
print("✓ habits.json is valid JSON")
|
|
return data
|
|
except json.JSONDecodeError as e:
|
|
raise AssertionError(f"habits.json is not valid JSON: {e}")
|
|
|
|
|
|
def test_root_structure(data):
|
|
"""Test that root structure has required fields"""
|
|
assert 'lastUpdated' in data, "Root should have 'lastUpdated' field"
|
|
assert 'habits' in data, "Root should have 'habits' field"
|
|
print("✓ Root structure has lastUpdated and habits fields")
|
|
|
|
|
|
def test_last_updated_format(data):
|
|
"""Test that lastUpdated is a valid ISO timestamp"""
|
|
try:
|
|
datetime.fromisoformat(data['lastUpdated'].replace('Z', '+00:00'))
|
|
print("✓ lastUpdated is valid ISO timestamp")
|
|
except (ValueError, AttributeError) as e:
|
|
raise AssertionError(f"lastUpdated is not a valid ISO timestamp: {e}")
|
|
|
|
|
|
def test_habits_is_array(data):
|
|
"""Test that habits is an array"""
|
|
assert isinstance(data['habits'], list), "habits should be an array"
|
|
print("✓ habits is an array")
|
|
|
|
|
|
def test_habit_schema():
|
|
"""Test habit schema structure with sample data"""
|
|
# Sample habit to validate schema
|
|
sample_habit = {
|
|
"id": "habit-123",
|
|
"name": "Bazin",
|
|
"frequency": "daily",
|
|
"createdAt": "2026-02-10T10:57:00.000Z",
|
|
"completions": ["2026-02-10T10:00:00.000Z", "2026-02-09T10:00:00.000Z"]
|
|
}
|
|
|
|
# Validate required fields
|
|
required_fields = ['id', 'name', 'frequency', 'createdAt', 'completions']
|
|
for field in required_fields:
|
|
assert field in sample_habit, f"Habit should have '{field}' field"
|
|
|
|
# Validate types
|
|
assert isinstance(sample_habit['id'], str), "id should be string"
|
|
assert isinstance(sample_habit['name'], str), "name should be string"
|
|
assert sample_habit['frequency'] in ['daily', 'weekly'], "frequency should be 'daily' or 'weekly'"
|
|
assert isinstance(sample_habit['completions'], list), "completions should be array"
|
|
|
|
# Validate ISO dates
|
|
datetime.fromisoformat(sample_habit['createdAt'].replace('Z', '+00:00'))
|
|
for completion in sample_habit['completions']:
|
|
datetime.fromisoformat(completion.replace('Z', '+00:00'))
|
|
|
|
print("✓ Habit schema structure is valid")
|
|
|
|
|
|
def test_initial_state(data):
|
|
"""Test that initial file has empty habits array"""
|
|
assert len(data['habits']) == 0, "Initial habits array should be empty"
|
|
print("✓ Initial habits array is empty")
|
|
|
|
|
|
def run_all_tests():
|
|
"""Run all schema validation tests"""
|
|
print("Running habits.json schema validation tests...\n")
|
|
|
|
try:
|
|
test_habits_file_exists()
|
|
data = test_valid_json()
|
|
test_root_structure(data)
|
|
test_last_updated_format(data)
|
|
test_habits_is_array(data)
|
|
test_habit_schema()
|
|
test_initial_state(data)
|
|
|
|
print("\n✅ All tests passed!")
|
|
return True
|
|
except AssertionError as e:
|
|
print(f"\n❌ Test failed: {e}")
|
|
return False
|
|
except Exception as e:
|
|
print(f"\n❌ Unexpected error: {e}")
|
|
return False
|
|
|
|
|
|
if __name__ == '__main__':
|
|
success = run_all_tests()
|
|
exit(0 if success else 1)
|