From 62e245e642d157c860ce5b0731ccfe8f21e4ac86 Mon Sep 17 00:00:00 2001
From: eileen <emcnaughton@wikimedia.org>
Date: Thu, 13 Jun 2019 14:53:17 -0400
Subject: [PATCH] Add PrematureExitException to support unit testing where the
 code would otherwise exit

---
 CRM/Core/Exception/PrematureExitException.php | 44 +++++++++++++++++++
 CRM/Utils/System.php                          |  5 +++
 2 files changed, 49 insertions(+)
 create mode 100644 CRM/Core/Exception/PrematureExitException.php

diff --git a/CRM/Core/Exception/PrematureExitException.php b/CRM/Core/Exception/PrematureExitException.php
new file mode 100644
index 0000000000..3b0f574d16
--- /dev/null
+++ b/CRM/Core/Exception/PrematureExitException.php
@@ -0,0 +1,44 @@
+<?php
+/*
+ +--------------------------------------------------------------------+
+ | CiviCRM version 5                                                  |
+ +--------------------------------------------------------------------+
+ | Copyright CiviCRM LLC (c) 2004-2019                                |
+ +--------------------------------------------------------------------+
+ | This file is a part of CiviCRM.                                    |
+ |                                                                    |
+ | CiviCRM is free software; you can copy, modify, and distribute it  |
+ | under the terms of the GNU Affero General Public License           |
+ | Version 3, 19 November 2007 and the CiviCRM Licensing Exception.   |
+ |                                                                    |
+ | CiviCRM is distributed in the hope that it will be useful, but     |
+ | WITHOUT ANY WARRANTY; without even the implied warranty of         |
+ | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.               |
+ | See the GNU Affero General Public License for more details.        |
+ |                                                                    |
+ | You should have received a copy of the GNU Affero General Public   |
+ | License and the CiviCRM Licensing Exception along                  |
+ | with this program; if not, contact CiviCRM LLC                     |
+ | at info[AT]civicrm[DOT]org. If you have questions about the        |
+ | GNU Affero General Public License or the licensing of CiviCRM,     |
+ | see the CiviCRM license FAQ at http://civicrm.org/licensing        |
+ +--------------------------------------------------------------------+
+ */
+
+/**
+ * Exception thrown during tests where live code would exit.
+ *
+ * This is when the code would exit in live mode.
+ *
+ * @param string $message
+ *   The human friendly error message.
+ * @param string $error_code
+ *   A computer friendly error code. By convention, no space (but underscore allowed).
+ *   ex: mandatory_missing, duplicate, invalid_format
+ * @param array $data
+ *   Extra params to return. eg an extra array of ids. It is not mandatory, but can help the computer using the api.
+ * Keep in mind the api consumer isn't to be trusted. eg. the database password is NOT a good extra data.
+ */
+class CRM_Core_Exception_PrematureExitException extends RuntimeException {
+
+}
diff --git a/CRM/Utils/System.php b/CRM/Utils/System.php
index d52e3d5ef4..7508fe1573 100644
--- a/CRM/Utils/System.php
+++ b/CRM/Utils/System.php
@@ -1395,9 +1395,14 @@ class CRM_Utils_System {
    *
    * @param int $status
    *   (optional) Code with which to exit.
+   *
+   * @throws \CRM_Core_PrematureExit_Exception
    */
   public static function civiExit($status = 0) {
 
+    if (CIVICRM_UF === 'UnitTests') {
+      throw new CRM_Core_Exception_PrematureExitException('civiExit called');
+    }
     if ($status > 0) {
       http_response_code(500);
     }
-- 
2.25.1