EBR-79: update existing tests

the-virtual-brain · Jul 19, 2024 · 2f3bdf0 · 2f3bdf0
1 parent fec5ba2
commit 2f3bdf0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tvbwidgets/tests/test_drive_widget.py b/tvbwidgets/tests/test_drive_widget.py
@@ -122,8 +122,8 @@ def mockk(token):
     os.environ[CLB_AUTH] = "test_auth_token"
     widget = StorageWidget()
 
-    widget.api.repos_dropdown.value = widget.api.repos_dropdown.options['repo1']
-    widget.api.files_list.value = widget.api.files_list.options[1]
+    widget.drive_api.repos_dropdown.value = widget.drive_api.repos_dropdown.options['repo1']
+    widget.drive_api.files_list.value = widget.drive_api.files_list.options[1]
 
     assert widget.get_selected_file_name() == DUMMY_FILENAME
     assert widget.get_selected_file_content() == DUMMY_CONTENT
diff --git a/tvbwidgets/tests/test_head_widget.py b/tvbwidgets/tests/test_head_widget.py
@@ -133,8 +133,8 @@ def mockk(token):
     widget._TVBWidgetWithBrowser__display_message('ABC')
     assert widget.message_label.value == HeadBrowser.MSG_TEMPLATE.format('ABC', HeadBrowser.MSG_COLOR)
 
-    widget.storage_widget.api.repos_dropdown.value = widget.storage_widget.api.repos_dropdown.options['repo1']
-    widget.storage_widget.api.files_list.value = widget.storage_widget.api.files_list.options[1]
+    widget.storage_widget.drive_api.repos_dropdown.value = widget.storage_widget.drive_api.repos_dropdown.options['repo1']
+    widget.storage_widget.drive_api.files_list.value = widget.storage_widget.drive_api.files_list.options[1]
 
     widget.load_selected_file(Surface)
     assert 'Only .zip' in widget.message_label.value