diff --git a/tests/smoke/test_app.py b/tests/smoke/test_app.py index 0f161e8..28f2fc9 100644 --- a/tests/smoke/test_app.py +++ b/tests/smoke/test_app.py @@ -2,7 +2,14 @@ def test_app(mocker, default_voice_chat_configs): - mocker.patch("streamlit.session_state", {}) + class MockAttrDict(dict): + def __getattr__(self, attr): + return self.get(attr, mocker.MagicMock()) + + def __setattr__(self, attr, value): + self[attr] = value + + mocker.patch("streamlit.session_state", MockAttrDict()) mocker.patch("streamlit.chat_input", return_value="foobar") mocker.patch( "pyrobbot.chat_configs.VoiceChatConfigs.from_file",