Minor changes required for running tests

[ipypbx] / src / ipypbx / main.py

diff --git a/src/ipypbx/main.py b/src/ipypbx/main.py
index 9c8ebfe..f0422d6 100644 (file)
--- a/src/ipypbx/main.py
+++ b/src/ipypbx/main.py
@@ -58,7 +58,7 @@ def setupDb(prefix=PREFIX, dbname=DB_NAME):
             for query_string in sql_queries[:-1]:
                 query.exec_(query_string)
 
-        return created
+        return db, created
     else:
         # Something went horribly wrong.
         QtGui.QMessageBox.warning(
@@ -77,7 +77,7 @@ if __name__ == '__main__':
         QtGui.QApplication.installTranslator(translator)
 
     # Initialize main window.
-    created = setupDb()
+    db, created = setupDb()
     main = QtGui.QMainWindow()
     views = ui.Ui_MainWindow()
     views.setupUi(main)
@@ -107,6 +107,7 @@ if __name__ == '__main__':
             QtCore.SIGNAL('currentRowChanged(QModelIndex, QModelIndex)'),
             connection_index, connection_index)
 
+    # TODO: not needed anymore?
     if created:
         main.controllers['connection'].add()
     sys.exit(app.exec_())