Wordpress/tests/phpunit/includes/factory/class-wp-unittest-factory-for-attachment.php
Gary Pendergast 8f95800d52 Code is Poetry.
WordPress' code just... wasn't.
This is now dealt with.

Props jrf, pento, netweb, GaryJ, jdgrimes, westonruter, Greg Sherwood from PHPCS, and everyone who's ever contributed to WPCS and PHPCS.
Fixes #41057.



git-svn-id: https://develop.svn.wordpress.org/trunk@42343 602fd350-edb4-49c9-b593-d223f7449a82
2017-11-30 23:09:33 +00:00

66 lines
1.8 KiB
PHP

<?php
class WP_UnitTest_Factory_For_Attachment extends WP_UnitTest_Factory_For_Post {
/**
* Create an attachment fixture.
*
* @param array $args {
* Array of arguments. Accepts all arguments that can be passed to
* wp_insert_attachment(), in addition to the following:
* @type int $post_parent ID of the post to which the attachment belongs.
* @type string $file Path of the attached file.
* }
* @param int $legacy_parent Deprecated.
* @param array $legacy_args Deprecated
*/
function create_object( $args, $legacy_parent = 0, $legacy_args = array() ) {
// Backward compatibility for legacy argument format.
if ( is_string( $args ) ) {
$file = $args;
$args = $legacy_args;
$args['post_parent'] = $legacy_parent;
$args['file'] = $file;
}
$r = array_merge(
array(
'file' => '',
'post_parent' => 0,
), $args
);
return wp_insert_attachment( $r, $r['file'], $r['post_parent'] );
}
function create_upload_object( $file, $parent = 0 ) {
$contents = file_get_contents( $file );
$upload = wp_upload_bits( basename( $file ), null, $contents );
$type = '';
if ( ! empty( $upload['type'] ) ) {
$type = $upload['type'];
} else {
$mime = wp_check_filetype( $upload['file'] );
if ( $mime ) {
$type = $mime['type'];
}
}
$attachment = array(
'post_title' => basename( $upload['file'] ),
'post_content' => '',
'post_type' => 'attachment',
'post_parent' => $parent,
'post_mime_type' => $type,
'guid' => $upload['url'],
);
// Save the data
$id = wp_insert_attachment( $attachment, $upload['file'], $parent );
wp_update_attachment_metadata( $id, wp_generate_attachment_metadata( $id, $upload['file'] ) );
return $id;
}
}