mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
|
"""Tests for the library map injected into every chat."""
|
||
|
|
import time
|
||
|
|
from unittest import mock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from modules.ai import context
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_build_library_context(async_client, test_session, video_factory, archive_factory,
|
||
|
|
test_zim, food_inventory_factory):
|
||
|
|
"""The map carries today's date, counts, and exact names."""
|
||
|
|
from wrolpi.collections.models import Collection
|
||
|
|
test_session.add(Collection(name='Talking Sasquach', kind='channel'))
|
||
|
|
video_factory(title='a video')
|
||
|
|
archive_factory(domain='example.com', title='a page', contents='contents')
|
||
|
|
food_inventory_factory(name='Pantry')
|
||
|
|
test_session.commit()
|
||
|
|
|
||
|
|
text = context.build_library_context()
|
||
|
|
assert 'Today is' in text
|
||
|
|
assert '1 videos, 1 archived web pages' in text
|
||
|
|
assert 'Talking Sasquach' in text
|
||
|
|
assert 'example.com' in text
|
||
|
|
assert 'WROLPi example ZIM' in text
|
||
|
|
assert 'Pantry' in text
|
||
|
|
assert 'user spellings may differ' in text
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_library_context_caps_names(async_client, test_session, archive_factory):
|
||
|
|
"""A huge library lists at most MAX_NAMES names per category, plus a count of the rest."""
|
||
|
|
for i in range(context.MAX_NAMES + 5):
|
||
|
|
archive_factory(domain=f'site{i:02}.com', title=f'page {i}', contents='x')
|
||
|
|
test_session.commit()
|
||
|
|
|
||
|
|
text = context.build_library_context()
|
||
|
|
assert f'Archived websites ({context.MAX_NAMES + 5}):' in text
|
||
|
|
assert 'and 5 more' in text
|
||
|
|
listed = [i for i in range(context.MAX_NAMES + 5) if f'site{i:02}.com' in text]
|
||
|
|
assert len(listed) == context.MAX_NAMES
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_library_context_cache_and_degradation(async_client, test_session):
|
||
|
|
"""The map is cached briefly; a build failure serves the stale copy rather than crashing chat."""
|
||
|
|
context._cache.update(text=None, expires=0.0)
|
||
|
|
first = context.get_library_context()
|
||
|
|
assert first and 'Today is' in first
|
||
|
|
|
||
|
|
with mock.patch('modules.ai.context.build_library_context', side_effect=RuntimeError('db down')) as build:
|
||
|
|
# Cached: the builder is not called.
|
||
|
|
assert context.get_library_context() == first
|
||
|
|
build.assert_not_called()
|
||
|
|
|
||
|
|
# Expired + failing builder: the stale copy is served.
|
||
|
|
context._cache['expires'] = time.time() - 1
|
||
|
|
assert context.get_library_context() == first
|
||
|
|
build.assert_called_once()
|