Fixes to web app test

paulovcmedeiros · Feb 25, 2024 · f54024a · f54024a
1 parent d5fe814
commit f54024a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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",