Changeset 32919 for trunk/tests/phpunit/tests/post/attachments.php

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;
+    }
 }