mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-06 15:02:20 +02:00
pewdiepie-archdaemon
2e8413a54a
Snapshot current maintainer-preview application changes and regression fixtures for integration into lab. Excludes local runtime data, evaluation outputs and source backups. Focused Python regression selection: 140 passed; full suite not certified.
64 lines
2.9 KiB
Python
64 lines
2.9 KiB
Python
import json
|
|
import pytest
|
|
from src.agent_tools.web_tools import YouTubeTool
|
|
from src.clean_agent_preview import youtube_reference_error
|
|
|
|
|
|
@pytest.mark.parametrize('role,content,allowed', [
|
|
('assistant', 'https://youtube.com/watch?v=abcdefghijk', False),
|
|
('tool', '{"exit_code":1,"video_url":"https://youtube.com/watch?v=abcdefghijk"}', False),
|
|
('tool', 'Video ID: abcdefghijk', True),
|
|
('tool', '- link "Video title" [ref=e78]', False),
|
|
('user', 'https://youtube.com/watch?v=abcdefghijk', True),
|
|
])
|
|
def test_video_identity_requires_evidence(role, content, allowed):
|
|
error = youtube_reference_error('youtube_tool', {'action': 'comments', 'video_id': 'abcdefghijk'},
|
|
user_text='Read comments on it', history=[{'role': role, 'content': content}])
|
|
assert bool(error) is not allowed
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('target', [
|
|
'https://www.youtube.com/watch?v=example_last_vlog',
|
|
'https://www.youtube.com/watch?v=abcdefghijkEXTRA',
|
|
'https://www.youtube.com/@creator/videos',
|
|
'https://notyoutube.example/watch?v=abcdefghijk',
|
|
'invented_title',
|
|
])
|
|
async def test_invalid_target_rejected_before_network(monkeypatch, target):
|
|
async def unexpected(*args, **kwargs):
|
|
pytest.fail('Invalid target must not reach retrieval')
|
|
monkeypatch.setattr(YouTubeTool, '_comments_from_data_api', unexpected)
|
|
result = await YouTubeTool().execute(json.dumps({'action': 'comments', 'video_url': target}), {})
|
|
assert result['exit_code'] == 1
|
|
assert result['failure_kind'] == 'invalid_target'
|
|
assert 'actual video URL' in result['error']
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('target', ['abcdefghijk',
|
|
'https://www.youtube.com/watch?v=abcdefghijk&t=42',
|
|
'https://youtu.be/abcdefghijk', 'https://www.youtube.com/shorts/abcdefghijk'])
|
|
async def test_valid_targets_reach_retrieval(monkeypatch, target):
|
|
seen = []
|
|
async def metadata(self, url):
|
|
seen.append(url)
|
|
return {'output': 'fixture', 'exit_code': 0}
|
|
monkeypatch.setattr(YouTubeTool, '_metadata', metadata)
|
|
result = await YouTubeTool().execute(json.dumps({'action': 'metadata', 'video_url': target}), {})
|
|
assert result['exit_code'] == 0 and len(seen) == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_comment_failure_distinguishes_retrieval_from_invalid_target(monkeypatch):
|
|
import services.youtube.youtube_handler as handler
|
|
async def unavailable(*args, **kwargs):
|
|
return {'success': False, 'error': 'Fixture transport unavailable'}
|
|
monkeypatch.setattr(YouTubeTool, '_comments_from_data_api', unavailable)
|
|
monkeypatch.setattr(handler, 'fetch_youtube_comments', unavailable)
|
|
result = await YouTubeTool().execute(json.dumps({
|
|
'action': 'comments', 'video_id': 'abcdefghijk'}), {})
|
|
assert result['failure_kind'] == 'comments_unavailable'
|
|
assert result['video_url'] == 'https://www.youtube.com/watch?v=abcdefghijk'
|
|
assert result['exit_code'] == 1
|