mirror of
https://github.com/gosticks/wordpress-develop.git
synced 2025-10-16 12:05:38 +00:00
c6c78490e2
All PHP files in `/tests` now conform to the PHP coding standards, or have exceptions appropriately marked. Travis now also runs `phpcs` on the `/tests` directory, any future changes to these files must conform entirely to the WordPress PHP coding standards. 🎉 See #47632. git-svn-id: https://develop.svn.wordpress.org/trunk@45607 602fd350-edb4-49c9-b593-d223f7449a82
105 lines
3.0 KiB
PHP
105 lines
3.0 KiB
PHP
<?php
|
|
|
|
if ( is_multisite() ) :
|
|
|
|
/**
|
|
* @group multisite
|
|
* @covers ::get_space_used
|
|
*/
|
|
class Tests_Multisite_Get_Space_Used extends WP_UnitTestCase {
|
|
protected $suppress = false;
|
|
|
|
function setUp() {
|
|
global $wpdb;
|
|
parent::setUp();
|
|
$this->suppress = $wpdb->suppress_errors();
|
|
}
|
|
|
|
function tearDown() {
|
|
global $wpdb;
|
|
$wpdb->suppress_errors( $this->suppress );
|
|
parent::tearDown();
|
|
}
|
|
|
|
function test_get_space_used_switched_site() {
|
|
$blog_id = self::factory()->blog->create();
|
|
switch_to_blog( $blog_id );
|
|
|
|
// Our comparison of space relies on an initial value of 0. If a previous test has failed or if the
|
|
// src directory already contains a content directory with site content, then the initial expectation
|
|
// will be polluted. We create sites until an empty one is available.
|
|
while ( 0 !== get_space_used() ) {
|
|
restore_current_blog();
|
|
$blog_id = self::factory()->blog->create();
|
|
switch_to_blog( $blog_id );
|
|
}
|
|
|
|
// Upload a file to the new site.
|
|
$filename = __FUNCTION__ . '.jpg';
|
|
$contents = __FUNCTION__ . '_contents';
|
|
$file = wp_upload_bits( $filename, null, $contents );
|
|
|
|
// get_space_used() is measures in MB, get the size of the new file in MB.
|
|
$size = filesize( $file['file'] ) / 1024 / 1024;
|
|
|
|
delete_transient( 'dirsize_cache' );
|
|
|
|
$this->assertEquals( $size, get_space_used() );
|
|
$upload_dir = wp_upload_dir();
|
|
$this->remove_added_uploads();
|
|
$this->delete_folders( $upload_dir['basedir'] );
|
|
restore_current_blog();
|
|
}
|
|
|
|
/**
|
|
* Directories of sub sites on a network should not count against the same spaced used total for
|
|
* the main site.
|
|
*/
|
|
function test_get_space_used_main_site() {
|
|
$space_used = get_space_used();
|
|
|
|
$blog_id = self::factory()->blog->create();
|
|
switch_to_blog( $blog_id );
|
|
|
|
// We don't rely on an initial value of 0 for space used, but should have a clean space available
|
|
// so that we can remove any uploaded files and directories without concern of a conflict with
|
|
// existing content directories in src.
|
|
while ( 0 !== get_space_used() ) {
|
|
restore_current_blog();
|
|
$blog_id = self::factory()->blog->create();
|
|
switch_to_blog( $blog_id );
|
|
}
|
|
|
|
// Upload a file to the new site.
|
|
$filename = __FUNCTION__ . '.jpg';
|
|
$contents = __FUNCTION__ . '_contents';
|
|
wp_upload_bits( $filename, null, $contents );
|
|
|
|
restore_current_blog();
|
|
|
|
delete_transient( 'dirsize_cache' );
|
|
|
|
$this->assertEquals( $space_used, get_space_used() );
|
|
|
|
// Switch back to the new site to remove the uploaded file.
|
|
switch_to_blog( $blog_id );
|
|
$upload_dir = wp_upload_dir();
|
|
$this->remove_added_uploads();
|
|
$this->delete_folders( $upload_dir['basedir'] );
|
|
restore_current_blog();
|
|
}
|
|
|
|
function test_get_space_used_pre_get_spaced_used_filter() {
|
|
add_filter( 'pre_get_space_used', array( $this, '_filter_space_used' ) );
|
|
|
|
$this->assertEquals( 300, get_space_used() );
|
|
|
|
remove_filter( 'pre_get_space_used', array( $this, '_filter_space_used' ) );
|
|
}
|
|
|
|
function _filter_space_used() {
|
|
return 300;
|
|
}
|
|
}
|
|
endif;
|