From f6b5e1293d43b57c6778d993d6ac16062d423cdc Mon Sep 17 00:00:00 2001 From: Alex Pyrgiotis Date: Fri, 21 Jul 2023 18:07:17 +0300 Subject: [PATCH] gui: Add references to dialog buttons Add references to dialog buttons, so that we can click on them from our GUI tests. --- dangerzone/gui/logic.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/dangerzone/gui/logic.py b/dangerzone/gui/logic.py index c1810b1..05f5bbc 100644 --- a/dangerzone/gui/logic.py +++ b/dangerzone/gui/logic.py @@ -149,21 +149,22 @@ class Dialog(QtWidgets.QDialog): message_layout = self.create_layout() - ok_button = QtWidgets.QPushButton(ok_text) - ok_button.clicked.connect(self.clicked_ok) + self.ok_button = QtWidgets.QPushButton(ok_text) + self.ok_button.clicked.connect(self.clicked_ok) + if extra_button_text: - extra_button = QtWidgets.QPushButton(extra_button_text) - extra_button.clicked.connect(self.clicked_extra) + self.extra_button = QtWidgets.QPushButton(extra_button_text) + self.extra_button.clicked.connect(self.clicked_extra) buttons_layout = QtWidgets.QHBoxLayout() buttons_layout.addStretch() - buttons_layout.addWidget(ok_button) + buttons_layout.addWidget(self.ok_button) if extra_button_text: - buttons_layout.addWidget(extra_button) + buttons_layout.addWidget(self.extra_button) if has_cancel: - cancel_button = QtWidgets.QPushButton(cancel_text) - cancel_button.clicked.connect(self.clicked_cancel) - buttons_layout.addWidget(cancel_button) + self.cancel_button = QtWidgets.QPushButton(cancel_text) + self.cancel_button.clicked.connect(self.clicked_cancel) + buttons_layout.addWidget(self.cancel_button) layout = QtWidgets.QVBoxLayout() layout.addLayout(message_layout)