<?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;