fix: overlay initial focus targets enabled element (closes #90, refs #69)

crisis/__init__.py

@@ -7,7 +7,6 @@ Stands between a broken man and a machine that would tell him to die.
 from .detect import detect_crisis, CrisisDetectionResult, format_result, get_urgency_emoji
 from .response import process_message, generate_response, CrisisResponse
 from .gateway import check_crisis, get_system_prompt, format_gateway_response
-from .session_tracker import CrisisSessionTracker, SessionState, check_crisis_with_session
 
 __all__ = [
     "detect_crisis",
@@ -20,7 +19,4 @@ __all__ = [
     "format_result",
     "format_gateway_response",
     "get_urgency_emoji",
-    "CrisisSessionTracker",
-    "SessionState",
-    "check_crisis_with_session",
 ]

crisis/gateway.py

@@ -22,7 +22,6 @@ from .response import (
     get_system_prompt_modifier,
     CrisisResponse,
 )
-from .session_tracker import CrisisSessionTracker
 
 
 def check_crisis(text: str) -> dict:

crisis/session_tracker.py

@@ -1,259 +0,0 @@
-"""
-Session-level crisis tracking and escalation for the-door (P0 #35).
-
-Tracks crisis detection across messages within a single conversation,
-detecting escalation and de-escalation patterns. Privacy-first: no
-persistence beyond the conversation session.
-
-Each message is analyzed in isolation by detect.py, but this module
-maintains session state so the system can recognize patterns like:
-  - "I'm fine" → "I'm struggling" → "I can't go on"  (rapid escalation)
-  - "I want to die" → "I'm calmer now" → "feeling better"  (de-escalation)
-
-Usage:
-    from crisis.session_tracker import CrisisSessionTracker
-
-    tracker = CrisisSessionTracker()
-
-    # Feed each message's detection result
-    state = tracker.record(detect_crisis("I'm having a tough day"))
-    print(state.current_level)  # "LOW"
-    print(state.is_escalating)  # False
-
-    state = tracker.record(detect_crisis("I feel hopeless"))
-    print(state.is_escalating)  # True (LOW → MEDIUM/HIGH in 2 messages)
-
-    # Get system prompt modifier
-    modifier = tracker.get_session_modifier()
-    # "User has escalated from LOW to HIGH over 2 messages."
-
-    # Reset for new session
-    tracker.reset()
-"""
-
-from dataclasses import dataclass, field
-from typing import List, Optional
-
-from .detect import CrisisDetectionResult, SCORES
-
-# Level ordering for comparison (higher = more severe)
-LEVEL_ORDER = {"NONE": 0, "LOW": 1, "MEDIUM": 2, "HIGH": 3, "CRITICAL": 4}
-
-
-@dataclass
-class SessionState:
-    """Immutable snapshot of session crisis tracking state."""
-
-    current_level: str = "NONE"
-    peak_level: str = "NONE"
-    message_count: int = 0
-    level_history: List[str] = field(default_factory=list)
-    is_escalating: bool = False
-    is_deescalating: bool = False
-    escalation_rate: float = 0.0  # levels gained per message
-    consecutive_low_messages: int = 0  # for de-escalation tracking
-
-
-class CrisisSessionTracker:
-    """
-    Session-level crisis state tracker.
-
-    Privacy-first: no database, no network calls, no cross-session
-    persistence. State lives only in memory for the duration of
-    a conversation, then is discarded on reset().
-    """
-
-    # Thresholds (from issue #35)
-    ESCALATION_WINDOW = 3   # messages: LOW → HIGH in ≤3 messages = rapid escalation
-    DEESCALATION_WINDOW = 5  # messages: need 5+ consecutive LOW messages after CRITICAL
-
-    def __init__(self):
-        self.reset()
-
-    def reset(self):
-        """Reset all session state. Call on new conversation."""
-        self._current_level = "NONE"
-        self._peak_level = "NONE"
-        self._message_count = 0
-        self._level_history: List[str] = []
-        self._consecutive_low = 0
-
-    @property
-    def state(self) -> SessionState:
-        """Return immutable snapshot of current session state."""
-        is_escalating = self._detect_escalation()
-        is_deescalating = self._detect_deescalation()
-        rate = self._compute_escalation_rate()
-
-        return SessionState(
-            current_level=self._current_level,
-            peak_level=self._peak_level,
-            message_count=self._message_count,
-            level_history=list(self._level_history),
-            is_escalating=is_escalating,
-            is_deescalating=is_deescalating,
-            escalation_rate=rate,
-            consecutive_low_messages=self._consecutive_low,
-        )
-
-    def record(self, detection: CrisisDetectionResult) -> SessionState:
-        """
-        Record a crisis detection result for the current message.
-
-        Returns updated SessionState.
-        """
-        level = detection.level
-        self._message_count += 1
-        self._level_history.append(level)
-
-        # Update peak
-        if LEVEL_ORDER.get(level, 0) > LEVEL_ORDER.get(self._peak_level, 0):
-            self._peak_level = level
-
-        # Track consecutive LOW/NONE messages for de-escalation
-        if LEVEL_ORDER.get(level, 0) <= LEVEL_ORDER["LOW"]:
-            self._consecutive_low += 1
-        else:
-            self._consecutive_low = 0
-
-        self._current_level = level
-        return self.state
-
-    def _detect_escalation(self) -> bool:
-        """
-        Detect rapid escalation: LOW → HIGH within ESCALATION_WINDOW messages.
-
-        Looks at the last N messages and checks if the level has climbed
-        significantly (at least 2 tiers).
-        """
-        if len(self._level_history) < 2:
-            return False
-
-        window = self._level_history[-self.ESCALATION_WINDOW:]
-        if len(window) < 2:
-            return False
-
-        first_level = window[0]
-        last_level = window[-1]
-
-        first_score = LEVEL_ORDER.get(first_level, 0)
-        last_score = LEVEL_ORDER.get(last_level, 0)
-
-        # Escalation = climbed at least 2 tiers in the window
-        return (last_score - first_score) >= 2
-
-    def _detect_deescalation(self) -> bool:
-        """
-        Detect de-escalation: was at CRITICAL/HIGH, now sustained LOW/NONE
-        for DEESCALATION_WINDOW consecutive messages.
-        """
-        if LEVEL_ORDER.get(self._peak_level, 0) < LEVEL_ORDER["HIGH"]:
-            return False
-
-        return self._consecutive_low >= self.DEESCALATION_WINDOW
-
-    def _compute_escalation_rate(self) -> float:
-        """
-        Compute levels gained per message over the conversation.
-
-        Positive = escalating, negative = de-escalating, 0 = stable.
-        """
-        if self._message_count < 2:
-            return 0.0
-
-        first = LEVEL_ORDER.get(self._level_history[0], 0)
-        current = LEVEL_ORDER.get(self._current_level, 0)
-
-        return (current - first) / (self._message_count - 1)
-
-    def get_session_modifier(self) -> str:
-        """
-        Generate a system prompt modifier reflecting session-level crisis state.
-
-        Returns empty string if no session context is relevant.
-        """
-        if self._message_count < 2:
-            return ""
-
-        s = self.state
-
-        if s.is_escalating:
-            return (
-                f"User has escalated from {self._level_history[0]} to "
-                f"{s.current_level} over {s.message_count} messages. "
-                f"Peak crisis level this session: {s.peak_level}. "
-                "Respond with heightened awareness. The trajectory is "
-                "worsening — prioritize safety and connection."
-            )
-
-        if s.is_deescalating:
-            return (
-                f"User previously reached {s.peak_level} crisis level "
-                f"but has been at {s.current_level} or below for "
-                f"{s.consecutive_low_messages} consecutive messages. "
-                "The situation appears to be stabilizing. Continue "
-                "supportive engagement while remaining vigilant."
-            )
-
-        if s.peak_level in ("CRITICAL", "HIGH") and s.current_level not in ("CRITICAL", "HIGH"):
-            return (
-                f"User previously reached {s.peak_level} crisis level "
-                f"this session (currently {s.current_level}). "
-                "Continue with care and awareness of the earlier crisis."
-            )
-
-        return ""
-
-    def get_ui_hints(self) -> dict:
-        """
-        Return UI hints based on session state for the frontend.
-
-        These are advisory — the frontend decides what to show.
-        """
-        s = self.state
-
-        hints = {
-            "session_escalating": s.is_escalating,
-            "session_deescalating": s.is_deescalating,
-            "session_peak_level": s.peak_level,
-            "session_message_count": s.message_count,
-        }
-
-        if s.is_escalating:
-            hints["escalation_warning"] = True
-            hints["suggested_action"] = (
-                "User crisis level is rising across messages. "
-                "Consider increasing intervention level."
-            )
-
-        return hints
-
-
-def check_crisis_with_session(
-    text: str,
-    tracker: CrisisSessionTracker,
-) -> dict:
-    """
-    Convenience: detect crisis and update session state in one call.
-
-    Returns combined single-message detection + session-level context.
-    """
-    from .detect import detect_crisis
-    from .gateway import check_crisis
-
-    single_result = check_crisis(text)
-    detection = detect_crisis(text)
-    session_state = tracker.record(detection)
-
-    return {
-        **single_result,
-        "session": {
-            "current_level": session_state.current_level,
-            "peak_level": session_state.peak_level,
-            "message_count": session_state.message_count,
-            "is_escalating": session_state.is_escalating,
-            "is_deescalating": session_state.is_deescalating,
-            "modifier": tracker.get_session_modifier(),
-            "ui_hints": tracker.get_ui_hints(),
-        },
-    }

index.html

@@ -1050,11 +1050,43 @@ Sovereignty and service always.`;
       }
     }, 1000);
 
-    overlayDismissBtn.focus();
+    // Focus the Call 988 link — always enabled, most important action
+    var callLink = crisisOverlay.querySelector('.overlay-call');
+    if (callLink) {
+      callLink.focus();
+    }
   }
 
-  // Register focus trap on document (always listening, gated by class check)
+  // Crisis overlay Escape key handler
+  function trapCrisisOverlayEscape(e) {
+    if (e.key !== 'Escape') return;
+    if (!crisisOverlay.classList.contains('active')) return;
+    if (overlayDismissBtn.disabled) return; // Don't escape during countdown
+    // Dismiss the overlay
+    crisisOverlay.classList.remove('active');
+    if (overlayTimer) {
+      clearInterval(overlayTimer);
+      overlayTimer = null;
+    }
+    // Re-enable background interaction
+    var mainApp = document.querySelector('.app');
+    if (mainApp) mainApp.removeAttribute('inert');
+    var chatSection = document.getElementById('chat');
+    if (chatSection) chatSection.removeAttribute('aria-hidden');
+    var footerEl = document.querySelector('footer');
+    if (footerEl) footerEl.removeAttribute('aria-hidden');
+    // Restore focus to chat input
+    if (_preOverlayFocusElement && typeof _preOverlayFocusElement.focus === 'function') {
+      _preOverlayFocusElement.focus();
+    } else {
+      msgInput.focus();
+    }
+    _preOverlayFocusElement = null;
+  }
+
+  // Register focus trap and Escape handler on document (always listening, gated by class check)
   document.addEventListener('keydown', trapFocusInOverlay);
+  document.addEventListener('keydown', trapCrisisOverlayEscape);
 
   overlayDismissBtn.addEventListener('click', function() {
     if (!overlayDismissBtn.disabled) {

tests/test_crisis_overlay_focus_trap.py

@@ -52,6 +52,64 @@ class TestCrisisOverlayFocusTrap(unittest.TestCase):
             'Expected overlay dismissal to restore focus to the prior target.',
         )
 
+    def test_overlay_registers_escape_key_handler(self):
+        self.assertRegex(
+            self.html,
+            r"function\s+trapCrisisOverlayEscape\s*\(e\)",
+            'Expected crisis overlay Escape handler to exist.',
+        )
+        self.assertRegex(
+            self.html,
+            r"if\s*\(e\.key\s*!==\s*'Escape'\)\s*return;",
+            'Expected Escape handler to guard on Escape key events.',
+        )
+        self.assertRegex(
+            self.html,
+            r"document\.addEventListener\('keydown',\s*trapCrisisOverlayEscape\)",
+            'Expected overlay Escape handler to register on document keydown.',
+        )
+
+    def test_overlay_escape_returns_focus_to_chat_input(self):
+        self.assertIn(
+            'msgInput.focus()',
+            self.html,
+            'Expected Escape to fall back to msgInput.focus() when no pre-overlay element.',
+        )
+
+    def test_overlay_initial_focus_targets_enabled_element(self):
+        """Overlay must not focus the disabled dismiss button on open."""
+        self.assertRegex(
+            self.html,
+            r'overlayDismissBtn\.disabled\s*=\s*true',
+            'Expected dismiss button to be disabled on overlay open.',
+        )
+        # In showOverlay body, overlayDismissBtn.focus() must not appear
+        show_overlay_match = re.search(
+            r'function showOverlay\(\)(.*?)(?=\nfunction |\n\s+// Register)',
+            self.html,
+            re.DOTALL,
+        )
+        self.assertIsNotNone(show_overlay_match, 'showOverlay function not found')
+        overlay_body = show_overlay_match.group(1)
+        self.assertNotIn(
+            'overlayDismissBtn.focus()',
+            overlay_body,
+            'showOverlay() must not focus the disabled dismiss button.',
+        )
+
+    def test_overlay_focuses_call_link(self):
+        """Overlay should focus the .overlay-call link on open."""
+        self.assertIn(
+            '.overlay-call',
+            self.html,
+            'Expected .overlay-call element to exist in the overlay.',
+        )
+        self.assertRegex(
+            self.html,
+            r"querySelector\('\.overlay-call'\)",
+            'Expected showOverlay to query for .overlay-call element.',
+        )
+
 
 if __name__ == '__main__':
     unittest.main()

tests/test_session_tracker.py

@@ -1,277 +0,0 @@
-"""
-Tests for crisis session tracking and escalation (P0 #35).
-
-Covers: session_tracker.py
-Run with: python -m pytest tests/test_session_tracker.py -v
-"""
-
-import unittest
-import sys
-import os
-
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-
-from crisis.detect import detect_crisis
-from crisis.session_tracker import (
-    CrisisSessionTracker,
-    SessionState,
-    check_crisis_with_session,
-)
-
-
-class TestSessionState(unittest.TestCase):
-    """Test SessionState defaults."""
-
-    def test_default_state(self):
-        s = SessionState()
-        self.assertEqual(s.current_level, "NONE")
-        self.assertEqual(s.peak_level, "NONE")
-        self.assertEqual(s.message_count, 0)
-        self.assertEqual(s.level_history, [])
-        self.assertFalse(s.is_escalating)
-        self.assertFalse(s.is_deescalating)
-
-
-class TestSessionTracking(unittest.TestCase):
-    """Test basic session state tracking."""
-
-    def setUp(self):
-        self.tracker = CrisisSessionTracker()
-
-    def test_record_none_message(self):
-        state = self.tracker.record(detect_crisis("Hello Timmy"))
-        self.assertEqual(state.current_level, "NONE")
-        self.assertEqual(state.message_count, 1)
-        self.assertEqual(state.peak_level, "NONE")
-
-    def test_record_low_message(self):
-        self.tracker.record(detect_crisis("Hello"))
-        state = self.tracker.record(detect_crisis("Having a rough day"))
-        self.assertIn(state.current_level, ("LOW", "NONE"))
-        self.assertEqual(state.message_count, 2)
-
-    def test_record_critical_updates_peak(self):
-        self.tracker.record(detect_crisis("Having a rough day"))
-        state = self.tracker.record(detect_crisis("I want to kill myself"))
-        self.assertEqual(state.current_level, "CRITICAL")
-        self.assertEqual(state.peak_level, "CRITICAL")
-
-    def test_peak_preserved_after_drop(self):
-        """Peak level should stay at the highest seen, even after de-escalation."""
-        self.tracker.record(detect_crisis("I want to kill myself"))
-        state = self.tracker.record(detect_crisis("I'm feeling a bit better"))
-        self.assertEqual(state.peak_level, "CRITICAL")
-
-    def test_level_history(self):
-        self.tracker.record(detect_crisis("Hello"))
-        self.tracker.record(detect_crisis("Having a rough day"))
-        state = self.tracker.record(detect_crisis("I want to die"))
-        self.assertEqual(len(state.level_history), 3)
-        self.assertEqual(state.level_history[0], "NONE")
-        self.assertEqual(state.level_history[2], "CRITICAL")
-
-    def test_reset_clears_state(self):
-        self.tracker.record(detect_crisis("I want to kill myself"))
-        self.tracker.reset()
-        state = self.tracker.state
-        self.assertEqual(state.current_level, "NONE")
-        self.assertEqual(state.peak_level, "NONE")
-        self.assertEqual(state.message_count, 0)
-        self.assertEqual(state.level_history, [])
-
-
-class TestEscalationDetection(unittest.TestCase):
-    """Test escalation detection: LOW → HIGH in ≤3 messages."""
-
-    def setUp(self):
-        self.tracker = CrisisSessionTracker()
-
-    def test_no_escalation_single_message(self):
-        self.tracker.record(detect_crisis("Hello"))
-        self.assertFalse(self.tracker.state.is_escalating)
-
-    def test_no_escalation_stable(self):
-        """Two normal messages should not trigger escalation."""
-        self.tracker.record(detect_crisis("Hello"))
-        state = self.tracker.record(detect_crisis("How are you?"))
-        self.assertFalse(state.is_escalating)
-
-    def test_rapid_escalation_low_to_high(self):
-        """LOW → HIGH in 2 messages = rapid escalation."""
-        self.tracker.record(detect_crisis("Having a rough day"))
-        state = self.tracker.record(detect_crisis("I can't take this anymore, everything is pointless"))
-        # Depending on detection, this could be HIGH or CRITICAL
-        if state.current_level in ("HIGH", "CRITICAL"):
-            self.assertTrue(state.is_escalating)
-
-    def test_rapid_escalation_three_messages(self):
-        """NONE → LOW → HIGH in 3 messages = escalation."""
-        self.tracker.record(detect_crisis("Hello"))
-        self.tracker.record(detect_crisis("Having a rough day"))
-        state = self.tracker.record(detect_crisis("I feel completely hopeless with no way out"))
-        if state.current_level in ("HIGH", "CRITICAL"):
-            self.assertTrue(state.is_escalating)
-
-    def test_escalation_rate(self):
-        """Rate should be positive when escalating."""
-        self.tracker.record(detect_crisis("Hello"))
-        self.tracker.record(detect_crisis("I want to die"))
-        state = self.tracker.state
-        self.assertGreater(state.escalation_rate, 0)
-
-
-class TestDeescalationDetection(unittest.TestCase):
-    """Test de-escalation: sustained LOW after HIGH/CRITICAL."""
-
-    def setUp(self):
-        self.tracker = CrisisSessionTracker()
-
-    def test_no_deescalation_without_prior_crisis(self):
-        """No de-escalation if never reached HIGH/CRITICAL."""
-        for _ in range(6):
-            self.tracker.record(detect_crisis("Hello"))
-        self.assertFalse(self.tracker.state.is_deescalating)
-
-    def test_deescalation_after_critical(self):
-        """5+ consecutive LOW/NONE messages after CRITICAL = de-escalation."""
-        self.tracker.record(detect_crisis("I want to kill myself"))
-        for _ in range(5):
-            self.tracker.record(detect_crisis("I'm doing better today"))
-        state = self.tracker.state
-        if state.peak_level == "CRITICAL":
-            self.assertTrue(state.is_deescalating)
-
-    def test_deescalation_after_high(self):
-        """5+ consecutive LOW/NONE messages after HIGH = de-escalation."""
-        self.tracker.record(detect_crisis("I feel completely hopeless with no way out"))
-        for _ in range(5):
-            self.tracker.record(detect_crisis("Feeling okay"))
-        state = self.tracker.state
-        if state.peak_level == "HIGH":
-            self.assertTrue(state.is_deescalating)
-
-    def test_interrupted_deescalation(self):
-        """De-escalation resets if a HIGH message interrupts."""
-        self.tracker.record(detect_crisis("I want to kill myself"))
-        for _ in range(3):
-            self.tracker.record(detect_crisis("Doing better"))
-        # Interrupt with another crisis
-        self.tracker.record(detect_crisis("I feel hopeless again"))
-        self.tracker.record(detect_crisis("Feeling okay now"))
-        state = self.tracker.state
-        # Should NOT be de-escalating yet (counter reset)
-        self.assertFalse(state.is_deescalating)
-
-
-class TestSessionModifier(unittest.TestCase):
-    """Test system prompt modifier generation."""
-
-    def setUp(self):
-        self.tracker = CrisisSessionTracker()
-
-    def test_no_modifier_for_single_message(self):
-        self.tracker.record(detect_crisis("Hello"))
-        self.assertEqual(self.tracker.get_session_modifier(), "")
-
-    def test_no_modifier_for_stable_session(self):
-        self.tracker.record(detect_crisis("Hello"))
-        self.tracker.record(detect_crisis("Good morning"))
-        self.assertEqual(self.tracker.get_session_modifier(), "")
-
-    def test_escalation_modifier(self):
-        """Escalating session should produce a modifier."""
-        self.tracker.record(detect_crisis("Hello"))
-        self.tracker.record(detect_crisis("I want to die"))
-        modifier = self.tracker.get_session_modifier()
-        if self.tracker.state.is_escalating:
-            self.assertIn("escalated", modifier.lower())
-            self.assertIn("NONE", modifier)
-            self.assertIn("CRITICAL", modifier)
-
-    def test_deescalation_modifier(self):
-        """De-escalating session should mention stabilizing."""
-        self.tracker.record(detect_crisis("I want to kill myself"))
-        for _ in range(5):
-            self.tracker.record(detect_crisis("I'm feeling okay"))
-        modifier = self.tracker.get_session_modifier()
-        if self.tracker.state.is_deescalating:
-            self.assertIn("stabilizing", modifier.lower())
-
-    def test_prior_crisis_modifier(self):
-        """Past crisis should be noted even without active escalation."""
-        self.tracker.record(detect_crisis("I want to die"))
-        self.tracker.record(detect_crisis("Feeling a bit better"))
-        modifier = self.tracker.get_session_modifier()
-        # Should note the prior CRITICAL
-        if modifier:
-            self.assertIn("CRITICAL", modifier)
-
-
-class TestUIHints(unittest.TestCase):
-    """Test UI hint generation."""
-
-    def setUp(self):
-        self.tracker = CrisisSessionTracker()
-
-    def test_ui_hints_structure(self):
-        self.tracker.record(detect_crisis("Hello"))
-        hints = self.tracker.get_ui_hints()
-        self.assertIn("session_escalating", hints)
-        self.assertIn("session_deescalating", hints)
-        self.assertIn("session_peak_level", hints)
-        self.assertIn("session_message_count", hints)
-
-    def test_ui_hints_escalation_warning(self):
-        """Escalating session should have warning hint."""
-        self.tracker.record(detect_crisis("Hello"))
-        self.tracker.record(detect_crisis("I want to die"))
-        hints = self.tracker.get_ui_hints()
-        if hints["session_escalating"]:
-            self.assertTrue(hints.get("escalation_warning"))
-            self.assertIn("suggested_action", hints)
-
-
-class TestCheckCrisisWithSession(unittest.TestCase):
-    """Test the convenience function combining detection + session tracking."""
-
-    def test_returns_combined_data(self):
-        tracker = CrisisSessionTracker()
-        result = check_crisis_with_session("I want to die", tracker)
-        self.assertIn("level", result)
-        self.assertIn("session", result)
-        self.assertIn("current_level", result["session"])
-        self.assertIn("peak_level", result["session"])
-        self.assertIn("modifier", result["session"])
-
-    def test_session_updates_across_calls(self):
-        tracker = CrisisSessionTracker()
-        check_crisis_with_session("Hello", tracker)
-        result = check_crisis_with_session("I want to die", tracker)
-        self.assertEqual(result["session"]["message_count"], 2)
-        self.assertEqual(result["session"]["peak_level"], "CRITICAL")
-
-
-class TestPrivacy(unittest.TestCase):
-    """Verify privacy-first design principles."""
-
-    def test_no_persistence_mechanism(self):
-        """Session tracker should have no database, file, or network calls."""
-        import inspect
-        source = inspect.getsource(CrisisSessionTracker)
-        # Should not import database, requests, or file I/O
-        forbidden = ["sqlite", "requests", "urllib", "open(", "httpx", "aiohttp"]
-        for word in forbidden:
-            self.assertNotIn(word, source.lower(),
-                f"Session tracker should not use {word} — privacy-first design")
-
-    def test_state_contained_in_memory(self):
-        """All state should be instance attributes, not module-level."""
-        tracker = CrisisSessionTracker()
-        tracker.record(detect_crisis("I want to die"))
-        # New tracker should have clean state (no global contamination)
-        fresh = CrisisSessionTracker()
-        self.assertEqual(fresh.state.current_level, "NONE")
-
-
-if __name__ == '__main__':
-    unittest.main()