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

Timestamp:

03/06/2015 08:37:08 PM (10 years ago)

Author:

wonderboymusic

Message:

Add unit tests for wp_attachment_is(), checks the whitelist and arbitrary extension.

See #25275.

File:

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

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

@@ -440 +440 @@
     }
 
+    public function test_wp_attachment_is() {
+        $filename = DIR_TESTDATA . '/images/test-image.jpg';
+        $contents = file_get_contents( $filename );
+
+        $upload = wp_upload_bits( basename( $filename ), null, $contents );
+        $attachment_id = $this->_make_attachment( $upload );
+
+        $this->assertTrue( wp_attachment_is_image( $attachment_id ) );
+        $this->assertTrue( wp_attachment_is( 'image', $attachment_id ) );
+        $this->assertFalse( wp_attachment_is( 'audio', $attachment_id ) );
+        $this->assertFalse( wp_attachment_is( 'video', $attachment_id ) );
+    }
+
+    public function test_wp_attachment_is_default() {
+        $filename = DIR_TESTDATA . '/images/test-image.psd';
+        $contents = file_get_contents( $filename );
+
+        $upload = wp_upload_bits( basename( $filename ), null, $contents );
+        $attachment_id = $this->_make_attachment( $upload );
+
+        $this->assertFalse( wp_attachment_is_image( $attachment_id ) );
+        $this->assertTrue( wp_attachment_is( 'psd', $attachment_id ) );
+        $this->assertFalse( wp_attachment_is( 'audio', $attachment_id ) );
+        $this->assertFalse( wp_attachment_is( 'video', $attachment_id ) );
+    }
 }