Changeset 7664 for trunk/tests/includes/testcase.php

Timestamp:

12/12/2013 08:43:28 PM (11 years ago)

Author:

boonebgorges

Message:

Always fake $_SERVERSERVER_NAME? when invoking wp_mail() during tests

This is a blanket fix for PHP errors that otherwise arise when calling any BP
function that results in an email notification being sent.

File:

• trunk/tests/includes/testcase.php (modified) (4 diffs)

trunk/tests/includes/testcase.php

@@ -12 +12 @@
 
     protected $temp_has_bp_moderate = array();
+    protected $cached_SERVER_NAME = null;
 
     public function setUp() {
@@ -23 +24 @@
         global $wpdb;
         $wpdb->query( "TRUNCATE TABLE {$wpdb->users}" );
+
+        // Fake WP mail globals, to avoid errors
+        add_filter( 'wp_mail', array( $this, 'setUp_wp_mail' ) );
+        add_filter( 'wp_mail_from', array( $this, 'tearDown_wp_mail' ) );
 
         $this->factory = new BP_UnitTest_Factory;
@@ -305 +310 @@
      * Set up globals necessary to avoid errors when using wp_mail()
      */
-    public function setUp_wp_mail() {
+    public function setUp_wp_mail( $args ) {
+        if ( isset( $_SERVER['SERVER_NAME'] ) ) {
+            $this->cached_SERVER_NAME = $_SERVER['SERVER_NAME'];
+        }
+
         $_SERVER['SERVER_NAME'] = 'example.com';
+
+        // passthrough
+        return $args;
     }
 
@@ -312 +324 @@
      * Tear down globals set up in setUp_wp_mail()
      */
-    public function tearDown_wp_mail() {
-        unset( $_SERVER['SERVER_NAME'] );
+    public function tearDown_wp_mail( $args ) {
+        if ( ! empty( $this->cached_SERVER_NAME ) ) {
+            $_SERVER['SERVER_NAME'] = $this->cached_SERVER_NAME;
+            unset( $this->cached_SERVER_NAME );
+        } else {
+            unset( $_SERVER['SERVER_NAME'] );
+        }
+
+        // passthrough
+        return $args;
     }
 }