mirror of
https://github.com/buddypress/buddypress.git
synced 2026-07-23 21:06:56 +08:00
f15348c102
We are introducing a new `BP_LoggedIn_User` class to fetch data about a BuddyPress logged-in user. This new addition fixes an issue where a user could be de-authenticated when making REST API requests. Props dcavins, DJPaul, johnjamesjacoby, and imath. Closes https://github.com/buddypress/buddypress/pull/395 See #9229 and #9145 Fixes #7658 git-svn-id: https://buddypress.svn.wordpress.org/trunk@14070 cdf35c40-ae34-48e0-9cc9-0c9da1808c22
541 lines
14 KiB
PHP
541 lines
14 KiB
PHP
<?php
|
|
|
|
require_once BP_TESTS_DIR . 'assets/invitations-extensions.php';
|
|
|
|
/**
|
|
* @group core
|
|
* @group invitations
|
|
*/
|
|
class BP_Tests_Invitations extends BP_UnitTestCase {
|
|
|
|
/**
|
|
* @ticket BP8552
|
|
* @group cache
|
|
*/
|
|
public function test_invitation_query_with_ids_cache_results() {
|
|
global $wpdb;
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create a couple of invitations.
|
|
$invite_args = array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 'sent',
|
|
);
|
|
|
|
$invites_class->add_invitation( $invite_args );
|
|
|
|
$invite_args['inviter_id'] = $u2;
|
|
|
|
$invites_class->add_invitation( $invite_args );
|
|
|
|
$wpdb->num_queries = 0;
|
|
|
|
$first_query = BP_Invitation::get(
|
|
array(
|
|
'cache_results' => true,
|
|
'fields' => 'ids',
|
|
)
|
|
);
|
|
|
|
$queries_before = get_num_queries();
|
|
|
|
$second_query = BP_Invitation::get(
|
|
array(
|
|
'cache_results' => false,
|
|
'fields' => 'ids',
|
|
)
|
|
);
|
|
|
|
$queries_after = get_num_queries();
|
|
|
|
$this->assertNotSame( $queries_before, $queries_after, 'Assert that queries are run' );
|
|
$this->assertSame( 2, $queries_after, 'Assert that the uncached query was run' );
|
|
$this->assertSameSets( $first_query, $second_query, 'Results of the query are expected to match.' );
|
|
}
|
|
|
|
/**
|
|
* @ticket BP8552
|
|
* @group cache
|
|
*/
|
|
public function test_invitation_query_with_all_cache_results() {
|
|
global $wpdb;
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create a couple of invitations.
|
|
$invite_args = array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 'sent',
|
|
);
|
|
|
|
$invites_class->add_invitation( $invite_args );
|
|
|
|
$invite_args['inviter_id'] = $u2;
|
|
|
|
$invites_class->add_invitation( $invite_args );
|
|
|
|
$wpdb->num_queries = 0;
|
|
|
|
$first_query = BP_Invitation::get(
|
|
array( 'cache_results' => true )
|
|
);
|
|
|
|
$queries_before = get_num_queries();
|
|
|
|
$second_query = BP_Invitation::get(
|
|
array( 'cache_results' => false )
|
|
);
|
|
|
|
$queries_after = get_num_queries();
|
|
|
|
$this->assertNotSame( $queries_before, $queries_after, 'Assert that queries are run' );
|
|
$this->assertSame( 3, $queries_after, 'Assert that the uncached query was run' );
|
|
$this->assertEquals( $first_query, $second_query, 'Results of the query are expected to match.' );
|
|
}
|
|
|
|
public function test_bp_invitations_add_invitation_vanilla() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create a couple of invitations.
|
|
$invite_args = array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 'sent',
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
$invite_args['inviter_id'] = $u2;
|
|
$i2 = $invites_class->add_invitation( $invite_args );
|
|
|
|
$get_invites = array(
|
|
'user_id' => $u3,
|
|
'fields' => 'ids',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEqualSets( array( $i1, $i2 ), $invites );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_invitation_avoid_duplicates() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$invite_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 'sent',
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
// Attempt to create a duplicate. Should return existing invite.
|
|
$i2 = $invites_class->add_invitation( $invite_args );
|
|
$this->assertEquals( $i1, $i2 );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_invitation_invite_plus_request_should_accept() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$i1 = $invites_class->add_invitation(
|
|
array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 'sent',
|
|
)
|
|
);
|
|
|
|
$this->assertIsInt( $i1, 'Invitation ID is not an integer.' );
|
|
|
|
// Create a request.
|
|
$invites_class->add_request(
|
|
array(
|
|
'user_id' => $u3,
|
|
'item_id' => 1,
|
|
)
|
|
);
|
|
|
|
$get_invites = array(
|
|
'user_id' => $u3,
|
|
'accepted' => 'accepted',
|
|
);
|
|
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$invite_ids = wp_list_pluck( $invites, 'id' );
|
|
|
|
$this->assertNotEmpty( $invite_ids );
|
|
$this->assertEqualSets( array( $i1 ), $invite_ids );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_invitation_unsent_invite_plus_request_should_not_accept() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$invite_args = array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 0,
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
|
|
// Create a request.
|
|
$request_args = array(
|
|
'user_id' => $u3,
|
|
'item_id' => 1,
|
|
);
|
|
$r1 = $invites_class->add_request( $request_args );
|
|
|
|
$get_invites = array(
|
|
'user_id' => $u3,
|
|
'accepted' => 'accepted',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEqualSets( array(), wp_list_pluck( $invites, 'id' ) );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_invitation_unsent_invite_plus_request_then_send_invite_should_accept() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$invite_args = array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 0,
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
|
|
// Create a request.
|
|
$request_args = array(
|
|
'user_id' => $u3,
|
|
'item_id' => 1,
|
|
);
|
|
$r1 = $invites_class->add_request( $request_args );
|
|
|
|
$invites_class->send_invitation_by_id( $i1 );
|
|
|
|
// Check that both the request and invitation are marked 'accepted'.
|
|
$get_invites = array(
|
|
'user_id' => $u3,
|
|
'type' => 'all',
|
|
'accepted' => 'accepted',
|
|
'fields' => 'ids',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEqualSets( array( $i1, $r1 ), $invites );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_request_vanilla() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create a couple of requests.
|
|
$request_args = array(
|
|
'user_id' => $u1,
|
|
'item_id' => 7,
|
|
);
|
|
$r1 = $invites_class->add_request( $request_args );
|
|
$request_args['item_id'] = 4;
|
|
$r2 = $invites_class->add_request( $request_args );
|
|
|
|
$get_requests = array(
|
|
'user_id' => $u1,
|
|
'fields' => 'ids',
|
|
);
|
|
$requests = $invites_class->get_requests( $get_requests );
|
|
$this->assertEqualSets( array( $r1, $r2 ), $requests );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_request_avoid_duplicates() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
// Create a couple of requests.
|
|
$request_args = array(
|
|
'user_id' => $u1,
|
|
'item_id' => 7,
|
|
);
|
|
$r1 = $invites_class->add_request( $request_args );
|
|
// Attempt to create a duplicate.
|
|
$this->assertFalse( $invites_class->add_request( $request_args ) );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_request_request_plus_sent_invite_should_accept() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create a request.
|
|
$request_args = array(
|
|
'user_id' => $u2,
|
|
'item_id' => 1,
|
|
);
|
|
$r1 = $invites_class->add_request( $request_args );
|
|
|
|
// Create an invitation.
|
|
$invite_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 1,
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
|
|
// Check that both the request and invitation are marked 'accepted'.
|
|
$get_invites = array(
|
|
'user_id' => $u2,
|
|
'type' => 'all',
|
|
'accepted' => 'accepted',
|
|
'fields' => 'ids',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEqualSets( array( $r1, $i1 ), $invites );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_sending_should_clear_cache() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$invite_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
|
|
$invite = new BP_Invitation( $i1 );
|
|
$this->assertEquals( 0, $invite->invite_sent );
|
|
|
|
$invites_class->send_invitation_by_id( $i1 );
|
|
|
|
$invite = new BP_Invitation( $i1 );
|
|
$this->assertEquals( 1, $invite->invite_sent );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_get_by_search_terms() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$i1_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'content' => 'Sometimes, the mystery is enough.',
|
|
);
|
|
$i1 = $invites_class->add_invitation( $i1_args );
|
|
$invites_class->send_invitation_by_id( $i1 );
|
|
|
|
// Create an invitation that uses an email address.
|
|
$i2_args = array(
|
|
'invitee_email' => 'findme@buddypress.org',
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
);
|
|
$i2 = $invites_class->add_invitation( $i2_args );
|
|
$invites_class->send_invitation_by_id( $i2 );
|
|
|
|
$get_invites = array(
|
|
'search_terms' => 'mystery',
|
|
'fields' => 'ids',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEqualSets( array( $i1 ), $invites );
|
|
|
|
$get_invites = array(
|
|
'search_terms' => 'findme',
|
|
'fields' => 'ids',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEqualSets( array( $i2 ), $invites );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_request_with_date_modified() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
$time = gmdate( 'Y-m-d H:i:s', time() - 100 );
|
|
$args = array(
|
|
'user_id' => $u1,
|
|
'item_id' => 7,
|
|
'date_modified' => $time,
|
|
);
|
|
$r1 = $invites_class->add_request( $args );
|
|
|
|
$req = new BP_Invitation( $r1 );
|
|
$this->assertEquals( $time, $req->date_modified );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_add_invite_with_date_modified() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
$time = gmdate( 'Y-m-d H:i:s', time() - 100 );
|
|
|
|
// Create an invitation.
|
|
$invite_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 1,
|
|
'send_invite' => 1,
|
|
'date_modified' => $time,
|
|
);
|
|
$i1 = $invites_class->add_invitation( $invite_args );
|
|
|
|
$inv = new BP_Invitation( $i1 );
|
|
$this->assertEquals( $time, $inv->date_modified );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
|
|
public function test_bp_invitations_orderby_item_id() {
|
|
$old_current_user = get_current_user_id();
|
|
|
|
$u1 = self::factory()->user->create();
|
|
$u2 = self::factory()->user->create();
|
|
$u3 = self::factory()->user->create();
|
|
wp_set_current_user( $u1 );
|
|
|
|
$invites_class = new BPTest_Invitation_Manager_Extension();
|
|
|
|
// Create an invitation.
|
|
$i1_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 6,
|
|
);
|
|
$i1 = $invites_class->add_invitation( $i1_args );
|
|
$invites_class->send_invitation_by_id( $i1 );
|
|
|
|
$i2_args = array(
|
|
'user_id' => $u3,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 4,
|
|
);
|
|
$i2 = $invites_class->add_invitation( $i2_args );
|
|
$invites_class->send_invitation_by_id( $i2 );
|
|
|
|
$i3_args = array(
|
|
'user_id' => $u2,
|
|
'inviter_id' => $u1,
|
|
'item_id' => 8,
|
|
);
|
|
$i3 = $invites_class->add_invitation( $i3_args );
|
|
$invites_class->send_invitation_by_id( $i3 );
|
|
|
|
$get_invites = array(
|
|
'order_by' => 'item_id',
|
|
'sort_order' => 'ASC',
|
|
'fields' => 'ids',
|
|
);
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEquals( array( $i2, $i1, $i3 ), $invites );
|
|
|
|
$get_invites['sort_order'] = 'DESC';
|
|
$invites = $invites_class->get_invitations( $get_invites );
|
|
$this->assertEquals( array( $i3, $i1, $i2 ), $invites );
|
|
|
|
wp_set_current_user( $old_current_user );
|
|
}
|
|
}
|
