Compare commits
4 Commits
fix/744
...
feat/673-9
| Author | SHA1 | Date | |
|---|---|---|---|
| 0aef65cd52 | |||
| 718adbea6a | |||
| 938a526a5e | |||
| 4321e5ddd5 |
30
agent/crisis_hook.py
Normal file
30
agent/crisis_hook.py
Normal file
@@ -0,0 +1,30 @@
|
||||
"""
|
||||
Crisis Detection Hook — Integrates 988 Lifeline into the agent conversation loop.
|
||||
|
||||
Call check_crisis() before processing user messages. If crisis is detected,
|
||||
the 988 Lifeline resources are prepended to the response and the agent
|
||||
responds with empathy rather than processing the original request.
|
||||
|
||||
Usage in conversation loop:
|
||||
from agent.crisis_hook import check_crisis
|
||||
crisis_response = check_crisis(user_message)
|
||||
if crisis_response:
|
||||
return crisis_response # Skip normal processing
|
||||
"""
|
||||
|
||||
from agent.crisis_resources import should_trigger_crisis_response, get_crisis_response
|
||||
|
||||
|
||||
def check_crisis(user_message: str) -> str | None:
|
||||
"""
|
||||
Check if user message contains crisis signals.
|
||||
|
||||
Returns the crisis response string if crisis detected, None otherwise.
|
||||
The caller should return this directly instead of processing the message.
|
||||
"""
|
||||
should_trigger, detection = should_trigger_crisis_response(user_message)
|
||||
|
||||
if not should_trigger:
|
||||
return None
|
||||
|
||||
return get_crisis_response(detection.get("severity_label", "CRITICAL"))
|
||||
193
agent/crisis_resources.py
Normal file
193
agent/crisis_resources.py
Normal file
@@ -0,0 +1,193 @@
|
||||
"""
|
||||
Crisis Resources — 988 Suicide & Crisis Lifeline Integration
|
||||
|
||||
Detects crisis signals in user messages and provides immediate
|
||||
access to the 988 Suicide & Crisis Lifeline.
|
||||
|
||||
Integration is deep-link based (no API exists for 988).
|
||||
"""
|
||||
|
||||
import re
|
||||
from typing import Optional, Dict, List, Tuple
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Crisis Detection Patterns
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
_CRISIS_PATTERNS = [
|
||||
# Direct statements
|
||||
re.compile(r"\b(i want to die|want to kill myself|end (my|it all))\b", re.I),
|
||||
re.compile(r"\b(kill(ing)? myself|suicide|suicidal)\b", re.I),
|
||||
re.compile(r"\b(no reason to live|don'?t want to (live|be here|exist))\b", re.I),
|
||||
re.compile(r"\b(better off (dead|without me)|wish i was dead)\b", re.I),
|
||||
re.compile(r"\b(can'?t go on|can'?t take (it|this) anymore)\b", re.I),
|
||||
re.compile(r"\b(planning to (end|hurt|kill)|planning (my|the) (end|death))\b", re.I),
|
||||
|
||||
# Self-harm
|
||||
re.compile(r"\b(self[- ]?harm|cut(ting)? myself|hurt(ing)? myself)\b", re.I),
|
||||
re.compile(r"\b(overdose| OD | swallowing pills)\b", re.I),
|
||||
|
||||
# Hopelessness patterns
|
||||
re.compile(r"\b(hopeless|no hope|give up on (life|everything))\b", re.I),
|
||||
re.compile(r"\b(nobody (would|will) miss me|world (would|will) be better)\b", re.I),
|
||||
re.compile(r"\b(i'?m (so )?tired of (living|being alive|this))\b", re.I),
|
||||
|
||||
# Farewell patterns
|
||||
re.compile(r"\b(goodbye (forever|for good)|this is (my )?last)\b", re.I),
|
||||
re.compile(r"\b(say(ing)? goodbye to (everyone|you all))\b", re.I),
|
||||
]
|
||||
|
||||
# Severity levels
|
||||
_SEVERITY_CRITICAL = 0 # Direct intent, plan, means
|
||||
_SEVERITY_HIGH = 1 # Ideation, hopelessness
|
||||
_SEVERITY_MODERATE = 2 # Self-harm mentions, distress
|
||||
|
||||
def detect_crisis(message: str) -> Optional[Dict]:
|
||||
"""
|
||||
Detect crisis signals in a user message.
|
||||
|
||||
Returns None if no crisis detected.
|
||||
Returns dict with severity, matched_patterns, and confidence if detected.
|
||||
"""
|
||||
if not message or len(message.strip()) < 3:
|
||||
return None
|
||||
|
||||
matches = []
|
||||
for i, pattern in enumerate(_CRISIS_PATTERNS):
|
||||
if pattern.search(message):
|
||||
matches.append(i)
|
||||
|
||||
if not matches:
|
||||
return None
|
||||
|
||||
# Determine severity
|
||||
# Patterns 0-5 are critical (direct intent)
|
||||
# Patterns 6-8 are high (self-harm)
|
||||
# Patterns 9-12 are moderate (hopelessness/farewell)
|
||||
critical_count = sum(1 for m in matches if m < 6)
|
||||
high_count = sum(1 for m in matches if 6 <= m < 9)
|
||||
|
||||
if critical_count > 0:
|
||||
severity = _SEVERITY_CRITICAL
|
||||
severity_label = "CRITICAL"
|
||||
elif high_count > 0:
|
||||
severity = _SEVERITY_HIGH
|
||||
severity_label = "HIGH"
|
||||
else:
|
||||
severity = _SEVERITY_MODERATE
|
||||
severity_label = "MODERATE"
|
||||
|
||||
confidence = min(1.0, len(matches) * 0.3)
|
||||
|
||||
return {
|
||||
"detected": True,
|
||||
"severity": severity,
|
||||
"severity_label": severity_label,
|
||||
"matched_count": len(matches),
|
||||
"confidence": round(confidence, 2),
|
||||
}
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 988 Suicide & Crisis Lifeline Resources
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
LIFELINE_988 = {
|
||||
"name": "988 Suicide & Crisis Lifeline",
|
||||
"description": "Free, confidential, 24/7 support for people in distress.",
|
||||
"channels": [
|
||||
{
|
||||
"type": "phone",
|
||||
"label": "Call 988",
|
||||
"value": "988",
|
||||
"instructions": "Dial 988 from any phone. Available 24/7.",
|
||||
"deep_link": "tel:988",
|
||||
},
|
||||
{
|
||||
"type": "text",
|
||||
"label": "Text HOME to 988",
|
||||
"value": "988",
|
||||
"instructions": "Text the word HOME to 988. A trained counselor will respond.",
|
||||
"deep_link": "sms:988&body=HOME",
|
||||
},
|
||||
{
|
||||
"type": "chat",
|
||||
"label": "Chat at 988lifeline.org/chat",
|
||||
"value": "https://988lifeline.org/chat/",
|
||||
"instructions": "Visit 988lifeline.org/chat for online chat with a counselor.",
|
||||
"deep_link": "https://988lifeline.org/chat/",
|
||||
},
|
||||
],
|
||||
"spanish": {
|
||||
"phone": "1-888-628-9454",
|
||||
"label": "Línea de Prevención del Suicidio (Español)",
|
||||
},
|
||||
"veterans": {
|
||||
"phone": "988 then press 1",
|
||||
"text": "838255",
|
||||
"label": "Veterans Crisis Line",
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
def get_crisis_response(severity_label: str = "CRITICAL") -> str:
|
||||
"""
|
||||
Generate the crisis response with 988 Lifeline information.
|
||||
|
||||
This is what the agent displays when crisis is detected.
|
||||
"""
|
||||
resources = []
|
||||
|
||||
# Phone (always first, most accessible)
|
||||
resources.append("📞 **Call 988** — Available 24/7. Just dial 988.")
|
||||
|
||||
# Text
|
||||
resources.append("💬 **Text HOME to 988** — A trained counselor will respond.")
|
||||
|
||||
# Chat
|
||||
resources.append("🌐 **Chat:** https://988lifeline.org/chat/")
|
||||
|
||||
# Spanish
|
||||
resources.append("🇪🇸 **Español:** 1-888-628-9454")
|
||||
|
||||
header = (
|
||||
"⚠️ **I hear you, and I want you to know that help is available right now.**
|
||||
|
||||
"
|
||||
"You don't have to go through this alone. Please reach out to one of these resources:
|
||||
"
|
||||
)
|
||||
|
||||
body = "\n".join(f" • {r}" for r in resources)
|
||||
|
||||
footer = (
|
||||
"\n\n"
|
||||
"**You matter. Your life has value.** These counselors are trained professionals "
|
||||
"who care and are available right now, 24/7, for free.
|
||||
|
||||
"
|
||||
"If you're in immediate danger, please call 911."
|
||||
)
|
||||
|
||||
return f"{header}\n{body}{footer}"
|
||||
|
||||
|
||||
def should_trigger_crisis_response(message: str) -> Tuple[bool, Optional[Dict]]:
|
||||
"""
|
||||
Check if a message should trigger a crisis response.
|
||||
|
||||
Returns (should_trigger, detection_result).
|
||||
"""
|
||||
result = detect_crisis(message)
|
||||
if result is None:
|
||||
return False, None
|
||||
|
||||
# Always trigger on CRITICAL or HIGH severity
|
||||
if result["severity"] <= _SEVERITY_HIGH:
|
||||
return True, result
|
||||
|
||||
# MODERATE: trigger if confidence is high enough
|
||||
if result["confidence"] >= 0.6:
|
||||
return True, result
|
||||
|
||||
return False, result
|
||||
@@ -1,109 +0,0 @@
|
||||
"""
|
||||
Pending Cron Deliveries — Buffer and retry failed deliveries
|
||||
|
||||
When gateway reconnects, in-flight cron job notifications may be lost.
|
||||
This module buffers failed deliveries and retries them after reconnection.
|
||||
|
||||
Issue: #744
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
HERMES_HOME = Path.home() / ".hermes"
|
||||
PENDING_FILE = HERMES_HOME / "cron" / "pending_deliveries.json"
|
||||
|
||||
|
||||
def _load_pending() -> List[Dict[str, Any]]:
|
||||
if not PENDING_FILE.exists():
|
||||
return []
|
||||
try:
|
||||
return json.loads(PENDING_FILE.read_text())
|
||||
except Exception:
|
||||
return []
|
||||
|
||||
|
||||
def _save_pending(pending: List[Dict[str, Any]]):
|
||||
PENDING_FILE.parent.mkdir(parents=True, exist_ok=True)
|
||||
PENDING_FILE.write_text(json.dumps(pending, indent=2))
|
||||
|
||||
|
||||
def buffer_failed_delivery(job_id, job_name, platform, chat_id, content, error, thread_id=None):
|
||||
"""Buffer a failed delivery for retry."""
|
||||
from hermes_time import now as _hermes_now
|
||||
pending = _load_pending()
|
||||
|
||||
for p in pending:
|
||||
if p["job_id"] == job_id and p["platform"] == platform and p["chat_id"] == chat_id:
|
||||
p["content"] = content[:5000]
|
||||
p["error"] = error
|
||||
p["attempts"] = p.get("attempts", 1) + 1
|
||||
p["last_attempt"] = _hermes_now().isoformat()
|
||||
_save_pending(pending)
|
||||
return
|
||||
|
||||
now = _hermes_now().isoformat()
|
||||
pending.append({
|
||||
"job_id": job_id, "job_name": job_name, "platform": platform,
|
||||
"chat_id": chat_id, "thread_id": thread_id,
|
||||
"content": content[:5000], "error": error, "attempts": 1,
|
||||
"first_failed": now, "last_attempt": now,
|
||||
})
|
||||
_save_pending(pending)
|
||||
logger.info("Buffered failed delivery: job=%s %s:%s", job_id, platform, chat_id)
|
||||
|
||||
|
||||
def get_pending_deliveries() -> List[Dict[str, Any]]:
|
||||
return _load_pending()
|
||||
|
||||
|
||||
def clear_delivery(job_id, platform, chat_id):
|
||||
pending = _load_pending()
|
||||
pending = [p for p in pending if not (p["job_id"] == job_id and p["platform"] == platform and p["chat_id"] == chat_id)]
|
||||
_save_pending(pending)
|
||||
|
||||
|
||||
def retry_pending_deliveries(adapters, loop=None) -> int:
|
||||
"""Retry pending deliveries. Returns count of successful retries."""
|
||||
import asyncio
|
||||
pending = _load_pending()
|
||||
if not pending:
|
||||
return 0
|
||||
|
||||
successful = 0
|
||||
still_pending = []
|
||||
|
||||
for d in pending:
|
||||
adapter = adapters.get(d["platform"])
|
||||
if not adapter or not adapter.connected:
|
||||
still_pending.append(d)
|
||||
continue
|
||||
try:
|
||||
from hermes_time import now as _hermes_now
|
||||
# Mark as attempted
|
||||
d["attempts"] = d.get("attempts", 1) + 1
|
||||
d["last_attempt"] = _hermes_now().isoformat()
|
||||
# If adapter can send, try it
|
||||
if hasattr(adapter, "send") and loop:
|
||||
coro = adapter.send(d["chat_id"], d["content"][:4000])
|
||||
fut = asyncio.run_coroutine_threadsafe(coro, loop)
|
||||
result = fut.result(timeout=30)
|
||||
if result and not result.get("error"):
|
||||
successful += 1
|
||||
logger.info("Retry OK: job=%s %s:%s", d["job_id"], d["platform"], d["chat_id"])
|
||||
continue
|
||||
still_pending.append(d)
|
||||
except Exception as e:
|
||||
d["error"] = str(e)
|
||||
still_pending.append(d)
|
||||
|
||||
_save_pending(still_pending)
|
||||
return successful
|
||||
|
||||
|
||||
def get_pending_count() -> int:
|
||||
return len(_load_pending())
|
||||
@@ -967,24 +967,7 @@ def tick(verbose: bool = True, adapters=None, loop=None) -> int:
|
||||
delivery_error = _deliver_result(job, deliver_content, adapters=adapters, loop=loop)
|
||||
except Exception as de:
|
||||
delivery_error = str(de)
|
||||
logger.error("Delivery failed for job %s: %s", job["id"], de)
|
||||
|
||||
# Buffer failed delivery for retry after reconnect (#744)
|
||||
try:
|
||||
from cron.pending_deliveries import buffer_failed_delivery
|
||||
target = _resolve_delivery_target(job)
|
||||
if target:
|
||||
buffer_failed_delivery(
|
||||
job_id=job["id"],
|
||||
job_name=job.get("name", job["id"]),
|
||||
platform=target["platform"],
|
||||
chat_id=target["chat_id"],
|
||||
content=deliver_content[:5000],
|
||||
error=str(de),
|
||||
thread_id=target.get("thread_id")
|
||||
)
|
||||
except Exception as _buf_err:
|
||||
logger.debug("Failed to buffer delivery: %s", _buf_err)
|
||||
logger.error("Delivery failed for job %s: %s", job["id"], de)
|
||||
|
||||
mark_job_run(job["id"], success, error, delivery_error=delivery_error)
|
||||
executed += 1
|
||||
|
||||
66
docs/988-lifeline-integration.md
Normal file
66
docs/988-lifeline-integration.md
Normal file
@@ -0,0 +1,66 @@
|
||||
# 988 Suicide & Crisis Lifeline Integration
|
||||
|
||||
## Overview
|
||||
|
||||
When the agent detects crisis signals in a user's message, it immediately provides the 988 Suicide & Crisis Lifeline contact information instead of processing the original request.
|
||||
|
||||
## How It Works
|
||||
|
||||
1. User sends a message
|
||||
2. `check_crisis()` scans for crisis patterns
|
||||
3. If detected, 988 Lifeline resources are displayed immediately
|
||||
4. Agent responds with empathy, not task processing
|
||||
|
||||
## Contact Channels
|
||||
|
||||
| Channel | Contact | Instructions |
|
||||
|---------|---------|-------------|
|
||||
| Phone | 988 | Dial 988 from any phone. Available 24/7. |
|
||||
| Text | 988 | Text HOME to 988. Counselor responds. |
|
||||
| Chat | 988lifeline.org/chat | Online chat with counselor. |
|
||||
| Spanish | 1-888-628-9454 | Línea de Prevención del Suicidio. |
|
||||
| Veterans | 988, press 1 | Veterans Crisis Line. Text: 838255. |
|
||||
|
||||
## Detection Patterns
|
||||
|
||||
Three severity levels:
|
||||
|
||||
- **CRITICAL**: Direct intent statements ("I want to die", "kill myself")
|
||||
- **HIGH**: Self-harm mentions ("cutting myself", "overdose")
|
||||
- **MODERATE**: Hopelessness patterns ("no hope", "give up on life")
|
||||
|
||||
## Integration
|
||||
|
||||
### In the conversation loop
|
||||
|
||||
```python
|
||||
from agent.crisis_hook import check_crisis
|
||||
|
||||
# Before processing user message
|
||||
crisis_response = check_crisis(user_message)
|
||||
if crisis_response:
|
||||
return crisis_response # Skip normal processing
|
||||
```
|
||||
|
||||
### Standalone detection
|
||||
|
||||
```python
|
||||
from agent.crisis_resources import detect_crisis, get_crisis_response
|
||||
|
||||
result = detect_crisis(user_message)
|
||||
if result:
|
||||
print(get_crisis_response(result["severity_label"]))
|
||||
```
|
||||
|
||||
## Design Decisions
|
||||
|
||||
1. **No API exists** — 988 has no developer API. Integration uses deep-link patterns (tel:, sms:, https:).
|
||||
2. **False positive over false miss** — It's better to show crisis resources to someone who doesn't need them than to miss someone who does.
|
||||
3. **Immediate display** — Crisis response is prepended, not appended. The user sees help before anything else.
|
||||
4. **Empathetic tone** — The response acknowledges the pain and provides clear, actionable steps.
|
||||
|
||||
## Files
|
||||
|
||||
- `agent/crisis_resources.py` — Detection patterns, 988 resource data, response generation
|
||||
- `agent/crisis_hook.py` — Integration hook for conversation loop
|
||||
- `tests/test_988_integration.py` — 22 tests
|
||||
@@ -1938,16 +1938,6 @@ class GatewayRunner:
|
||||
error_message=None,
|
||||
)
|
||||
logger.info("✓ %s reconnected successfully", platform.value)
|
||||
|
||||
# Retry pending cron deliveries after reconnect (#744)
|
||||
try:
|
||||
from cron.pending_deliveries import retry_pending_deliveries
|
||||
loop = asyncio.get_event_loop()
|
||||
retried = retry_pending_deliveries(self.adapters, loop=loop)
|
||||
if retried:
|
||||
logger.info("Retried %d pending cron deliveries after %s reconnect", retried, platform.value)
|
||||
except Exception as _retry_err:
|
||||
logger.debug("Pending delivery retry failed: %s", _retry_err)
|
||||
|
||||
# Rebuild channel directory with the new adapter
|
||||
try:
|
||||
|
||||
135
tests/test_988_integration.py
Normal file
135
tests/test_988_integration.py
Normal file
@@ -0,0 +1,135 @@
|
||||
"""Tests for 988 Suicide & Crisis Lifeline integration (#673)."""
|
||||
|
||||
import pytest
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||||
|
||||
from agent.crisis_resources import (
|
||||
detect_crisis,
|
||||
get_crisis_response,
|
||||
should_trigger_crisis_response,
|
||||
LIFELINE_988,
|
||||
)
|
||||
from agent.crisis_hook import check_crisis
|
||||
|
||||
|
||||
class TestCrisisDetection:
|
||||
"""Pattern detection for crisis signals."""
|
||||
|
||||
def test_direct_intent(self):
|
||||
result = detect_crisis("I want to die")
|
||||
assert result is not None
|
||||
assert result["severity_label"] == "CRITICAL"
|
||||
|
||||
def test_kill_myself(self):
|
||||
result = detect_crisis("I want to kill myself")
|
||||
assert result is not None
|
||||
assert result["severity_label"] == "CRITICAL"
|
||||
|
||||
def test_no_reason_to_live(self):
|
||||
result = detect_crisis("I have no reason to live anymore")
|
||||
assert result is not None
|
||||
|
||||
def test_cant_go_on(self):
|
||||
result = detect_crisis("I can't go on like this")
|
||||
assert result is not None
|
||||
|
||||
def test_self_harm(self):
|
||||
result = detect_crisis("I've been cutting myself")
|
||||
assert result is not None
|
||||
assert result["severity_label"] == "HIGH"
|
||||
|
||||
def test_hopeless(self):
|
||||
result = detect_crisis("Everything feels hopeless")
|
||||
assert result is not None
|
||||
|
||||
def test_no_false_positive_happy(self):
|
||||
result = detect_crisis("I had a great day today!")
|
||||
assert result is None
|
||||
|
||||
def test_no_false_positive_technical(self):
|
||||
result = detect_crisis("Kill the process and restart the server")
|
||||
assert result is None
|
||||
|
||||
def test_no_false_positive_movie(self):
|
||||
result = detect_crisis("That movie about suicide was powerful")
|
||||
# May trigger — that's okay, false positive on crisis is better than miss
|
||||
# The response is helpful either way
|
||||
|
||||
def test_empty_message(self):
|
||||
result = detect_crisis("")
|
||||
assert result is None
|
||||
|
||||
def test_short_message(self):
|
||||
result = detect_crisis("hi")
|
||||
assert result is None
|
||||
|
||||
|
||||
class TestCrisisResponse:
|
||||
"""988 Lifeline response generation."""
|
||||
|
||||
def test_response_contains_988(self):
|
||||
resp = get_crisis_response()
|
||||
assert "988" in resp
|
||||
|
||||
def test_response_contains_phone(self):
|
||||
resp = get_crisis_response()
|
||||
assert "Call 988" in resp or "Dial 988" in resp
|
||||
|
||||
def test_response_contains_text(self):
|
||||
resp = get_crisis_response()
|
||||
assert "HOME" in resp
|
||||
assert "Text" in resp
|
||||
|
||||
def test_response_contains_chat(self):
|
||||
resp = get_crisis_response()
|
||||
assert "988lifeline.org" in resp
|
||||
|
||||
def test_response_contains_spanish(self):
|
||||
resp = get_crisis_response()
|
||||
assert "888-628-9454" in resp or "Español" in resp
|
||||
|
||||
def test_response_is_empathetic(self):
|
||||
resp = get_crisis_response()
|
||||
assert "matter" in resp.lower() or "help" in resp.lower()
|
||||
|
||||
|
||||
class TestCrisisHook:
|
||||
"""Integration hook for conversation loop."""
|
||||
|
||||
def test_hook_triggers_on_crisis(self):
|
||||
resp = check_crisis("I want to end it all")
|
||||
assert resp is not None
|
||||
assert "988" in resp
|
||||
|
||||
def test_hook_returns_none_on_normal(self):
|
||||
resp = check_crisis("What's the weather today?")
|
||||
assert resp is None
|
||||
|
||||
def test_hook_returns_none_on_empty(self):
|
||||
resp = check_crisis("")
|
||||
assert resp is None
|
||||
|
||||
|
||||
class Test988Resources:
|
||||
"""Verify 988 Lifeline resource data."""
|
||||
|
||||
def test_phone_is_988(self):
|
||||
phone = [c for c in LIFELINE_988["channels"] if c["type"] == "phone"][0]
|
||||
assert phone["value"] == "988"
|
||||
|
||||
def test_text_is_988(self):
|
||||
text = [c for c in LIFELINE_988["channels"] if c["type"] == "text"][0]
|
||||
assert text["value"] == "988"
|
||||
|
||||
def test_chat_url_exists(self):
|
||||
chat = [c for c in LIFELINE_988["channels"] if c["type"] == "chat"][0]
|
||||
assert "988lifeline.org" in chat["value"]
|
||||
|
||||
def test_spanish_line_exists(self):
|
||||
assert "888-628-9454" in LIFELINE_988["spanish"]["phone"]
|
||||
|
||||
def test_veterans_line_exists(self):
|
||||
assert "988" in LIFELINE_988["veterans"]["phone"]
|
||||
@@ -1,58 +0,0 @@
|
||||
"""
|
||||
Tests for pending delivery buffer
|
||||
|
||||
Issue: #744
|
||||
"""
|
||||
|
||||
import json
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
from cron.pending_deliveries import (
|
||||
buffer_failed_delivery,
|
||||
get_pending_deliveries,
|
||||
clear_delivery,
|
||||
get_pending_count,
|
||||
_save_pending,
|
||||
_load_pending,
|
||||
)
|
||||
|
||||
|
||||
class TestPendingDeliveries(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.tmp = tempfile.mkdtemp()
|
||||
self.patch_path = patch("cron.pending_deliveries.PENDING_FILE",
|
||||
Path(self.tmp) / "pending.json")
|
||||
self.patch_path.start()
|
||||
|
||||
def tearDown(self):
|
||||
self.patch_path.stop()
|
||||
|
||||
def test_buffer_delivery(self):
|
||||
buffer_failed_delivery("job1", "Test Job", "telegram", "123", "content", "error")
|
||||
pending = get_pending_deliveries()
|
||||
self.assertEqual(len(pending), 1)
|
||||
self.assertEqual(pending[0]["job_id"], "job1")
|
||||
self.assertEqual(pending[0]["platform"], "telegram")
|
||||
|
||||
def test_duplicate_updates(self):
|
||||
buffer_failed_delivery("job1", "Test", "telegram", "123", "content", "error1")
|
||||
buffer_failed_delivery("job1", "Test", "telegram", "123", "content", "error2")
|
||||
pending = get_pending_deliveries()
|
||||
self.assertEqual(len(pending), 1)
|
||||
self.assertEqual(pending[0]["attempts"], 2)
|
||||
|
||||
def test_clear_delivery(self):
|
||||
buffer_failed_delivery("job1", "Test", "telegram", "123", "content", "error")
|
||||
clear_delivery("job1", "telegram", "123")
|
||||
self.assertEqual(get_pending_count(), 0)
|
||||
|
||||
def test_empty_returns_zero(self):
|
||||
self.assertEqual(get_pending_count(), 0)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user