fixup!: test_deleted_file_filewidget

freedomofpress · Oct 25, 2024 · 907ebcd · 907ebcd
1 parent 4e20eee
commit 907ebcd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/client/tests/functional/test_deleted_file_filewidget.py b/client/tests/functional/test_deleted_file_filewidget.py
@@ -43,15 +43,15 @@ def conversation_with_file_is_rendered():
         assert gui.main_view.view_layout.currentIndex() == gui.main_view.CONVERSATION_INDEX
         conversation = gui.main_view.view_layout.widget(gui.main_view.view_layout.currentIndex())
         assert isinstance(conversation, SourceConversationWrapper)
-        file_id = list(conversation.current_messages.keys())[2]
-        file_widget = conversation.current_messages[file_id]
+        file_id = list(conversation.conversation_view.current_messages.keys())[2]
+        file_widget = conversation.conversation_view.current_messages[file_id]
         assert isinstance(file_widget, FileWidget)
 
     # Get the selected source conversation that contains a file attachment
     qtbot.waitUntil(conversation_with_file_is_rendered, timeout=TIME_RENDER_CONV_VIEW)
     conversation = gui.main_view.view_layout.widget(gui.main_view.view_layout.currentIndex())
-    file_id = list(conversation.current_messages.keys())[2]
-    file_widget = conversation.current_messages[file_id]
+    file_id = list(conversation.conversation_view.current_messages.keys())[2]
+    file_widget = conversation.conversation_view.current_messages[file_id]
 
     deleteLater = mocker.patch.object(file_widget, "deleteLater")