Adjust capability checks in AJAX callback tests

Author: JLG-WOCFR-DEV

tests/BlcAjaxCallbacksTest.php

@@ -1250,14 +1250,24 @@ public function test_edit_link_returns_success_when_post_has_been_deleted(): voi
 
         Functions\when('check_ajax_referer')->justReturn(true);
         Functions\expect('get_post')->once()->with(11)->andReturn(null);
-        Functions\expect('blc_current_user_can_fix_links')->once()->andReturn(true);
+        $capabilityCalls = 0;
+        Functions\when('blc_current_user_can_fix_links')->alias(static function () use (&$capabilityCalls) {
+            $capabilityCalls++;
+
+            return true;
+        });
 
         global $wpdb;
         $wpdb = $this->createAjaxWpdbStub();
         $wpdb->prefix = 'wp_';
         $wpdb->delete_return_value = 1;
 
-        Functions\expect('wp_send_json_success')->once()->with(['purged' => true])->andReturnUsing(function () {
+        Functions\expect('wp_send_json_success')->once()->with([
+            'purged'      => true,
+            'row_removed' => true,
+            'rowRemoved'  => true,
+            'previewOnly' => false,
+        ])->andReturnUsing(function () {
             throw new \Exception('success');
         });
 
@@ -1268,6 +1278,8 @@ public function test_edit_link_returns_success_when_post_has_been_deleted(): voi
             $this->assertSame('success', $exception->getMessage());
         }
 
+        $this->assertGreaterThanOrEqual(2, $capabilityCalls, 'The capability should be checked for the request and during cleanup.');
+
         $this->assertCount(1, $wpdb->delete_calls, 'Cleanup should remove the orphaned database row.');
         $deleted = $wpdb->delete_calls[0];
         $this->assertSame('wp_blc_broken_links', $deleted[0]);
@@ -2063,7 +2075,12 @@ public function test_unlink_returns_success_when_post_has_been_deleted(): void
 
         Functions\when('check_ajax_referer')->justReturn(true);
         Functions\expect('get_post')->once()->with(21)->andReturn(null);
-        Functions\expect('blc_current_user_can_fix_links')->once()->andReturn(true);
+        $capabilityCalls = 0;
+        Functions\when('blc_current_user_can_fix_links')->alias(static function () use (&$capabilityCalls) {
+            $capabilityCalls++;
+
+            return true;
+        });
 
         Functions\when('esc_url_raw')->alias(function ($url) {
             return $url;
@@ -2085,6 +2102,8 @@ public function test_unlink_returns_success_when_post_has_been_deleted(): void
             $this->assertSame('success', $exception->getMessage());
         }
 
+        $this->assertGreaterThanOrEqual(2, $capabilityCalls, 'The capability should be checked for the request and during cleanup.');
+
         $this->assertCount(1, $wpdb->delete_calls, 'Cleanup should remove the orphaned database row.');
         $deleted = $wpdb->delete_calls[0];
         $this->assertSame('wp_blc_broken_links', $deleted[0]);