instance = new Favorites(); $this->user = $this->factory->user->create( array( 'role' => 'administrator', ) ); } /** * Test that favorites can be added. */ public function test_add_favorites() { wp_set_current_user( $this->user ); $result = $this->instance->add_item( 'menu-item', get_current_user_id() ); $this->assertTrue( $result ); $result = $this->instance->add_item( 'menu-item2', get_current_user_id() ); $this->assertTrue( $result ); $favorites = $this->instance->get_all( get_current_user_id() ); $this->assertContains( 'menu-item', $favorites ); $this->assertContains( 'menu-item2', $favorites ); } /** * Test that favorites can be removed. */ public function test_remove_favorites() { wp_set_current_user( $this->user ); $result = $this->instance->add_item( 'item-to-remove', get_current_user_id() ); $this->assertTrue( $result ); $favorites = $this->instance->get_all( get_current_user_id() ); $this->assertContains( 'item-to-remove', $favorites ); $result = $this->instance->remove_item( 'item-to-remove', get_current_user_id() ); $favorites = $this->instance->get_all( get_current_user_id() ); $this->assertNotContains( 'item-to-remove', $favorites ); } /** * Test that existing favorites can not be added again. */ public function test_add_previously_added_favorite() { wp_set_current_user( $this->user ); $result = $this->instance->add_item( 'duplicate-item', get_current_user_id() ); $this->assertTrue( $result ); $result = $this->instance->add_item( 'duplicate-item', get_current_user_id() ); $this->assertInstanceOf( 'WP_Error', $result ); } /** * Test removing a favorite that does not exist. */ public function test_remove_invalid_favorite() { wp_set_current_user( $this->user ); $result = $this->instance->remove_item( 'does-not-exist', get_current_user_id() ); $this->assertInstanceOf( 'WP_Error', $result ); } }