55 lines
2.9 KiB
Python
55 lines
2.9 KiB
Python
|
import asyncio
|
||
|
|
from contextlib import asynccontextmanager
|
||
|
|
from dataclasses import replace
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
import pytest
|
||
|
|
from fastapi import FastAPI
|
||
|
|
|
||
|
|
from src.api.endpoints import router
|
||
|
|
from src.core import fastgpt_client as wiring
|
||
|
|
from src.core.config import Config
|
||
|
|
from src.storage.runtime import open_storage
|
||
|
|
from .test_persistence import RecordingLLM, run, settings
|
||
|
|
|
||
|
|
|
||
|
|
def test_http_restart_replay_headers_and_switching(settings, monkeypatch):
|
||
|
|
import src.agent.llm as llm_module
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_BACKEND', 'langgraph')
|
||
|
|
monkeypatch.setattr(Config, 'LLM_PROVIDER', 'fake')
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_STORAGE', settings.kind)
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_SQLITE_PATH', settings.sqlite_path)
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_POSTGRES_DSN', settings.postgres_dsn)
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_DB_POOL_SIZE', settings.pool_size)
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_DB_AUTO_MIGRATE', True)
|
||
|
|
monkeypatch.setattr(llm_module, 'build_llm_service', lambda: RecordingLLM())
|
||
|
|
app = FastAPI()
|
||
|
|
app.include_router(router)
|
||
|
|
@asynccontextmanager
|
||
|
|
async def client():
|
||
|
|
async with wiring.lifespan(app):
|
||
|
|
async with httpx.AsyncClient(transport=httpx.ASGITransport(app=app), base_url='http://test') as value:
|
||
|
|
yield value
|
||
|
|
async def scenario():
|
||
|
|
import uuid
|
||
|
|
payload = {'sessionId': uuid.uuid4().hex, 'timeStamp':'20261022120000'}
|
||
|
|
request = {**payload, 'text':'新对话'}
|
||
|
|
async with client() as http:
|
||
|
|
assert (await http.post('/set_info', json={**payload,'key':'hphm1','value':'http-plate'})).json()['code']=='200'
|
||
|
|
first = (await http.post('/chat', json=request, headers={'Idempotency-Key':'http-key'})).json()
|
||
|
|
assert first['nextStageCode']=='1001'
|
||
|
|
monkeypatch.setattr(Config, 'AGENT_DB_AUTO_MIGRATE', False)
|
||
|
|
async with client() as http:
|
||
|
|
assert (await http.post('/chat', json=request, headers={'Idempotency-Key':'http-key'})).json()==first
|
||
|
|
conflict = await http.post('/chat', json={**request,'text':'different'}, headers={'Idempotency-Key':'http-key'})
|
||
|
|
assert conflict.json()['code']=='409'
|
||
|
|
stream = await http.post('/chat?stream=true', json={**payload,'text':'继续办理','needFormUpdate':False})
|
||
|
|
assert 'event: done' in stream.text
|
||
|
|
assert 'event: formUpdate' not in stream.text
|
||
|
|
assert '"0"' == (await http.post('/get_info', json={**payload,'key':'jdcsl'})).json()['value']
|
||
|
|
assert (await http.request('DELETE','/delete_session',json=payload)).json()['code']=='200'
|
||
|
|
async with client() as http:
|
||
|
|
assert (await http.post('/get_info',json={**payload,'key':'hphm1'})).json()['value']=='""'
|
||
|
|
assert (await http.post('/chat',json=request,headers={'Idempotency-Key':'http-key'})).json()['nextStageCode']=='1001'
|
||
|
|
run(scenario())
|