test_threads.py

import openai

import openai_responses
from openai_responses import OpenAIMock


@openai_responses.mock()
def test_create_thread(openai_mock: OpenAIMock):
    client = openai.Client(api_key="sk-fake123")

    thread = client.beta.threads.create()

    assert thread.id
    assert openai_mock.beta.threads.create.route.call_count == 1


@openai_responses.mock()
def test_create_thread_with_additional_messages(openai_mock: OpenAIMock):
    client = openai.Client(api_key="sk-fake123")

    thread = client.beta.threads.create(
        messages=[{"role": "assistant", "content": "How can I help?"}]
    )

    messages = client.beta.threads.messages.list(thread.id)

    assert thread.id
    assert len(messages.data) == 1
    assert openai_mock.beta.threads.create.route.call_count == 1


@openai_responses.mock()
def test_retrieve_thread(openai_mock: OpenAIMock):
    client = openai.Client(api_key="sk-fake123")

    thread = client.beta.threads.create()
    found = client.beta.threads.retrieve(thread.id)

    assert found.id == thread.id
    assert openai_mock.beta.threads.create.route.call_count == 1
    assert openai_mock.beta.threads.retrieve.route.call_count == 1


@openai_responses.mock()
def test_update_thread(openai_mock: OpenAIMock):
    client = openai.Client(api_key="sk-fake123")

    thread = client.beta.threads.create(metadata={"foo": "bar"})
    updated = client.beta.threads.update(thread.id, metadata={"foo": "baz"})

    assert updated.id == thread.id
    assert thread.metadata == {"foo": "bar"}
    assert updated.metadata == {"foo": "baz"}
    assert openai_mock.beta.threads.create.route.call_count == 1
    assert openai_mock.beta.threads.update.route.call_count == 1


@openai_responses.mock()
def test_delete_thread(openai_mock: OpenAIMock):
    client = openai.Client(api_key="sk-fake123")

    thread = client.beta.threads.create()

    assert client.beta.threads.delete(thread.id).deleted
    assert openai_mock.beta.threads.create.route.call_count == 1
    assert openai_mock.beta.threads.delete.route.call_count == 1