/
pv_hum
/
schemaforge
Обзор
Документация
Войти
/
pv_hum
/
schemaforge
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
tests/agent/test_payload_builder.py
43 строки
2 KB
pv_hum
Initial commit
21 май 2026, 22:25
21 май 2026, 22:25
0de9fa8
Код
Авторство
О чём код?
from app.models import ChangedFlags, Entity, SchemaModel, Attribute from app.payload_builder import build_agent_input_payload def test_build_prompt_only(): payload = build_agent_input_payload( mode='prompt', active_view='er', text_prompt='Схема магазина', changed_flags=ChangedFlags(promptChanged=True, sqlChanged=False, erChanged=False), ) assert 'Схема магазина' in payload.userPrompt assert payload.includedContext['prompt'] is True assert payload.includedContext['sql'] is False def test_build_sql_only(): payload = build_agent_input_payload( mode='sql', active_view='sql', sql='CREATE TABLE t (id INT);', changed_flags=ChangedFlags(promptChanged=False, sqlChanged=True, erChanged=False), ) assert 'CREATE TABLE' in payload.userPrompt assert payload.includedContext['sql'] is True def test_build_er_revision(): schema = SchemaModel( entities=[Entity(name='a', attributes=[Attribute(name='id', type='int', nullable=False, primaryKey=True)])], relationships=[], ) payload = build_agent_input_payload( mode='er', active_view='er', er_schema=schema, revision_prompt='добавь таблицу b', changed_flags=ChangedFlags(promptChanged=False, sqlChanged=False, erChanged=True), iteration=2, previous_issues=[{'severity': 'warning', 'issueCode': 'X', 'message': 'test'}], ) assert 'добавь таблицу b' in payload.userPrompt assert 'Итерация доработки: 2' in payload.userPrompt