Wordpress/tests/phpunit/includes/factory/class-wp-unittest-factory-f...

63 lines
1.7 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;
}
}