Chainlit context not found in unit tests for applications

[dokato]

Describe the bug
My unittests fails with:

chainlit.context.ChainlitContextException: Chainlit context not found

To Reproduce

I have a function that generates chainlit text:

def _generate_message_text(txt:str) ->cl.Text:
    return cl.Text(name="Message", content=f"message: {txt}", display="inline")

but any sort of unit test like here

def test_generate_message_text():
    cl_text = _generate_message_text("cat")
    assert 1==1

results in an error as above. What is recommended way to write unittests for functions that generate chainlit objects?

Expected behavior
An easy way to unit test chainlit generation elements outside of contex, or easy mechanism to mock the context.

chainlit ver.: 1.0.200

[dokterbob]

Hi @dokato, excellent question!

In our internal unit tests, we use a pytest fixture to create a ChainlitContext.

chainlit/backend/tests/conftest.py

Line 56 in ff26451

async def create_chainlit_context(mock_session):

I'd be very curious to learn whether and how you manage to use something similar to do testing of applications built with chainlit.

Lastly, I would strongly suggest you upgrade if you are offering a public-facing application. There's a lot of security issues in chainlit as well as the dependencies fixed in later releases.