summaryrefslogtreecommitdiffstatshomepage
path: root/tests/phpunit/includes/factory/class-wp-unittest-factory-for-bookmark.php
blob: d9f8e06fbe353dcfce745c4c77f93b5d2e86ab67 (plain) (blame)
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
69
70
71
72
73
74
<?php

/**
 * Factory for creating fixtures for the deprecated Links/Bookmarks API.
 *
 * 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.
 *
 * @since 4.6.0
 *
 * @method int|WP_Error     create( $args = array(), $generation_definitions = null )
 * @method object|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_Bookmark extends WP_UnitTest_Factory_For_Thing {

	public function __construct( $factory = null ) {
		parent::__construct( $factory );
		$this->default_generation_definitions = array(
			'link_name' => new WP_UnitTest_Generator_Sequence( 'Bookmark name %s' ),
			'link_url'  => new WP_UnitTest_Generator_Sequence( 'Bookmark URL %s' ),
		);
	}

	/**
	 * Creates a link object.
	 *
	 * @since 4.6.0
	 * @since 6.2.0 Returns a WP_Error object on failure.
	 *
	 * @param array $args Arguments for the link object.
	 *
	 * @return int|WP_Error The link ID on success, WP_Error object on failure.
	 */
	public function create_object( $args ) {
		return wp_insert_link( $args, true );
	}

	/**
	 * Updates a link object.
	 *
	 * @since 4.6.0
	 * @since 6.2.0 Returns a WP_Error object on failure.
	 *
	 * @param int   $link_id ID of the link to update.
	 * @param array $fields  The fields to update.
	 *
	 * @return int|WP_Error The link ID on success, WP_Error object on failure.
	 */
	public function update_object( $link_id, $fields ) {
		$fields['link_id'] = $link_id;

		$result = wp_update_link( $fields );

		if ( 0 === $result ) {
			return new WP_Error( 'link_update_error', __( 'Could not update link.' ) );
		}

		return $result;
	}

	/**
	 * Retrieves a link by a given ID.
	 *
	 * @since 4.6.0
	 *
	 * @param int $link_id ID of the link to retrieve.
	 *
	 * @return object|null The link object on success, null on failure.
	 */
	public function get_object_by_id( $link_id ) {
		return get_bookmark( $link_id );
	}
}