102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
import json
|
||
|
|
import pytest
|
||
|
|
from unittest.mock import Mock, patch
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
# Add scripts to path
|
||
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "scripts"))
|
||
|
|
|
||
|
|
from briefing import generate_and_send
|
||
|
|
|
||
|
|
def test_generate_and_send_success():
|
||
|
|
# Mock subprocess.run for summarize.py
|
||
|
|
mock_briefing_data = {
|
||
|
|
"macro_message": "Macro Summary",
|
||
|
|
"portfolio_message": "Portfolio Summary",
|
||
|
|
"summary": "Full Summary"
|
||
|
|
}
|
||
|
|
|
||
|
|
with patch("briefing.subprocess.run") as mock_run:
|
||
|
|
mock_result = Mock()
|
||
|
|
mock_result.returncode = 0
|
||
|
|
mock_result.stdout = json.dumps(mock_briefing_data)
|
||
|
|
mock_run.return_value = mock_result
|
||
|
|
|
||
|
|
args = Mock()
|
||
|
|
args.time = "morning"
|
||
|
|
args.style = "briefing"
|
||
|
|
args.lang = "en"
|
||
|
|
args.deadline = 300
|
||
|
|
args.fast = False
|
||
|
|
args.llm = False
|
||
|
|
args.debug = False
|
||
|
|
args.json = True
|
||
|
|
args.send = False
|
||
|
|
|
||
|
|
result = generate_and_send(args)
|
||
|
|
|
||
|
|
assert result == "Macro Summary"
|
||
|
|
assert mock_run.called
|
||
|
|
# Check if summarize.py was called with correct args
|
||
|
|
call_args = mock_run.call_args[0][0]
|
||
|
|
assert "summarize.py" in str(call_args[1])
|
||
|
|
assert "--time" in call_args
|
||
|
|
assert "morning" in call_args
|
||
|
|
|
||
|
|
def test_generate_and_send_with_whatsapp():
|
||
|
|
mock_briefing_data = {
|
||
|
|
"macro_message": "Macro Summary",
|
||
|
|
"portfolio_message": "Portfolio Summary"
|
||
|
|
}
|
||
|
|
|
||
|
|
with patch("briefing.subprocess.run") as mock_run, \
|
||
|
|
patch("briefing.send_to_whatsapp") as mock_send:
|
||
|
|
|
||
|
|
# First call is summarize.py
|
||
|
|
mock_result = Mock()
|
||
|
|
mock_result.returncode = 0
|
||
|
|
mock_result.stdout = json.dumps(mock_briefing_data)
|
||
|
|
mock_run.return_value = mock_result
|
||
|
|
|
||
|
|
args = Mock()
|
||
|
|
args.time = "evening"
|
||
|
|
args.style = "briefing"
|
||
|
|
args.lang = "en"
|
||
|
|
args.deadline = None
|
||
|
|
args.fast = True
|
||
|
|
args.llm = False
|
||
|
|
args.json = False
|
||
|
|
args.send = True
|
||
|
|
args.group = "Test Group"
|
||
|
|
args.debug = False
|
||
|
|
|
||
|
|
generate_and_send(args)
|
||
|
|
|
||
|
|
# Check if send_to_whatsapp was called for both messages
|
||
|
|
assert mock_send.call_count == 2
|
||
|
|
mock_send.assert_any_call("Macro Summary", "Test Group")
|
||
|
|
mock_send.assert_any_call("Portfolio Summary", "Test Group")
|
||
|
|
|
||
|
|
def test_generate_and_send_failure():
|
||
|
|
with patch("briefing.subprocess.run") as mock_run:
|
||
|
|
mock_result = Mock()
|
||
|
|
mock_result.returncode = 1
|
||
|
|
mock_result.stderr = "Error occurred"
|
||
|
|
mock_run.return_value = mock_result
|
||
|
|
|
||
|
|
args = Mock()
|
||
|
|
args.time = "morning"
|
||
|
|
args.style = "briefing"
|
||
|
|
args.lang = "en"
|
||
|
|
args.deadline = None
|
||
|
|
args.fast = False
|
||
|
|
args.llm = False
|
||
|
|
args.json = False
|
||
|
|
args.send = False
|
||
|
|
args.debug = False
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit):
|
||
|
|
generate_and_send(args)
|