@opslane/claude-code-game 0.1.0

package/levels/06-sub-agents/exercise/tests/test_models.py

@@ -0,0 +1,240 @@
+"""Tests for expense models.
+
+This test file includes intentionally failing tests that students fix
+during the course:
+
+- test_add_negative_amount_should_fail: Fails until Lesson 3 (validation bug)
+- test_list_by_category_case_insensitive: Fails until a later lesson (case bug)
+"""
+
+import pytest
+from datetime import datetime
+from models import (
+    add_expense,
+    list_expenses,
+    delete_expense,
+    get_expense,
+    get_total_by_category,
+    Expense
+)
+
+
+class TestAddExpense:
+    """Tests for the add_expense function."""
+
+    def test_add_basic_expense(self):
+        """Test adding a simple expense."""
+        expense = add_expense(50.0, "Food", "Lunch at cafe")
+
+        assert expense.amount == 50.0
+        assert expense.category == "Food"
+        assert expense.description == "Lunch at cafe"
+        assert expense.id is not None
+        assert len(expense.id) == 8  # UUID prefix
+
+    def test_add_expense_with_date(self):
+        """Test adding expense with specific date."""
+        date = datetime(2026, 1, 15, 12, 30)
+        expense = add_expense(25.0, "Transport", "Uber to meeting", date)
+
+        assert expense.date == date
+        assert expense.amount == 25.0
+
+    def test_add_expense_default_date(self):
+        """Test that expense gets current date if not specified."""
+        before = datetime.now()
+        expense = add_expense(10.0, "Food", "Snack")
+        after = datetime.now()
+
+        assert before <= expense.date <= after
+
+    def test_add_expense_persists(self):
+        """Test that added expense can be retrieved."""
+        expense = add_expense(75.0, "Shopping", "New book")
+
+        retrieved = get_expense(expense.id)
+        assert retrieved is not None
+        assert retrieved.amount == 75.0
+        assert retrieved.description == "New book"
+
+    # =========================================================
+    # BUG TESTS - These fail until students fix them in Lesson 3
+    # =========================================================
+
+    def test_add_negative_amount_should_fail(self):
+        """BUG TEST: Negative amounts should raise ValueError.
+
+        This test FAILS initially - students fix this in Lesson 3
+        by adding validation to add_expense().
+
+        Expected fix:
+            if amount <= 0:
+                raise ValueError("Amount must be greater than 0")
+        """
+        with pytest.raises(ValueError, match="[Aa]mount"):
+            add_expense(-50.0, "Food", "Invalid expense")
+
+    def test_add_zero_amount_should_fail(self):
+        """BUG TEST: Zero amount should raise ValueError.
+
+        This test FAILS initially - students fix this in Lesson 3.
+        """
+        with pytest.raises(ValueError, match="[Aa]mount"):
+            add_expense(0, "Food", "Free lunch doesn't exist")
+
+    def test_add_empty_category_should_fail(self):
+        """BUG TEST: Empty category should raise ValueError.
+
+        This test FAILS initially - students fix this in Lesson 3.
+        """
+        with pytest.raises(ValueError, match="[Cc]ategory"):
+            add_expense(50.0, "", "No category")
+
+    def test_add_whitespace_category_should_fail(self):
+        """BUG TEST: Whitespace-only category should raise ValueError."""
+        with pytest.raises(ValueError, match="[Cc]ategory"):
+            add_expense(50.0, "   ", "Whitespace category")
+
+
+class TestListExpenses:
+    """Tests for the list_expenses function."""
+
+    def test_list_all_expenses(self, sample_expenses):
+        """Test listing all expenses without filters."""
+        expenses = list_expenses()
+        assert len(expenses) == 5
+
+    def test_list_empty(self):
+        """Test listing when no expenses exist."""
+        expenses = list_expenses()
+        assert len(expenses) == 0
+
+    def test_list_by_category(self):
+        """Test filtering by category (exact match)."""
+        add_expense(50.0, "Food", "Lunch")
+        add_expense(30.0, "Transport", "Uber")
+        add_expense(25.0, "Food", "Coffee")
+
+        food = list_expenses(category="Food")
+        assert len(food) == 2
+        assert all(e.category == "Food" for e in food)
+
+    def test_list_by_month(self):
+        """Test filtering by month."""
+        add_expense(50.0, "Food", "Jan expense", datetime(2026, 1, 15))
+        add_expense(30.0, "Food", "Feb expense", datetime(2026, 2, 10))
+
+        jan = list_expenses(month="2026-01")
+        assert len(jan) == 1
+        assert jan[0].description == "Jan expense"
+
+    def test_list_with_limit(self):
+        """Test limiting number of results."""
+        for i in range(10):
+            add_expense(float(i + 1), "Food", f"Expense {i}")
+
+        limited = list_expenses(limit=5)
+        assert len(limited) == 5
+
+    def test_list_sorted_by_date(self):
+        """Test that results are sorted by date descending."""
+        add_expense(10.0, "Food", "Old", datetime(2026, 1, 1))
+        add_expense(20.0, "Food", "New", datetime(2026, 1, 15))
+        add_expense(30.0, "Food", "Middle", datetime(2026, 1, 10))
+
+        expenses = list_expenses()
+        dates = [e.date for e in expenses]
+        assert dates == sorted(dates, reverse=True)
+
+    # =========================================================
+    # BUG TESTS - These fail until students fix them in a later lesson
+    # =========================================================
+
+    def test_list_by_category_case_insensitive(self, mixed_case_expenses):
+        """BUG TEST: Category filter should be case-insensitive.
+
+        This test FAILS initially - students fix this in a later lesson.
+
+        The seed data includes lowercase "food" entries that won't
+        match "Food" until the bug is fixed.
+
+        Expected fix in list_expenses():
+            if category:
+                category_lower = category.lower()
+                expenses = [e for e in expenses if e.category.lower() == category_lower]
+        """
+        # mixed_case_expenses has: "Food", "food", "FOOD"
+        food = list_expenses(category="Food")
+        assert len(food) == 3, (
+            f"Expected 3 'Food' expenses (case-insensitive), got {len(food)}. "
+            "Fix: make category filtering case-insensitive."
+        )
+
+    def test_list_category_lowercase_query(self, mixed_case_expenses):
+        """BUG TEST: Lowercase category query should still find matches."""
+        food = list_expenses(category="food")  # lowercase query
+        assert len(food) == 3, "Lowercase 'food' should match all Food variants"
+
+
+class TestDeleteExpense:
+    """Tests for the delete_expense function."""
+
+    def test_delete_existing(self):
+        """Test deleting an existing expense."""
+        expense = add_expense(50.0, "Food", "To be deleted")
+
+        result = delete_expense(expense.id)
+        assert result is True
+
+        # Verify it's gone
+        assert get_expense(expense.id) is None
+
+    def test_delete_nonexistent(self):
+        """Test deleting non-existent expense returns False."""
+        result = delete_expense("fake-id-12345")
+        assert result is False
+
+    def test_delete_twice(self):
+        """Test that deleting same expense twice fails second time."""
+        expense = add_expense(50.0, "Food", "Delete me")
+
+        assert delete_expense(expense.id) is True
+        assert delete_expense(expense.id) is False
+
+
+class TestGetExpense:
+    """Tests for the get_expense function."""
+
+    def test_get_existing(self):
+        """Test getting an existing expense."""
+        created = add_expense(99.99, "Shopping", "Test item")
+
+        retrieved = get_expense(created.id)
+        assert retrieved is not None
+        assert retrieved.id == created.id
+        assert retrieved.amount == 99.99
+
+    def test_get_nonexistent(self):
+        """Test getting non-existent expense returns None."""
+        result = get_expense("nonexistent-id")
+        assert result is None
+
+
+class TestGetTotalByCategory:
+    """Tests for the get_total_by_category function."""
+
+    def test_total_single_category(self):
+        """Test total for a single category."""
+        add_expense(50.0, "Food", "Item 1")
+        add_expense(30.0, "Food", "Item 2")
+        add_expense(100.0, "Shopping", "Other category")
+
+        total = get_total_by_category("Food")
+        assert total == 80.0
+
+    def test_total_empty_category(self):
+        """Test total for category with no expenses."""
+        add_expense(50.0, "Food", "Food item")
+
+        total = get_total_by_category("Transport")
+        assert total == 0.0

package/levels/06-sub-agents/exercise/tests/test_reports.py

@@ -0,0 +1,190 @@
+"""Tests for report generation.
+
+Note: These tests will FAIL until students implement reports.py in Lesson 4.
+The tests serve as a specification for what needs to be implemented.
+"""
+
+import pytest
+from datetime import datetime
+from models import add_expense, Expense
+from reports import (
+    generate_monthly_report,
+    get_category_breakdown,
+    get_top_expenses,
+    save_report
+)
+
+
+class TestGenerateMonthlyReport:
+    """Tests for the generate_monthly_report function."""
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_report_for_empty_month(self):
+        """Test report generation for month with no expenses."""
+        report = generate_monthly_report("2020-01")
+        assert "No expenses" in report or "no expenses" in report.lower()
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_report_includes_total(self):
+        """Test that report includes total amount."""
+        add_expense(100.0, "Food", "Groceries", datetime(2026, 1, 15))
+        add_expense(50.0, "Transport", "Uber", datetime(2026, 1, 15))
+
+        report = generate_monthly_report("2026-01")
+
+        # Should contain total of $150
+        assert "150" in report
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_report_includes_categories(self):
+        """Test that report includes category breakdown."""
+        add_expense(100.0, "Food", "Groceries", datetime(2026, 1, 15))
+        add_expense(50.0, "Transport", "Uber", datetime(2026, 1, 15))
+
+        report = generate_monthly_report("2026-01")
+
+        assert "Food" in report
+        assert "Transport" in report
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_report_defaults_to_current_month(self):
+        """Test that no month argument uses current month."""
+        # This test is tricky since it depends on current date
+        # Just verify it doesn't error
+        report = generate_monthly_report()
+        assert report is not None
+        assert isinstance(report, str)
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_report_format_is_markdown(self):
+        """Test that report is formatted as markdown."""
+        add_expense(100.0, "Food", "Test", datetime(2026, 1, 15))
+        report = generate_monthly_report("2026-01")
+
+        # Should have markdown headers
+        assert "#" in report
+
+
+class TestGetCategoryBreakdown:
+    """Tests for the get_category_breakdown function."""
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_single_category(self):
+        """Test breakdown with single category."""
+        expenses = [
+            Expense("1", 50.0, "Food", "Lunch", datetime.now()),
+            Expense("2", 25.0, "Food", "Coffee", datetime.now()),
+        ]
+
+        breakdown = get_category_breakdown(expenses)
+
+        assert "Food" in breakdown
+        assert breakdown["Food"] == 75.0
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_multiple_categories(self):
+        """Test breakdown with multiple categories."""
+        expenses = [
+            Expense("1", 50.0, "Food", "Lunch", datetime.now()),
+            Expense("2", 30.0, "Transport", "Bus", datetime.now()),
+            Expense("3", 20.0, "Food", "Snack", datetime.now()),
+        ]
+
+        breakdown = get_category_breakdown(expenses)
+
+        assert breakdown["Food"] == 70.0
+        assert breakdown["Transport"] == 30.0
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_empty_list(self):
+        """Test breakdown with empty expense list."""
+        breakdown = get_category_breakdown([])
+        assert breakdown == {}
+
+
+class TestGetTopExpenses:
+    """Tests for the get_top_expenses function."""
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_top_expenses_ordering(self):
+        """Test that expenses are sorted by amount descending."""
+        expenses = [
+            Expense("1", 25.0, "Food", "Small", datetime.now()),
+            Expense("2", 100.0, "Bills", "Big", datetime.now()),
+            Expense("3", 50.0, "Shopping", "Medium", datetime.now()),
+        ]
+
+        top = get_top_expenses(expenses, 3)
+
+        assert top[0].amount == 100.0
+        assert top[1].amount == 50.0
+        assert top[2].amount == 25.0
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_top_n_limit(self):
+        """Test limiting to N expenses."""
+        expenses = [
+            Expense(str(i), float(i * 10), "Cat", "Desc", datetime.now())
+            for i in range(1, 11)
+        ]
+
+        top = get_top_expenses(expenses, 3)
+        assert len(top) == 3
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_top_with_fewer_expenses(self):
+        """Test when requesting more than available."""
+        expenses = [
+            Expense("1", 50.0, "Food", "Only one", datetime.now()),
+        ]
+
+        top = get_top_expenses(expenses, 5)
+        assert len(top) == 1
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_default_n_is_5(self):
+        """Test that default n value is 5."""
+        expenses = [
+            Expense(str(i), float(i), "Cat", "Desc", datetime.now())
+            for i in range(10)
+        ]
+
+        top = get_top_expenses(expenses)
+        assert len(top) == 5
+
+
+class TestSaveReport:
+    """Tests for the save_report function."""
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_save_creates_file(self, tmp_path, monkeypatch):
+        """Test that save_report creates a file."""
+        import os
+        monkeypatch.chdir(tmp_path)
+
+        report_content = "# Test Report\n\nThis is a test."
+        path = save_report(report_content, "test-report.md")
+
+        assert path.exists()
+        assert path.read_text() == report_content
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_save_creates_directory(self, tmp_path, monkeypatch):
+        """Test that save_report creates reports directory if needed."""
+        import os
+        monkeypatch.chdir(tmp_path)
+
+        save_report("# Test", "2026-01.md")
+
+        assert (tmp_path / "reports").is_dir()
+
+    @pytest.mark.skip(reason="Implement in Lesson 4")
+    def test_save_returns_path(self, tmp_path, monkeypatch):
+        """Test that save_report returns the file path."""
+        import os
+        monkeypatch.chdir(tmp_path)
+
+        path = save_report("# Test", "my-report.md")
+
+        assert path.name == "my-report.md"
+        assert "reports" in str(path)

package/levels/06-sub-agents/exercise/utils.py

@@ -0,0 +1,163 @@
+"""Utility functions for the expense tracker."""
+
+from datetime import datetime, timedelta
+from typing import Optional
+import locale
+
+
+def format_currency(amount: float) -> str:
+    """Format amount as currency.
+
+    Args:
+        amount: The numeric amount
+
+    Returns:
+        Formatted string like '$1,234.56'
+
+    Example:
+        >>> format_currency(1234.5)
+        '$1,234.50'
+    """
+    try:
+        locale.setlocale(locale.LC_ALL, '')
+        return locale.currency(amount, grouping=True)
+    except (locale.Error, ValueError):
+        # Fallback if locale not available
+        return f"${amount:,.2f}"
+
+
+def parse_date(date_str: Optional[str]) -> Optional[datetime]:
+    """Parse a date string in various formats.
+
+    Supports:
+    - YYYY-MM-DD (2026-01-15)
+    - MM/DD/YYYY (01/15/2026)
+    - 'today'
+    - 'yesterday'
+
+    Args:
+        date_str: The date string to parse
+
+    Returns:
+        datetime object or None if input is None
+
+    Raises:
+        ValueError: If date format is not recognized
+
+    Example:
+        >>> parse_date('2026-01-15')
+        datetime(2026, 1, 15, 0, 0, 0)
+        >>> parse_date('today')
+        datetime(...)  # Today at midnight
+    """
+    if not date_str:
+        return None
+
+    date_str = date_str.lower().strip()
+
+    # Handle relative dates
+    if date_str == 'today':
+        return datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
+
+    if date_str == 'yesterday':
+        return (datetime.now() - timedelta(days=1)).replace(
+            hour=0, minute=0, second=0, microsecond=0
+        )
+
+    # Try YYYY-MM-DD (ISO format)
+    try:
+        return datetime.strptime(date_str, '%Y-%m-%d')
+    except ValueError:
+        pass
+
+    # Try MM/DD/YYYY (US format)
+    try:
+        return datetime.strptime(date_str, '%m/%d/%Y')
+    except ValueError:
+        pass
+
+    # Try DD/MM/YYYY (European format)
+    try:
+        return datetime.strptime(date_str, '%d/%m/%Y')
+    except ValueError:
+        pass
+
+    raise ValueError(f"Could not parse date: {date_str}. Use YYYY-MM-DD format.")
+
+
+def validate_category(category: str) -> str:
+    """Normalize category name.
+
+    Capitalizes first letter, strips whitespace.
+
+    Args:
+        category: Raw category input
+
+    Returns:
+        Normalized category name
+
+    Example:
+        >>> validate_category('  food  ')
+        'Food'
+    """
+    return category.strip().capitalize()
+
+
+def format_date(dt: datetime) -> str:
+    """Format a datetime for display.
+
+    Args:
+        dt: The datetime to format
+
+    Returns:
+        Formatted string like 'Jan 15, 2026'
+    """
+    return dt.strftime('%b %d, %Y')
+
+
+def format_month(month_str: str) -> str:
+    """Format a YYYY-MM string for display.
+
+    Args:
+        month_str: Month in YYYY-MM format
+
+    Returns:
+        Formatted string like 'January 2026'
+
+    Example:
+        >>> format_month('2026-01')
+        'January 2026'
+    """
+    dt = datetime.strptime(month_str, '%Y-%m')
+    return dt.strftime('%B %Y')
+
+
+# Valid categories for validation/suggestions
+VALID_CATEGORIES = [
+    "Food",
+    "Transport",
+    "Entertainment",
+    "Shopping",
+    "Bills",
+    "Health",
+    "Education",
+    "Travel",
+    "Other"
+]
+
+
+def suggest_category(partial: str) -> list[str]:
+    """Suggest categories matching a partial input.
+
+    Args:
+        partial: Partial category name
+
+    Returns:
+        List of matching categories
+
+    Example:
+        >>> suggest_category('fo')
+        ['Food']
+    """
+    partial_lower = partial.lower()
+    return [cat for cat in VALID_CATEGORIES if cat.lower().startswith(partial_lower)]

package/levels/06-sub-agents/lesson.yaml

@@ -0,0 +1,49 @@
+id: "sub-agents"
+number: 6
+module: "Building Blocks"
+title: "Sub-Agents for Context Isolation"
+
+video:
+  url: "https://youtu.be/jB_p_Pphb_E"
+  duration_seconds: 300
+
+exercise:
+  intro: |
+    Use a sub-agent to explore the codebase without polluting your main context.
+  objective: "Use sub-agents to keep your main context clean"
+
+intro: |
+  Sub-agents = fresh context window. Parent gets a clean summary.
+
+  Remember Lesson 1? Context is finite. Sub-agents solve that.
+
+  You want to understand how the reporting module works, but don't
+  want all those details filling your main context.
+
+  Type: claude
+
+  First, run /context to see your current context usage.
+
+  Say: "Use a sub-agent to analyze reports.py and summarize how it
+  generates reports. I just want a summary, not all the details."
+
+  After getting the summary, run /context again.
+  Your main context should barely have moved.
+
+verification:
+  - type: message_exists
+  - type: tool_called
+    tool_name: Task
+
+success: |
+  You used sub-agents for context isolation!
+
+  The sub-agent explored reports.py in its own context.
+  You got a clean summary without the noise.
+
+  Sub-agents are for research and exploration.
+  Keep your main session clean for actual work.
+
+limits:
+  max_duration_minutes: 15
+  max_claude_messages: 20