def new_config_btn_clicked(self): """ Open the Config Entry dialog window. If database is not connected, display error message. """ if not db_connected(): QMessageBox.critical( self, 'Database connection required', 'Database connection is required to edit the PV configuration.', QMessageBox.Ok) return self.config_entry_w.show() self.config_entry_w.activateWindow()
def update_db_connection_status(self): if db_connected(): set_colored_text(self.db_connection_status, 'connected', QColor('green')) else: set_colored_text(self.db_connection_status, 'not connected', QColor('red')) self.db_connection_last_checked.setText( f"Connection status last updated on: " f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S:%f')}")
def connect_to_db(self): db_models.initialize_database(name=self.HeliumDB.name, host=self.HeliumDB.host, user=self.HeliumDB.user, password=self.HeliumDB.password) try: db_connect() except DBConnectionError: log_exception(*sys.exc_info()) finally: return db_connected()
def edit_config_btn_clicked(self): """ On Edit, open Config Entry dialog window as normal with object selected and PV config loaded. """ if not db_connected(): QMessageBox.critical( self, 'Database connection required', 'Database connection is required to edit the PV configuration.', QMessageBox.Ok) return self.config_entry_w.show() selected_row_items = self.config_table.selectedItems() if selected_row_items: selected_object_name = selected_row_items[1].text() self.config_entry_w.edit_object_config( obj_name=selected_object_name) self.config_entry_w.activateWindow()
def test_get_max_object_id_WHEN_not_present_THEN_returns_zero(self): with mock_database.Database(): self.assertTrue(db_func.db_connected()) self.assertEqual(db_func.get_max_object_id(), 0)
def test_db_connected_WHEN_db_not_connected_THEN_returns_false( self, mock_func): mock_func.return_value = False self.assertFalse(db_func.db_connected()) mock_func.assert_called()
def test_db_connected_WHEN_db_connected_THEN_returns_true(self, mock_func): mock_func.return_value = True self.assertTrue(db_func.db_connected()) mock_func.assert_called()