Changeset 7364

Timestamp:

09/02/2013 02:52:44 AM (11 years ago)

Author:

boonebgorges

Message:

Introduces grant_bp_moderate() to testcase

In some cases, we have to test on the fly whether a given user has the
bp_moderate cap. This utility method makes it easy to do so in the unit
tests. Note that this implementation depends on using bp_current_user_can(),
rather than the WP equivalent.

File:

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

trunk/tests/includes/testcase.php

@@ -10 +10 @@
 
 class BP_UnitTestCase extends WP_UnitTestCase {
+
+    protected $temp_has_bp_moderate = array();
 
     public function setUp() {
@@ -264 +266 @@
     }
 
+    public function grant_bp_moderate( $user_id ) {
+        if ( ! isset( $this->temp_has_bp_moderate[ $user_id ] ) ) {
+            $this->temp_has_bp_moderate[ $user_id ] = 1;
+        }
+        add_filter( 'bp_current_user_can', array( $this, 'grant_bp_moderate_cb' ), 10, 2 );
+    }
+
+    public function revoke_bp_moderate( $user_id ) {
+        if ( isset( $this->temp_has_bp_moderate[ $user_id ] ) ) {
+            unset( $this->temp_has_bp_moderate[ $user_id ] );
+        }
+        remove_filter( 'bp_current_user_can', array( $this, 'grant_bp_moderate_cb' ), 10, 2 );
+    }
+
+    public function grant_bp_moderate_cb( $retval, $capability ) {
+        $current_user = bp_loggedin_user_id();
+        if ( ! isset( $this->temp_has_bp_moderate[ $current_user ] ) ) {
+            return $retval;
+        }
+
+        if ( 'bp_moderate' == $capability ) {
+            $retval = true;
+        }
+
+        return $retval;
+    }
+
     /**
      * Go to the root blog. This helps reset globals after moving between