def test_show_request_exception(self, mock_db, mock_print): mock_db().get_requests.side_effect = Exception bdo = BDO() bdo.show_requests() mock_print.assert_called_once() mock_db().get_requests.assert_called_once() mock_db().rollback_data.assert_called_once()
def test_show_request_rejected_approval(self, mock_menu, mock_db, mock_print, mock_input, mock_table): mock_db().get_requests.return_value.description.return_value = [ 'dummy', 'dummy', 'dummy', 'dummy', 'dummy' ] mock_db().get_requests.return_value.fetchall.return_value = [[ 'APPROVAL', 'dummy', 'MEMBER|sample_member|ea32d0e8-' '298a-4967-9fbd-72dc595294fb|334aabf5-' '8663-492d-bf75-6a5d7ece032d|project1', 'dummy', 'dummy' ]] mock_menu().draw_menu.return_value = 'REJECTED' mock_input.side_effect = ['0', ''] bdo = BDO() bdo.show_requests() mock_menu().draw_menu.assert_called_once() mock_table.assert_called_once() mock_db().get_requests.assert_called_once() mock_db().resolve_request.assert_called_once() mock_db().commit_data.assert_called_once() mock_db().get_requests().fetchall.assert_called_once() self.assertEqual(mock_print.call_count, 3) self.assertEqual(mock_input.call_count, 2)
def test_show_request_no_request(self, mock_db, mock_print): mock_db().get_requests.return_value.fetchall.return_value = [] mock_db().get_requests.return_value.description.return_value = () bdo = BDO() bdo.show_requests() mock_db().get_requests.assert_called_once() mock_db().get_requests().fetchall.assert_called_once() self.assertEqual(mock_print.call_count, 1)