686 lines
20 KiB
PHP
686 lines
20 KiB
PHP
<?php
|
|
|
|
require_once dirname( __FILE__ ) . '/factory.php';
|
|
require_once dirname( __FILE__ ) . '/trac.php';
|
|
|
|
class WP_UnitTestCase extends PHPUnit_Framework_TestCase {
|
|
|
|
protected static $forced_tickets = array();
|
|
protected $expected_deprecated = array();
|
|
protected $caught_deprecated = array();
|
|
protected $expected_doing_it_wrong = array();
|
|
protected $caught_doing_it_wrong = array();
|
|
|
|
protected static $hooks_saved = array();
|
|
protected static $ignore_files;
|
|
|
|
/**
|
|
* @var WP_UnitTest_Factory
|
|
*/
|
|
protected static $factory;
|
|
|
|
public static function get_called_class() {
|
|
if ( function_exists( 'get_called_class' ) ) {
|
|
return get_called_class();
|
|
}
|
|
|
|
// PHP 5.2 only
|
|
$backtrace = debug_backtrace();
|
|
// [0] WP_UnitTestCase::get_called_class()
|
|
// [1] WP_UnitTestCase::setUpBeforeClass()
|
|
if ( 'call_user_func' === $backtrace[2]['function'] ) {
|
|
return $backtrace[2]['args'][0][0];
|
|
}
|
|
return $backtrace[2]['class'];
|
|
}
|
|
|
|
public static function setUpBeforeClass() {
|
|
parent::setUpBeforeClass();
|
|
|
|
if ( ! self::$factory ) {
|
|
self::$factory = new WP_UnitTest_Factory();
|
|
}
|
|
|
|
$c = self::get_called_class();
|
|
if ( ! method_exists( $c, 'wpSetUpBeforeClass' ) ) {
|
|
return;
|
|
}
|
|
|
|
call_user_func( array( $c, 'wpSetUpBeforeClass' ), self::$factory );
|
|
|
|
self::commit_transaction();
|
|
}
|
|
|
|
public static function tearDownAfterClass() {
|
|
parent::tearDownAfterClass();
|
|
|
|
$c = self::get_called_class();
|
|
if ( ! method_exists( $c, 'wpTearDownAfterClass' ) ) {
|
|
return;
|
|
}
|
|
|
|
call_user_func( array( $c, 'wpTearDownAfterClass' ) );
|
|
|
|
self::commit_transaction();
|
|
}
|
|
|
|
function setUp() {
|
|
set_time_limit(0);
|
|
|
|
if ( ! self::$ignore_files ) {
|
|
self::$ignore_files = $this->scan_user_uploads();
|
|
}
|
|
|
|
if ( ! self::$hooks_saved ) {
|
|
$this->_backup_hooks();
|
|
}
|
|
|
|
global $wpdb, $wp_rewrite;
|
|
$wpdb->suppress_errors = false;
|
|
$wpdb->show_errors = true;
|
|
$wpdb->db_connect();
|
|
ini_set('display_errors', 1 );
|
|
$this->clean_up_global_scope();
|
|
|
|
/*
|
|
* When running core tests, ensure that post types and taxonomies
|
|
* are reset for each test. We skip this step for non-core tests,
|
|
* given the large number of plugins that register post types and
|
|
* taxonomies at 'init'.
|
|
*/
|
|
if ( defined( 'WP_RUN_CORE_TESTS' ) && WP_RUN_CORE_TESTS ) {
|
|
$this->reset_post_types();
|
|
$this->reset_taxonomies();
|
|
$this->reset_post_statuses();
|
|
|
|
if ( $wp_rewrite->permalink_structure ) {
|
|
$this->set_permalink_structure( '' );
|
|
}
|
|
}
|
|
|
|
$this->start_transaction();
|
|
$this->expectDeprecated();
|
|
add_filter( 'wp_die_handler', array( $this, 'get_wp_die_handler' ) );
|
|
|
|
add_filter( 'wp_mail', array( $this, 'set_wp_mail_globals' ) );
|
|
}
|
|
|
|
/**
|
|
* Detect post-test failure conditions.
|
|
*
|
|
* We use this method to detect expectedDeprecated and expectedIncorrectUsage annotations.
|
|
*
|
|
* @since 4.2.0
|
|
*/
|
|
protected function assertPostConditions() {
|
|
$this->expectedDeprecated();
|
|
}
|
|
|
|
function tearDown() {
|
|
global $wpdb, $wp_query, $post;
|
|
$wpdb->query( 'ROLLBACK' );
|
|
if ( is_multisite() ) {
|
|
while ( ms_is_switched() ) {
|
|
restore_current_blog();
|
|
}
|
|
}
|
|
$wp_query = new WP_Query();
|
|
$post = null;
|
|
remove_theme_support( 'html5' );
|
|
remove_filter( 'query', array( $this, '_create_temporary_tables' ) );
|
|
remove_filter( 'query', array( $this, '_drop_temporary_tables' ) );
|
|
remove_filter( 'wp_die_handler', array( $this, 'get_wp_die_handler' ) );
|
|
$this->_restore_hooks();
|
|
wp_set_current_user( 0 );
|
|
}
|
|
|
|
function clean_up_global_scope() {
|
|
$_GET = array();
|
|
$_POST = array();
|
|
$this->flush_cache();
|
|
}
|
|
|
|
/**
|
|
* Unregister existing post types and register defaults.
|
|
*
|
|
* Run before each test in order to clean up the global scope, in case
|
|
* a test forgets to unregister a post type on its own, or fails before
|
|
* it has a chance to do so.
|
|
*/
|
|
protected function reset_post_types() {
|
|
foreach ( get_post_types() as $pt ) {
|
|
_unregister_post_type( $pt );
|
|
}
|
|
create_initial_post_types();
|
|
}
|
|
|
|
/**
|
|
* Unregister existing taxonomies and register defaults.
|
|
*
|
|
* Run before each test in order to clean up the global scope, in case
|
|
* a test forgets to unregister a taxonomy on its own, or fails before
|
|
* it has a chance to do so.
|
|
*/
|
|
protected function reset_taxonomies() {
|
|
foreach ( get_taxonomies() as $tax ) {
|
|
_unregister_taxonomy( $tax );
|
|
}
|
|
create_initial_taxonomies();
|
|
}
|
|
|
|
/**
|
|
* Unregister non-built-in post statuses.
|
|
*/
|
|
protected function reset_post_statuses() {
|
|
foreach ( get_post_stati( array( '_builtin' => false ) ) as $post_status ) {
|
|
_unregister_post_status( $post_status );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Saves the action and filter-related globals so they can be restored later.
|
|
*
|
|
* Stores $merged_filters, $wp_actions, $wp_current_filter, and $wp_filter
|
|
* on a class variable so they can be restored on tearDown() using _restore_hooks().
|
|
*
|
|
* @global array $merged_filters
|
|
* @global array $wp_actions
|
|
* @global array $wp_current_filter
|
|
* @global array $wp_filter
|
|
* @return void
|
|
*/
|
|
protected function _backup_hooks() {
|
|
$globals = array( 'merged_filters', 'wp_actions', 'wp_current_filter', 'wp_filter' );
|
|
foreach ( $globals as $key ) {
|
|
self::$hooks_saved[ $key ] = $GLOBALS[ $key ];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Restores the hook-related globals to their state at setUp()
|
|
* so that future tests aren't affected by hooks set during this last test.
|
|
*
|
|
* @global array $merged_filters
|
|
* @global array $wp_actions
|
|
* @global array $wp_current_filter
|
|
* @global array $wp_filter
|
|
* @return void
|
|
*/
|
|
protected function _restore_hooks() {
|
|
$globals = array( 'merged_filters', 'wp_actions', 'wp_current_filter', 'wp_filter' );
|
|
foreach ( $globals as $key ) {
|
|
if ( isset( self::$hooks_saved[ $key ] ) ) {
|
|
$GLOBALS[ $key ] = self::$hooks_saved[ $key ];
|
|
}
|
|
}
|
|
}
|
|
|
|
function flush_cache() {
|
|
global $wp_object_cache;
|
|
$wp_object_cache->group_ops = array();
|
|
$wp_object_cache->stats = array();
|
|
$wp_object_cache->memcache_debug = array();
|
|
$wp_object_cache->cache = array();
|
|
if ( method_exists( $wp_object_cache, '__remoteset' ) ) {
|
|
$wp_object_cache->__remoteset();
|
|
}
|
|
wp_cache_flush();
|
|
wp_cache_add_global_groups( array( 'users', 'userlogins', 'usermeta', 'user_meta', 'site-transient', 'site-options', 'site-lookup', 'blog-lookup', 'blog-details', 'rss', 'global-posts', 'blog-id-cache' ) );
|
|
wp_cache_add_non_persistent_groups( array( 'comment', 'counts', 'plugins' ) );
|
|
}
|
|
|
|
function start_transaction() {
|
|
global $wpdb;
|
|
$wpdb->query( 'SET autocommit = 0;' );
|
|
$wpdb->query( 'START TRANSACTION;' );
|
|
add_filter( 'query', array( $this, '_create_temporary_tables' ) );
|
|
add_filter( 'query', array( $this, '_drop_temporary_tables' ) );
|
|
}
|
|
|
|
/**
|
|
* Commit the queries in a transaction.
|
|
*
|
|
* @since 4.1.0
|
|
*/
|
|
public static function commit_transaction() {
|
|
global $wpdb;
|
|
$wpdb->query( 'COMMIT;' );
|
|
}
|
|
|
|
function _create_temporary_tables( $query ) {
|
|
if ( 'CREATE TABLE' === substr( trim( $query ), 0, 12 ) )
|
|
return substr_replace( trim( $query ), 'CREATE TEMPORARY TABLE', 0, 12 );
|
|
return $query;
|
|
}
|
|
|
|
function _drop_temporary_tables( $query ) {
|
|
if ( 'DROP TABLE' === substr( trim( $query ), 0, 10 ) )
|
|
return substr_replace( trim( $query ), 'DROP TEMPORARY TABLE', 0, 10 );
|
|
return $query;
|
|
}
|
|
|
|
function get_wp_die_handler( $handler ) {
|
|
return array( $this, 'wp_die_handler' );
|
|
}
|
|
|
|
function wp_die_handler( $message ) {
|
|
throw new WPDieException( $message );
|
|
}
|
|
|
|
function expectDeprecated() {
|
|
$annotations = $this->getAnnotations();
|
|
foreach ( array( 'class', 'method' ) as $depth ) {
|
|
if ( ! empty( $annotations[ $depth ]['expectedDeprecated'] ) )
|
|
$this->expected_deprecated = array_merge( $this->expected_deprecated, $annotations[ $depth ]['expectedDeprecated'] );
|
|
if ( ! empty( $annotations[ $depth ]['expectedIncorrectUsage'] ) )
|
|
$this->expected_doing_it_wrong = array_merge( $this->expected_doing_it_wrong, $annotations[ $depth ]['expectedIncorrectUsage'] );
|
|
}
|
|
add_action( 'deprecated_function_run', array( $this, 'deprecated_function_run' ) );
|
|
add_action( 'deprecated_argument_run', array( $this, 'deprecated_function_run' ) );
|
|
add_action( 'doing_it_wrong_run', array( $this, 'doing_it_wrong_run' ) );
|
|
add_action( 'deprecated_function_trigger_error', '__return_false' );
|
|
add_action( 'deprecated_argument_trigger_error', '__return_false' );
|
|
add_action( 'doing_it_wrong_trigger_error', '__return_false' );
|
|
}
|
|
|
|
function expectedDeprecated() {
|
|
$errors = array();
|
|
|
|
$not_caught_deprecated = array_diff( $this->expected_deprecated, $this->caught_deprecated );
|
|
foreach ( $not_caught_deprecated as $not_caught ) {
|
|
$errors[] = "Failed to assert that $not_caught triggered a deprecated notice";
|
|
}
|
|
|
|
$unexpected_deprecated = array_diff( $this->caught_deprecated, $this->expected_deprecated );
|
|
foreach ( $unexpected_deprecated as $unexpected ) {
|
|
$errors[] = "Unexpected deprecated notice for $unexpected";
|
|
}
|
|
|
|
$not_caught_doing_it_wrong = array_diff( $this->expected_doing_it_wrong, $this->caught_doing_it_wrong );
|
|
foreach ( $not_caught_doing_it_wrong as $not_caught ) {
|
|
$errors[] = "Failed to assert that $not_caught triggered an incorrect usage notice";
|
|
}
|
|
|
|
$unexpected_doing_it_wrong = array_diff( $this->caught_doing_it_wrong, $this->expected_doing_it_wrong );
|
|
foreach ( $unexpected_doing_it_wrong as $unexpected ) {
|
|
$errors[] = "Unexpected incorrect usage notice for $unexpected";
|
|
}
|
|
|
|
if ( ! empty( $errors ) ) {
|
|
$this->fail( implode( "\n", $errors ) );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Declare an expected `_deprecated_function()` or `_deprecated_argument()` call from within a test.
|
|
*
|
|
* @since 4.2.0
|
|
*
|
|
* @param string $deprecated Name of the function, method, class, or argument that is deprecated. Must match
|
|
* first parameter of the `_deprecated_function()` or `_deprecated_argument()` call.
|
|
*/
|
|
public function setExpectedDeprecated( $deprecated ) {
|
|
array_push( $this->expected_deprecated, $deprecated );
|
|
}
|
|
|
|
/**
|
|
* Declare an expected `_doing_it_wrong()` call from within a test.
|
|
*
|
|
* @since 4.2.0
|
|
*
|
|
* @param string $deprecated Name of the function, method, or class that appears in the first argument of the
|
|
* source `_doing_it_wrong()` call.
|
|
*/
|
|
public function setExpectedIncorrectUsage( $doing_it_wrong ) {
|
|
array_push( $this->expected_doing_it_wrong, $doing_it_wrong );
|
|
}
|
|
|
|
function deprecated_function_run( $function ) {
|
|
if ( ! in_array( $function, $this->caught_deprecated ) )
|
|
$this->caught_deprecated[] = $function;
|
|
}
|
|
|
|
function doing_it_wrong_run( $function ) {
|
|
if ( ! in_array( $function, $this->caught_doing_it_wrong ) )
|
|
$this->caught_doing_it_wrong[] = $function;
|
|
}
|
|
|
|
function assertWPError( $actual, $message = '' ) {
|
|
$this->assertInstanceOf( 'WP_Error', $actual, $message );
|
|
}
|
|
|
|
function assertNotWPError( $actual, $message = '' ) {
|
|
if ( is_wp_error( $actual ) && '' === $message ) {
|
|
$message = $actual->get_error_message();
|
|
}
|
|
$this->assertNotInstanceOf( 'WP_Error', $actual, $message );
|
|
}
|
|
|
|
function assertEqualFields( $object, $fields ) {
|
|
foreach( $fields as $field_name => $field_value ) {
|
|
if ( $object->$field_name != $field_value ) {
|
|
$this->fail();
|
|
}
|
|
}
|
|
}
|
|
|
|
function assertDiscardWhitespace( $expected, $actual ) {
|
|
$this->assertEquals( preg_replace( '/\s*/', '', $expected ), preg_replace( '/\s*/', '', $actual ) );
|
|
}
|
|
|
|
function assertEqualSets( $expected, $actual ) {
|
|
sort( $expected );
|
|
sort( $actual );
|
|
$this->assertEquals( $expected, $actual );
|
|
}
|
|
|
|
function assertEqualSetsWithIndex( $expected, $actual ) {
|
|
ksort( $expected );
|
|
ksort( $actual );
|
|
$this->assertEquals( $expected, $actual );
|
|
}
|
|
|
|
function go_to( $url ) {
|
|
// note: the WP and WP_Query classes like to silently fetch parameters
|
|
// from all over the place (globals, GET, etc), which makes it tricky
|
|
// to run them more than once without very carefully clearing everything
|
|
$_GET = $_POST = array();
|
|
foreach (array('query_string', 'id', 'postdata', 'authordata', 'day', 'currentmonth', 'page', 'pages', 'multipage', 'more', 'numpages', 'pagenow') as $v) {
|
|
if ( isset( $GLOBALS[$v] ) ) unset( $GLOBALS[$v] );
|
|
}
|
|
$parts = parse_url($url);
|
|
if (isset($parts['scheme'])) {
|
|
$req = isset( $parts['path'] ) ? $parts['path'] : '';
|
|
if (isset($parts['query'])) {
|
|
$req .= '?' . $parts['query'];
|
|
// parse the url query vars into $_GET
|
|
parse_str($parts['query'], $_GET);
|
|
}
|
|
} else {
|
|
$req = $url;
|
|
}
|
|
if ( ! isset( $parts['query'] ) ) {
|
|
$parts['query'] = '';
|
|
}
|
|
|
|
$_SERVER['REQUEST_URI'] = $req;
|
|
unset($_SERVER['PATH_INFO']);
|
|
|
|
$this->flush_cache();
|
|
unset($GLOBALS['wp_query'], $GLOBALS['wp_the_query']);
|
|
$GLOBALS['wp_the_query'] = new WP_Query();
|
|
$GLOBALS['wp_query'] = $GLOBALS['wp_the_query'];
|
|
$GLOBALS['wp'] = new WP();
|
|
_cleanup_query_vars();
|
|
|
|
$GLOBALS['wp']->main($parts['query']);
|
|
}
|
|
|
|
protected function checkRequirements() {
|
|
parent::checkRequirements();
|
|
|
|
// Core tests no longer check against open Trac tickets, but others using WP_UnitTestCase may do so.
|
|
if ( defined( 'WP_RUN_CORE_TESTS' ) && WP_RUN_CORE_TESTS ) {
|
|
return;
|
|
}
|
|
|
|
if ( WP_TESTS_FORCE_KNOWN_BUGS )
|
|
return;
|
|
$tickets = PHPUnit_Util_Test::getTickets( get_class( $this ), $this->getName( false ) );
|
|
foreach ( $tickets as $ticket ) {
|
|
if ( is_numeric( $ticket ) ) {
|
|
$this->knownWPBug( $ticket );
|
|
} elseif ( 'UT' == substr( $ticket, 0, 2 ) ) {
|
|
$ticket = substr( $ticket, 2 );
|
|
if ( $ticket && is_numeric( $ticket ) )
|
|
$this->knownUTBug( $ticket );
|
|
} elseif ( 'Plugin' == substr( $ticket, 0, 6 ) ) {
|
|
$ticket = substr( $ticket, 6 );
|
|
if ( $ticket && is_numeric( $ticket ) )
|
|
$this->knownPluginBug( $ticket );
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Skips the current test if there is an open WordPress ticket with id $ticket_id
|
|
*/
|
|
function knownWPBug( $ticket_id ) {
|
|
if ( WP_TESTS_FORCE_KNOWN_BUGS || in_array( $ticket_id, self::$forced_tickets ) )
|
|
return;
|
|
if ( ! TracTickets::isTracTicketClosed( 'https://core.trac.wordpress.org', $ticket_id ) )
|
|
$this->markTestSkipped( sprintf( 'WordPress Ticket #%d is not fixed', $ticket_id ) );
|
|
}
|
|
|
|
/**
|
|
* Skips the current test if there is an open unit tests ticket with id $ticket_id
|
|
*/
|
|
function knownUTBug( $ticket_id ) {
|
|
if ( WP_TESTS_FORCE_KNOWN_BUGS || in_array( 'UT' . $ticket_id, self::$forced_tickets ) )
|
|
return;
|
|
if ( ! TracTickets::isTracTicketClosed( 'https://unit-tests.trac.wordpress.org', $ticket_id ) )
|
|
$this->markTestSkipped( sprintf( 'Unit Tests Ticket #%d is not fixed', $ticket_id ) );
|
|
}
|
|
|
|
/**
|
|
* Skips the current test if there is an open plugin ticket with id $ticket_id
|
|
*/
|
|
function knownPluginBug( $ticket_id ) {
|
|
if ( WP_TESTS_FORCE_KNOWN_BUGS || in_array( 'Plugin' . $ticket_id, self::$forced_tickets ) )
|
|
return;
|
|
if ( ! TracTickets::isTracTicketClosed( 'https://plugins.trac.wordpress.org', $ticket_id ) )
|
|
$this->markTestSkipped( sprintf( 'WordPress Plugin Ticket #%d is not fixed', $ticket_id ) );
|
|
}
|
|
|
|
public static function forceTicket( $ticket ) {
|
|
self::$forced_tickets[] = $ticket;
|
|
}
|
|
|
|
/**
|
|
* Define constants after including files.
|
|
*/
|
|
function prepareTemplate( Text_Template $template ) {
|
|
$template->setVar( array( 'constants' => '' ) );
|
|
$template->setVar( array( 'wp_constants' => PHPUnit_Util_GlobalState::getConstantsAsString() ) );
|
|
parent::prepareTemplate( $template );
|
|
}
|
|
|
|
/**
|
|
* Returns the name of a temporary file
|
|
*/
|
|
function temp_filename() {
|
|
$tmp_dir = '';
|
|
$dirs = array( 'TMP', 'TMPDIR', 'TEMP' );
|
|
foreach( $dirs as $dir )
|
|
if ( isset( $_ENV[$dir] ) && !empty( $_ENV[$dir] ) ) {
|
|
$tmp_dir = $dir;
|
|
break;
|
|
}
|
|
if ( empty( $tmp_dir ) ) {
|
|
$tmp_dir = '/tmp';
|
|
}
|
|
$tmp_dir = realpath( $tmp_dir );
|
|
return tempnam( $tmp_dir, 'wpunit' );
|
|
}
|
|
|
|
/**
|
|
* Check each of the WP_Query is_* functions/properties against expected boolean value.
|
|
*
|
|
* Any properties that are listed by name as parameters will be expected to be true; any others are
|
|
* expected to be false. For example, assertQueryTrue('is_single', 'is_feed') means is_single()
|
|
* and is_feed() must be true and everything else must be false to pass.
|
|
*
|
|
* @param string $prop,... Any number of WP_Query properties that are expected to be true for the current request.
|
|
*/
|
|
function assertQueryTrue(/* ... */) {
|
|
global $wp_query;
|
|
$all = array(
|
|
'is_single', 'is_preview', 'is_page', 'is_archive', 'is_date', 'is_year', 'is_month', 'is_day', 'is_time',
|
|
'is_author', 'is_category', 'is_tag', 'is_tax', 'is_search', 'is_feed', 'is_comment_feed', 'is_trackback',
|
|
'is_home', 'is_404', 'is_comments_popup', 'is_paged', 'is_admin', 'is_attachment', 'is_singular', 'is_robots',
|
|
'is_posts_page', 'is_post_type_archive',
|
|
);
|
|
$true = func_get_args();
|
|
|
|
$passed = true;
|
|
$not_false = $not_true = array(); // properties that were not set to expected values
|
|
|
|
foreach ( $all as $query_thing ) {
|
|
$result = is_callable( $query_thing ) ? call_user_func( $query_thing ) : $wp_query->$query_thing;
|
|
|
|
if ( in_array( $query_thing, $true ) ) {
|
|
if ( ! $result ) {
|
|
array_push( $not_true, $query_thing );
|
|
$passed = false;
|
|
}
|
|
} else if ( $result ) {
|
|
array_push( $not_false, $query_thing );
|
|
$passed = false;
|
|
}
|
|
}
|
|
|
|
$message = '';
|
|
if ( count($not_true) )
|
|
$message .= implode( $not_true, ', ' ) . ' should be true. ';
|
|
if ( count($not_false) )
|
|
$message .= implode( $not_false, ', ' ) . ' should be false.';
|
|
$this->assertTrue( $passed, $message );
|
|
}
|
|
|
|
function unlink( $file ) {
|
|
$exists = is_file( $file );
|
|
if ( $exists && ! in_array( $file, self::$ignore_files ) ) {
|
|
//error_log( $file );
|
|
unlink( $file );
|
|
} elseif ( ! $exists ) {
|
|
$this->fail( "Trying to delete a file that doesn't exist: $file" );
|
|
}
|
|
}
|
|
|
|
function rmdir( $path ) {
|
|
$files = $this->files_in_dir( $path );
|
|
foreach ( $files as $file ) {
|
|
if ( ! in_array( $file, self::$ignore_files ) ) {
|
|
$this->unlink( $file );
|
|
}
|
|
}
|
|
}
|
|
|
|
function remove_added_uploads() {
|
|
// Remove all uploads.
|
|
$uploads = wp_upload_dir();
|
|
$this->rmdir( $uploads['basedir'] );
|
|
}
|
|
|
|
function files_in_dir( $dir ) {
|
|
$files = array();
|
|
|
|
$iterator = new RecursiveDirectoryIterator( $dir );
|
|
$objects = new RecursiveIteratorIterator( $iterator );
|
|
foreach ( $objects as $name => $object ) {
|
|
if ( is_file( $name ) ) {
|
|
$files[] = $name;
|
|
}
|
|
}
|
|
|
|
return $files;
|
|
}
|
|
|
|
function scan_user_uploads() {
|
|
static $files = array();
|
|
if ( ! empty( $files ) ) {
|
|
return $files;
|
|
}
|
|
|
|
$uploads = wp_upload_dir();
|
|
$files = $this->files_in_dir( $uploads['basedir'] );
|
|
return $files;
|
|
}
|
|
|
|
function delete_folders( $path ) {
|
|
$this->matched_dirs = array();
|
|
if ( ! is_dir( $path ) ) {
|
|
return;
|
|
}
|
|
|
|
$this->scandir( $path );
|
|
foreach ( array_reverse( $this->matched_dirs ) as $dir ) {
|
|
rmdir( $dir );
|
|
}
|
|
rmdir( $path );
|
|
}
|
|
|
|
function scandir( $dir ) {
|
|
foreach ( scandir( $dir ) as $path ) {
|
|
if ( 0 !== strpos( $path, '.' ) && is_dir( $dir . '/' . $path ) ) {
|
|
$this->matched_dirs[] = $dir . '/' . $path;
|
|
$this->scandir( $dir . '/' . $path );
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Helper to Convert a microtime string into a float
|
|
*/
|
|
protected function _microtime_to_float($microtime ){
|
|
$time_array = explode( ' ', $microtime );
|
|
return array_sum( $time_array );
|
|
}
|
|
|
|
/**
|
|
* When `wp_mail()` is called, make sure `$_SERVER['SERVER_NAME']` is faked.
|
|
*
|
|
* @since 4.3.0
|
|
*
|
|
* @param array $args `wp_mail()` arguments.
|
|
* @return array $args
|
|
*/
|
|
public function set_wp_mail_globals( $args ) {
|
|
if ( ! isset( $_SERVER['SERVER_NAME'] ) ) {
|
|
$_SERVER['SERVER_NAME'] = 'example.com';
|
|
add_action( 'phpmailer_init', array( $this, 'tear_down_wp_mail_globals' ) );
|
|
}
|
|
|
|
return $args;
|
|
}
|
|
|
|
/**
|
|
* Tear down the faked `$_SERVER['SERVER_NAME']` global used in `wp_mail()`.
|
|
*
|
|
* @since 4.3.0
|
|
*/
|
|
public function tear_down_wp_mail_globals() {
|
|
unset( $_SERVER['SERVER_NAME'] );
|
|
}
|
|
|
|
/**
|
|
* Multisite-agnostic way to delete a user from the database.
|
|
*
|
|
* @since 4.3.0
|
|
*/
|
|
public static function delete_user( $user_id ) {
|
|
if ( is_multisite() ) {
|
|
return wpmu_delete_user( $user_id );
|
|
} else {
|
|
return wp_delete_user( $user_id );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Utility method that resets permalinks and flushes rewrites.
|
|
*
|
|
* @since 4.4.0
|
|
*
|
|
* @global WP_Rewrite $wp_rewrite
|
|
*
|
|
* @param string $structure Optional. Permalink structure to set. Default empty.
|
|
*/
|
|
public function set_permalink_structure( $structure = '' ) {
|
|
global $wp_rewrite;
|
|
|
|
$wp_rewrite->init();
|
|
$wp_rewrite->set_permalink_structure( $structure );
|
|
$wp_rewrite->flush_rules();
|
|
}
|
|
}
|