def mock_segment_win(self, player_name, segment_name): """Mock a player winning a segment.""" winner = self.table.q_board.winners[segment_name] QTest.mouseClick(winner, Qt.LeftButton) for _ in range(self.table.scorer.players.index(player_name) + 1): QTest.keyEvent(QTest.Press, winner, Qt.Key_Down) QTest.keyEvent(QTest.Press, winner, Qt.Key_Enter)
def testDelete(self, mockQuestion): w = main_form.MainForm(file=str(SX_EXAMPLE3)) sxf = w.dataTable.sxfile self.assertEqual(len(sxf), 6) mockQuestion.configure_mock(return_value=main_form.QMessageBox.No) w.slotDelete() # aborted self.assertEqual(len(sxf), 6) mockQuestion.configure_mock(return_value=main_form.QMessageBox.Yes) w.dataTable.setCurrentItem(w.dataTable.item(2, 0)) QTest.keyEvent(QTest.Click, w.dataTable, Qt.Key_Down, Qt.ShiftModifier) w.slotDelete() # success self.assertEqual(len(sxf), 4)