132 lines
3.5 KiB
Python
132 lines
3.5 KiB
Python
import pytest
|
|
import time
|
|
import sys
|
|
import os
|
|
import json
|
|
from pathlib import Path
|
|
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
|
|
|
|
from src import api_hook_client
|
|
|
|
@pytest.mark.integration
|
|
def test_undo_redo_lifecycle(live_gui):
|
|
client = api_hook_client.ApiHookClient()
|
|
assert client.wait_for_server(timeout=15), "Hook server did not start"
|
|
|
|
# 1. Set initial state
|
|
print("Setting initial state...")
|
|
client.set_value('temperature', 0.5)
|
|
client.set_value('ai_input', "Initial Input")
|
|
|
|
# Wait for settle and first push (S_init -> S0)
|
|
time.sleep(3.0)
|
|
|
|
# 2. Change state
|
|
print("Modifying state...")
|
|
client.set_value('temperature', 1.5)
|
|
client.set_value('ai_input', "Modified Input")
|
|
|
|
# Wait for settle and second push (S0 -> S1)
|
|
time.sleep(3.0)
|
|
|
|
# Verify current state
|
|
temp = client.get_value('temperature')
|
|
ai_in = client.get_value('ai_input')
|
|
print(f"Current state: temp={temp}, ai_input={ai_in}")
|
|
assert temp == 1.5
|
|
assert ai_in == "Modified Input"
|
|
|
|
# 3. Undo
|
|
print("Sending Undo...")
|
|
client.click('btn_undo')
|
|
|
|
# Wait for state to revert
|
|
time.sleep(1.0)
|
|
|
|
ai_in_undo = client.get_value('ai_input')
|
|
temp_undo = client.get_value('temperature')
|
|
print(f"After undo: ai_input={ai_in_undo}, temp={temp_undo}")
|
|
|
|
assert ai_in_undo == "Initial Input"
|
|
assert temp_undo == 0.5
|
|
|
|
# 4. Redo
|
|
print("Sending Redo...")
|
|
client.click('btn_redo')
|
|
time.sleep(1.0)
|
|
|
|
ai_in_redo = client.get_value('ai_input')
|
|
temp_redo = client.get_value('temperature')
|
|
print(f"After redo: ai_input={ai_in_redo}, temp={temp_redo}")
|
|
|
|
assert ai_in_redo == "Modified Input"
|
|
assert temp_redo == 1.5
|
|
|
|
print("Undo/Redo basic lifecycle PASSED.")
|
|
|
|
@pytest.mark.integration
|
|
def test_undo_redo_discussion_mutation(live_gui):
|
|
client = api_hook_client.ApiHookClient()
|
|
assert client.wait_for_server(timeout=15)
|
|
|
|
# Get initial entries count
|
|
initial_entries = client.get_value('disc_entries')
|
|
initial_count = len(initial_entries)
|
|
print(f"Initial entries: {initial_count}")
|
|
|
|
# 1. Add an entry (we simulate this by appending to disc_entries)
|
|
# Wait for settle
|
|
time.sleep(2.0)
|
|
|
|
new_entries = initial_entries + [{"role": "User", "content": "New Entry", "collapsed": False, "ts": "2026-03-11 12:00:00"}]
|
|
client.set_value('disc_entries', new_entries)
|
|
|
|
# Wait for debounce
|
|
time.sleep(2.0)
|
|
assert len(client.get_value('disc_entries')) == initial_count + 1
|
|
|
|
# 2. Undo addition
|
|
print("Undoing entry addition...")
|
|
client.click('btn_undo')
|
|
time.sleep(0.5)
|
|
assert len(client.get_value('disc_entries')) == initial_count
|
|
|
|
# 3. Redo addition
|
|
print("Redoing entry addition...")
|
|
client.click('btn_redo')
|
|
time.sleep(0.5)
|
|
assert len(client.get_value('disc_entries')) == initial_count + 1
|
|
|
|
print("Undo/Redo discussion mutation PASSED.")
|
|
|
|
@pytest.mark.integration
|
|
def test_undo_redo_context_mutation(live_gui):
|
|
client = api_hook_client.ApiHookClient()
|
|
assert client.wait_for_server(timeout=15)
|
|
|
|
# Wait for settle
|
|
time.sleep(2.0)
|
|
|
|
# 1. Add a file
|
|
client.set_value('ui_file_paths', ['test_undo.py'])
|
|
|
|
# Wait for debounce
|
|
time.sleep(2.0)
|
|
assert 'test_undo.py' in client.get_value('ui_file_paths')
|
|
|
|
# 2. Undo addition
|
|
print("Undoing file addition...")
|
|
client.click('btn_undo')
|
|
time.sleep(0.5)
|
|
assert 'test_undo.py' not in client.get_value('ui_file_paths')
|
|
|
|
# 3. Redo addition
|
|
print("Redoing file addition...")
|
|
client.click('btn_redo')
|
|
time.sleep(0.5)
|
|
assert 'test_undo.py' in client.get_value('ui_file_paths')
|
|
|
|
print("Undo/Redo context mutation PASSED.")
|