Ticket #30202: 30202.tests.diff

tests/phpunit/tests/multisite/site.php

@@ -1377 +1377 @@
         }
 
         /**
+         * Test that a site's upload space used is actually being counted
+         */
+        function test_get_space_used() {
+                $blog_id = $this->factory->blog->create();
+                switch_to_blog( $blog_id );
+
+                $this->assertEquals( 0, get_space_used() );
+
+                // Upload a file to the new site.
+                $filename = rand_str().'.jpg';
+                $contents = rand_str();
+                $file = wp_upload_bits( $filename, null, $contents );
+
+                // get_space_used() is measures in MB, get size of new file in MB
+                $size = filesize( $file['file'] ) / 1024 / 1024;
+
+                delete_transient( 'dirsize_cache' );
+
+                $this->assertEquals( $size, get_space_used() );
+
+                restore_current_blog();
+        }
+
+        /**
+         * Test that uploads for other sites are excluded from counting towards the space used by the main site
+         */
+        function test_get_space_used_exclude() {
+
+                $space_used = get_space_used();
+
+                $blog_id = $this->factory->blog->create();
+                switch_to_blog( $blog_id );
+
+                // Upload a file to the new site.
+                $filename = rand_str().'.jpg';
+                $contents = rand_str();
+                $file = wp_upload_bits( $filename, null, $contents );
+
+                restore_current_blog();
+
+                delete_transient( 'dirsize_cache' );
+
+                $this->assertEquals( $space_used, get_space_used() );
+        }
+
+        /**
          * Test the primary purpose of get_blog_post(), to retrieve a post from
          * another site on the network.
          */