Ticket #6681: 6681.3-tests.diff

tests/phpunit/testcases/notifications/functions.php

@@ -233 +233 @@
                 // assert
                 $this->assertEquals( 2, buddypress()->notifications->query_loop->total_notification_count );
         }
+
+        /**
+         * @group bp_notifications_delete_notifications_on_user_delete
+         */
+        public function test_bp_notifications_delete_notifications_on_user_delete_should_delete_all_notifications() {
+                $u = $this->factory->user->create();
+
+                // Create notifications
+                $n1 = $this->factory->notification->create( array(
+                        'component_name'    => 'messages',
+                        'component_action'  => 'new_message',
+                        'item_id'           => 99,
+                        'user_id'           => $u,
+                ) );
+
+                $n2 = $this->factory->notification->create( array(
+                        'component_name'    => 'activity',
+                        'component_action'  => 'new_at_mention',
+                        'item_id'           => 99,
+                        'user_id'           => $u,
+                ) );
+
+                $n3 = $this->factory->notification->create( array(
+                        'component_name' => 'groups',
+                        'user_id'        => $u,
+                ) );
+
+                $n4 = $this->factory->notification->create( array(
+                        'component_name'   => 'friends',
+                        'component_action' => 'friendship_request',
+                        'user_id'          => $u,
+                ) );
+
+                // Create notification for non-core component
+                $n5 = $this->factory->notification->create( array(
+                        'component_name'    => 'foo',
+                        'component_action'  => 'bar',
+                        'item_id'           => 99,
+                        'user_id'           => $u,
+                ) );
+
+                global $wpdb, $bp;
+
+                /**
+                 * Get use BP_Notifications_Notification::get(), because class::parse_args,
+                 * checks against bp_notifications_get_registered_components()
+                 * and if component is disabled it will be ignored.
+                 */
+                $query = $wpdb->prepare( "SELECT id FROM {$bp->notifications->table_name} WHERE user_id = %d and is_new = 1", $u );
+
+                // Make sure notifications are added
+                $found1 = $wpdb->get_col( $query );
+                $this->assertEqualSets( array( $n1, $n2, $n3, $n4, $n5 ), $found1 );
+
+                wp_delete_user( $u );
+
+                // Check, if notifications are deleted.
+                $found2 = $wpdb->get_col( $query );
+                $this->assertEmpty( $found2 );
+        }
 }