from app.services import chat_history as history


def test_edits_regeneration_and_activity_survive_version_switch():
    history.create('Versions', 'versions')
    def append(id, role, content, parent=None, activity=None):
        history.append_message('versions', message_id=id, role=role, content=content, parent_message_id=parent, activity=activity)
    append('u1', 'user', 'original')
    append('a1', 'assistant', 'original answer', 'u1')
    append('u2', 'user', 'follow-up')
    append('a2', 'assistant', 'follow-up answer', 'u2')
    history.prepare_retry('versions', 'u1')
    append('u1-edit', 'user', 'edited')
    history.reserve_response('versions', 'a1-edit')
    trace = [{'type': 'thinking', 'text': 'before'}, {'type': 'tool', 'tool_call_id': 'tool'}, {'type': 'thinking', 'text': 'after'}]
    append('a1-edit', 'assistant', 'edited answer', 'u1-edit', trace)
    items, _ = history.list_messages('versions', 500, 0)
    assert [m.message_id for m in items] == ['u1-edit', 'a1-edit']
    assert items[0].versions == ['u1', 'u1-edit']
    assert items[1].activity == trace
    history.select_version('versions', 'u1')
    assert [m.message_id for m in history.list_messages('versions', 500, 0)[0]] == ['u1', 'a1', 'u2', 'a2']
    history.prepare_retry('versions', 'a1')
    history.reserve_response('versions', 'a1-new')
    append('a1-new', 'assistant', 'regenerated', 'u1')
    items, _ = history.list_messages('versions', 500, 0)
    assert [m.message_id for m in items] == ['u1', 'a1-new']
    assert items[-1].versions == ['a1', 'a1-new']
    history.select_version('versions', 'a1')
    assert history.list_messages('versions', 500, 0)[0][-1].message_id == 'a2'


def test_late_response_does_not_replace_new_generation():
    history.create('Late', 'late')
    history.append_message('late', message_id='u', role='user', content='question')
    history.reserve_response('late', 'new')
    history.append_message('late', message_id='old', role='assistant', content='old', parent_message_id='u')
    assert history.list_messages('late', 500, 0)[0][-1].message_id == 'u'
    history.append_message('late', message_id='new', role='assistant', content='new', parent_message_id='u')
    assert history.list_messages('late', 500, 0)[0][-1].message_id == 'new'