import pytest
from src.graph import build_graph


def test_graph_flow():
    graph = build_graph()
    input_state = {"input": "Hello world"}
    result = graph.invoke(input_state)
    assert "rewritten" in result
    expected = (
        "I notice that you said: 'Hello world'. "
        "Let's reflect on that."
    )
    assert result["rewritten"] == expected