File size: 3,959 Bytes
999bb04 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | from __future__ import annotations
from dataclasses import dataclass
from datetime import datetime, timedelta, timezone
import pytest
from aura.agents.echo import tools as echo
from aura.agents.echo.models import EmailDraft
from aura.core.config import AppConfig, FeatureFlags, ModelSettings, PathsSettings
@dataclass
class FakeNotification:
ok: bool = True
message: str = "sent"
class FakeSMTP:
def __init__(self, host, port):
self.host = host
self.port = port
self.logged_in = False
self.sent = False
def login(self, username, password):
self.logged_in = True
def send_message(self, message):
self.sent = True
def __enter__(self):
return self
def __exit__(self, exc_type, exc, tb):
return False
@pytest.fixture()
def echo_config(tmp_path):
config = AppConfig(
name="AURA",
offline_mode=True,
log_level="INFO",
primary_model=ModelSettings(provider="ollama", name="llama3", host="http://127.0.0.1:11434"),
fallback_models=[],
paths=PathsSettings(
allowed_roots=[tmp_path],
data_dir=tmp_path,
log_dir=tmp_path / "logs",
memory_dir=tmp_path / "memory",
ipc_socket=tmp_path / "aura.sock",
),
features=FeatureFlags(hotkey=True, tray=True, ipc=True, api=True),
source_path=tmp_path / "config.yaml",
)
echo.CONFIG = config
return config
def test_parse_natural_time_and_reminder_persistence(monkeypatch, echo_config):
monkeypatch.setattr(echo, "notify_user", lambda *args, **kwargs: FakeNotification())
iso = echo.parse_natural_time("tomorrow 9am")
reminder = echo.set_reminder("Standup", "tomorrow 9am")
upcoming = echo.get_upcoming_reminders(hours_ahead=48)
assert iso.endswith("+00:00")
assert reminder.text == "Standup"
assert upcoming and upcoming[0].id == reminder.id
def test_create_update_list_and_cancel_meeting(echo_config):
start = (datetime.now(timezone.utc) + timedelta(hours=1)).isoformat()
end = (datetime.now(timezone.utc) + timedelta(hours=2)).isoformat()
meeting = echo.create_meeting("Planning", start, end, ["a@example.com"], "offline", "demo")
meetings = echo.list_meetings({"start": (datetime.now(timezone.utc) - timedelta(minutes=5)).isoformat(), "end": (datetime.now(timezone.utc) + timedelta(hours=3)).isoformat()})
updated = echo.update_meeting(meeting.id, {"title": "Planning 2"})
cancelled = echo.cancel_meeting(meeting.id)
meetings_after = echo.list_meetings({"start": (datetime.now(timezone.utc) - timedelta(minutes=5)).isoformat(), "end": (datetime.now(timezone.utc) + timedelta(hours=3)).isoformat()})
assert meeting.provider == "local-sqlite"
assert meetings and meetings[0].id == meeting.id
assert updated.title == "Planning 2"
assert cancelled.success is True
assert meetings_after == []
def test_join_draft_and_send_email(monkeypatch, echo_config):
monkeypatch.setattr(echo, "open_path", lambda link: type("Result", (), {"ok": True, "message": "opened", "details": {"link": link}})())
draft = echo.draft_email(["a@example.com"], "Subject", "Body")
assert isinstance(draft, EmailDraft)
assert draft.subject == "Subject"
echo.set_email_config({"smtp_host": "smtp.example.com", "smtp_port": 465, "username": "user", "password": "pass", "from_address": "user@example.com"})
monkeypatch.setattr(echo.smtplib, "SMTP_SSL", FakeSMTP)
send_result = echo.send_email(draft.id)
join_result = echo.join_meeting("https://example.com/meet")
assert send_result.success is True
assert join_result.success is True
def test_send_email_without_config_and_invalid_time(echo_config):
echo.set_email_config(None)
result = echo.send_email("missing")
assert result.success is False
with pytest.raises(Exception):
echo.parse_natural_time("not a time")
|