wordpress-develop/tests/phpunit/tests/multisite/getSpaceUsed.php
Tonya Mork 40ac5de838 Coding Standards: Add visibility to methods in tests/phpunit/tests/.
Adds a `public` visibility to test fixtures, tests, data providers, and callbacks methods.

Adds a `private` visibility to helper methods within test classes.

Renames callbacks and helpers that previously started with a `_` prefix. Why? For consistency and to leverage using the method visibility. Further naming standardizations is beyond the scope of this commit.

Props costdev, jrf, hellofromTonya.
Fixes #54177.

git-svn-id: https://develop.svn.wordpress.org/trunk@52010 602fd350-edb4-49c9-b593-d223f7449a82
2021-11-04 15:22:47 +00:00

92 lines
2.8 KiB
PHP

<?php
if ( is_multisite() ) :
/**
* @group multisite
* @covers ::get_space_used
*/
class Tests_Multisite_GetSpaceUsed extends WP_UnitTestCase {
public 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->assertSame( $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.
*/
public 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->assertSame( $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();
}
public function test_get_space_used_pre_get_spaced_used_filter() {
add_filter( 'pre_get_space_used', array( $this, 'filter_space_used' ) );
$this->assertSame( 300, get_space_used() );
remove_filter( 'pre_get_space_used', array( $this, 'filter_space_used' ) );
}
public function filter_space_used() {
return 300;
}
}
endif;