1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
<?php
/**
* Unit test factory for posts.
*
* Note: The below @method notations are defined solely for the benefit of IDEs,
* as a way to indicate expected return values from the given factory methods.
*
* @method int|WP_Error create( $args = array(), $generation_definitions = null )
* @method WP_Post|WP_Error create_and_get( $args = array(), $generation_definitions = null )
* @method (int|WP_Error)[] create_many( $count, $args = array(), $generation_definitions = null )
*/
class WP_UnitTest_Factory_For_Post extends WP_UnitTest_Factory_For_Thing {
public function __construct( $factory = null ) {
parent::__construct( $factory );
$this->default_generation_definitions = array(
'post_status' => 'publish',
'post_title' => new WP_UnitTest_Generator_Sequence( 'Post title %s' ),
'post_content' => new WP_UnitTest_Generator_Sequence( 'Post content %s' ),
'post_excerpt' => new WP_UnitTest_Generator_Sequence( 'Post excerpt %s' ),
'post_type' => 'post',
);
}
/**
* Creates a post object.
*
* @since UT (3.7.0)
* @since 6.2.0 Returns a WP_Error object on failure.
*
* @param array $args Array with elements for the post.
*
* @return int|WP_Error The post ID on success, WP_Error object on failure.
*/
public function create_object( $args ) {
return wp_insert_post( $args, true );
}
/**
* Updates an existing post object.
*
* @since UT (3.7.0)
* @since 6.2.0 Returns a WP_Error object on failure.
*
* @param int $post_id ID of the post to update.
* @param array $fields Post data.
*
* @return int|WP_Error The post ID on success, WP_Error object on failure.
*/
public function update_object( $post_id, $fields ) {
$fields['ID'] = $post_id;
return wp_update_post( $fields, true );
}
/**
* Retrieves a post by a given ID.
*
* @since UT (3.7.0)
*
* @param int $post_id ID of the post to retrieve.
*
* @return WP_Post|null WP_Post object on success, null on failure.
*/
public function get_object_by_id( $post_id ) {
return get_post( $post_id );
}
}
|
