Changeset 32919

Timestamp:

06/24/2015 12:42:33 AM (11 years ago)

Author:

johnbillion

Message:

Implement tests for the upload_mimes filter to ensure it prevents the upload of disallowed file types.

See #32693

File:

• trunk/tests/phpunit/tests/post/attachments.php (modified) (1 diff)

trunk/tests/phpunit/tests/post/attachments.php

@@ -501 +501 @@
         }
 
+        public function test_upload_mimes_filter_is_applied() {
+                $filename = DIR_TESTDATA . '/images/test-image.jpg';
+                $contents = file_get_contents( $filename );
+
+                $upload = wp_upload_bits( basename( $filename ), null, $contents );
+
+                $this->assertFalse( $upload['error'] );
+
+                add_filter( 'upload_mimes', array( $this, 'blacklist_jpg_mime_type' ) );
+
+                $upload = wp_upload_bits( basename( $filename ), null, $contents );
+
+                $this->assertNotEmpty( $upload['error'] );
+
+                remove_filter( 'upload_mimes', array( $this, 'blacklist_jpg_mime_type' ) );
+        }
+
         public function whitelist_psd_mime_type( $mimes ) {
                 $mimes['psd'] = 'application/octet-stream';
                 return $mimes;
         }
+
+        public function blacklist_jpg_mime_type( $mimes ) {
+                unset( $mimes['jpg|jpeg|jpe'] );
+                return $mimes;
+        }
 }