General testing improvements

activity_browser/actions/database/database_export.py

@@ -13,6 +13,8 @@ class DatabaseExport(ABAction):
     title = "Export database..."
     tool_tip = "Export a database from this project"
 
+    wizard: DatabaseExportWizard
+
     def onTrigger(self, toggled):
-        wizard = DatabaseExportWizard(application.main_window)
-        wizard.show()
+        self.wizard = DatabaseExportWizard(application.main_window)
+        self.wizard.show()

activity_browser/actions/database/database_import.py

@@ -9,7 +9,8 @@ class DatabaseImport(ABAction):
     icon = qicons.import_db
     title = "Import database..."
     tool_tip = "Import a new database"
+    wizard: DatabaseImportWizard
 
     def onTrigger(self, toggled):
-        wizard = DatabaseImportWizard(application.main_window)
-        wizard.show()
+        self.wizard = DatabaseImportWizard(application.main_window)
+        self.wizard.show()

tests/actions/test_activity_actions.py

@@ -5,8 +5,15 @@
 from activity_browser.ui.widgets.dialog import LocationLinkingDialog, ActivityLinkingDialog
 
 
-def test_activity_delete(ab_app):
+def test_activity_delete(ab_app, monkeypatch):
     key = ('activity_tests', '330b935a46bc4ad39530ab7df012f38b')
+
+    monkeypatch.setattr(
+        QtWidgets.QMessageBox, 'warning',
+        staticmethod(lambda *args, **kwargs: QtWidgets.QMessageBox.Yes)
+    )
+
+
     assert bw.projects.current == "default"
     assert bw.get_activity(key)
 

tests/actions/test_database_actions.py

@@ -45,11 +45,19 @@ def test_database_duplicate(ab_app, monkeypatch, qtbot):
 
 def test_database_export(ab_app):
     # TODO: implement when we've redone the export wizard and actions
+    action = actions.DatabaseExport(None)
+    action.trigger()
+    assert action.wizard.isVisible()
+    action.wizard.destroy()
     return
 
 
 def test_database_import(ab_app):
     # TODO: implement when we've redone the import wizard and actions
+    action = actions.DatabaseImport(None)
+    action.trigger()
+    assert action.wizard.isVisible()
+    action.wizard.destroy()
     return
 
 
@@ -61,13 +69,24 @@ def test_database_new(ab_app, monkeypatch):
         staticmethod(lambda *args, **kwargs: ('db_that_is_new', True))
     )
 
+    monkeypatch.setattr(
+        QtWidgets.QMessageBox, 'information',
+        staticmethod(lambda *args, **kwargs: True)
+    )
+
     assert bw.projects.current == "default"
     assert new_db not in bw.databases
 
     actions.DatabaseNew(None).trigger()
 
     assert new_db in bw.databases
 
+    db_number = len(bw.databases)
+
+    actions.DatabaseNew(None).trigger()
+
+    assert db_number == len(bw.databases)
+
 
 def test_database_relink(ab_app, monkeypatch):
     db = "db_to_relink"