Ticket #43438: 43438.13.diff

tests/phpunit/includes/testcase-ajax.php

@@ -119 +119 @@
                 'delete-theme',
                 'install-theme',
                 'get-post-thumbnail-html',
+                'wp-privacy-export-personal-data',
+                'wp-privacy-erase-personal-data',
         );
 
         public static function setUpBeforeClass() {

new file tests/phpunit/tests/ajax/PrivacyErasePersonalData.php

@@ +1 @@
+<?php
+/**
+ * Testing Ajax handler for erasing personal data.
+ *
+ * @package WordPress\UnitTests
+ *
+ * @since 4.9.7
+ */
+
+/**
+ * Tests_Ajax_PrivacyExportPersonalData class.
+ *
+ * @since 4.9.7
+ *
+ * @group ajax
+ * @group privacy
+ * @covers wp_ajax_wp_privacy_erase_personal_data
+ */
+class Tests_Ajax_PrivacyErasePersonalData extends WP_Ajax_UnitTestCase {
+        /**
+         * User Request ID.
+         *
+         * @since 4.9.7
+         *
+         * @var int
+         */
+        protected static $request_id;
+
+        /**
+         * User Request Email.
+         *
+         * @since 4.9.7
+         *
+         * @var string
+         */
+        protected static $request_email;
+
+        /**
+         * Create user erase request fixtures.
+         *
+         * @param WP_UnitTest_Factory $factory Factory.
+         */
+        public static function wpSetUpBeforeClass( $factory ) {
+                self::$request_email = 'requester@example.com';
+                self::$request_id    = wp_create_user_request( self::$request_email, 'remove_personal_data' );
+        }
+
+        /**
+         * Register a custom personal data eraser.
+         */
+        public function setUp() {
+                parent::setUp();
+
+                // Make sure the erasers response is not modified and avoid e.g. sending emails.
+                remove_all_filters( 'wp_privacy_personal_data_erasure_page' );
+                remove_all_actions( 'wp_privacy_personal_data_erased' );
+
+                // Only use our custom privacy personal data eraser.
+                remove_all_filters( 'wp_privacy_personal_data_erasers' );
+                add_filter( 'wp_privacy_personal_data_erasers', array( $this, 'register_custom_personal_data_eraser' ) );
+        }
+
+        /**
+         * The function should successfully send exporters response data when the current user has the required capabilites.
+         *
+         * @ticket 43438
+         */
+        public function test_wp_ajax_wp_privacy_export_personal_data_should_success_when_current_user_has_required_capabilities() {
+                $this->_setRole( 'administrator' );
+                $this->assertTrue( current_user_can( 'erase_others_personal_data' ) );
+                $this->assertTrue( current_user_can( 'delete_users' ) );
+
+                // Set up a request.
+                $_POST['action']   = 'wp-privacy-erase-personal-data';
+                $_POST['security'] = wp_create_nonce( 'wp-privacy-erase-personal-data-' . self::$request_id );
+                $_POST['eraser']   = 1;
+                $_POST['page']     = 1;
+                $_POST['id']       = self::$request_id;
+
+                // Make the request.
+                try {
+                        $this->_handleAjax( $_POST['action'] );
+                } catch ( WPAjaxDieContinueException $e ) {
+                        unset( $e );
+                }
+
+                // Get the response.
+                $response = json_decode( $this->_last_response, true );
+
+                $this->assertTrue( $response['success'] );
+                $this->assertSame( 'A message regarding retained data for requester@example.com.', $response['data']['messages'][0] );
+                $this->assertTrue( $response['data']['items_removed'] );
+                $this->assertTrue( $response['data']['items_retained'] );
+                $this->assertTrue( $response['data']['done'] );
+        }
+
+        /**
+         * The function should send an error when the current user is missing required capabilities.
+         *
+         * @ticket 43438
+         */
+        public function test_wp_ajax_wp_privacy_erase_personal_data_should_error_when_current_user_missing_required_capabilities() {
+                $this->_setRole( 'author' );
+                $this->assertFalse( current_user_can( 'erase_others_personal_data' ) );
+                $this->assertFalse( current_user_can( 'delete_users' ) );
+
+                // Set up a request.
+                $_POST['action']   = 'wp-privacy-erase-personal-data';
+                $_POST['security'] = wp_create_nonce( 'wp-privacy-erase-personal-data-' . self::$request_id );
+                $_POST['eraser']   = 1;
+                $_POST['page']     = 1;
+                $_POST['id']       = self::$request_id;
+
+                // Make the request.
+                try {
+                        $this->_handleAjax( $_POST['action'] );
+                } catch ( WPAjaxDieContinueException $e ) {
+                        unset( $e );
+                }
+
+                // Get the response.
+                $response = json_decode( $this->_last_response, true );
+
+                $this->assertFalse( $response['success'] );
+                $this->assertSame( 'Invalid request.', $response['data'] );
+        }
+
+        /**
+         * The function should send error when the request ID is missing.
+         *
+         * @ticket 43438
+         */
+        public function test_wp_ajax_wp_privacy_erase_personal_data_should_error_when_missing_request_id() {
+                $this->_setRole( 'administrator' );
+
+                $this->assertNotWPError( self::$request_id );
+
+                // Set up a request.
+                $_POST['action']   = 'wp-privacy-erase-personal-data';
+                $_POST['security'] = wp_create_nonce( 'wp-privacy-erase-personal-data-' . self::$request_id );
+                $_POST['eraser']   = 1;
+                $_POST['page']     = 1;
+                $_POST['id']       = null; // Missing request ID.
+
+                // Make the request.
+                try {
+                        $this->_handleAjax( $_POST['action'] );
+                } catch ( WPAjaxDieContinueException $e ) {
+                        unset( $e );
+                }
+
+                // Get the response.
+                $response = json_decode( $this->_last_response, true );
+
+                $this->assertFalse( $response['success'] );
+                $this->assertSame( 'Missing request ID.', $response['data'] );
+        }
+
+        /**
+        * Register handler for a custom personal data exporter.
+        *
+        * @since 4.9.7
+        *
+        * @param  array $erasers An array of personal data erasers.
+        * @return array $erasers An array of personal data erasers.
+        */
+        public function register_custom_personal_data_eraser( $erasers ) {
+                $erasers['custom_exporter'] = array(
+                        'eraser_friendly_name' => __( 'Custom Eraser' ),
+                        'callback'             => array( $this, 'custom_personal_data_eraser' ),
+                );
+                return $erasers;
+        }
+
+        /**
+        * Custom Personal Data Eraser.
+        *
+        * @since 4.9.7
+        *
+        * @param  string $email_address The comment author email address.
+        * @param  int    $page          Page number.
+        * @return array  $return        Erase data.
+        */
+        public function custom_personal_data_eraser( $email_address, $page = 1 ) {
+                if ( 1 === $page ) {
+                        return array(
+                                'items_removed'  => true,
+                                'items_retained' => true,
+                                'messages'       => array( sprintf( 'A message regarding retained data for %s.', $email_address ) ),
+                                'done'           => true,
+                        );
+                }
+
+                return array(
+                        'items_removed'  => false,
+                        'items_retained' => false,
+                        'messages'       => array(),
+                        'done'           => true,
+                );
+        }
+
+}

new file tests/phpunit/tests/ajax/PrivacyExportPersonalData.php

@@ +1 @@
+<?php
+/**
+ * Testing Ajax handler for exporting personal data.
+ *
+ * @package WordPress\UnitTests
+ *
+ * @since 4.9.7
+ */
+
+/**
+ * Tests_Ajax_PrivacyExportPersonalData class.
+ *
+ * @since 4.9.7
+ *
+ * @group ajax
+ * @group privacy
+ * @covers wp_ajax_wp_privacy_export_personal_data
+ */
+class Tests_Ajax_PrivacyExportPersonalData extends WP_Ajax_UnitTestCase {
+        /**
+         * User Request ID.
+         *
+         * @since 4.9.7
+         *
+         * @var int
+         */
+        protected static $request_id;
+
+        /**
+         * User Request Email.
+         *
+         * @since 4.9.7
+         *
+         * @var string
+         */
+        protected static $request_email;
+
+        /**
+         * Ajax Action.
+         *
+         * @since 4.9.7
+         *
+         * @var string
+         */
+        protected static $action;
+
+        /**
+         * Exporter Index.
+         *
+         * @since 4.9.7
+         *
+         * @var int
+         */
+        protected static $exporter;
+
+        /**
+         * Exporter Key.
+         *
+         * @since 4.9.7
+         *
+         * @var string
+         */
+        protected static $exporter_key;
+
+        /**
+         * Exporter Friendly Name.
+         *
+         * @since 4.9.7
+         *
+         * @var string
+         */
+        protected static $exporter_friendly_name;
+
+        /**
+         * Page Index.
+         *
+         * @since 4.9.7
+         *
+         * @var int
+         */
+        protected static $page;
+
+        /**
+         * Send As Email.
+         *
+         * @since 4.9.7
+         *
+         * @var bool
+         */
+        protected static $send_as_email;
+
+        /**
+         * Last response parsed.
+         *
+         * @since 4.9.7
+         *
+         * @var array|null
+         */
+        protected $_last_response_parsed;
+
+        /**
+         * Create user export request fixtures.
+         *
+         * @since 4.9.7
+         *
+         * @param WP_UnitTest_Factory $factory Factory.
+         */
+        public static function wpSetUpBeforeClass( $factory ) {
+                self::$request_email          = 'requester@example.com';
+                self::$request_id             = wp_create_user_request( self::$request_email, 'export_personal_data' );
+                self::$action                 = 'wp-privacy-export-personal-data';
+                self::$exporter               = 1;
+                self::$exporter_key           = 'custom-exporter';
+                self::$exporter_friendly_name = 'Custom Exporter';
+                self::$page                   = 1;
+                self::$send_as_email          = false;
+        }
+
+        /**
+         * Setup before each test method.
+         *
+         * @since 4.9.7
+         */
+        public function setUp() {
+                parent::setUp();
+
+                // Make sure the exporter response is not modified and avoid e.g. writing export file to disk.
+                remove_all_filters( 'wp_privacy_personal_data_export_page' );
+
+                // Only use our custom privacy personal data exporter.
+                remove_all_filters( 'wp_privacy_personal_data_exporters' );
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_register_custom_personal_data_exporter' ) );
+        }
+
+        /**
+         * Clean up before each test method.
+         */
+        public function tearDown() {
+                parent::tearDown();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_register_custom_personal_data_exporter' ) );
+                $_POST = array();
+        }
+
+        /**
+         * The function should successfully send exporter data response when the current user has the required capability.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_success_when_current_user_has_required_capability() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call();
+
+                $this->assertTrue( $this->_last_response_parsed['success'] );
+                $this->assertTrue( current_user_can( 'export_others_personal_data' ) );
+                $this->assertSame( 'custom-exporter-item-id', $this->_last_response_parsed['data']['data']['item_id'] );
+                $this->assertSame( 'Email', $this->_last_response_parsed['data']['data']['data'][0]['name'] );
+                $this->assertSame( self::$request_email, $this->_last_response_parsed['data']['data']['data'][0]['value'] );
+        }
+
+        /**
+         * The function should error when the current user is missing the required capability.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_current_user_missing_required_capability() {
+                $this->_setRole( 'author' );
+
+                $this->_make_ajax_call();
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertFalse( current_user_can( 'export_others_personal_data' ) );
+                $this->assertSame( 'Invalid request.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the request ID is missing.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_missing_request_id() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'id' => null, // Missing request ID.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Missing request ID.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the request ID is invalid.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_invalid_id() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'id' => -1, // Invalid request ID.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Invalid request ID.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the requester's email address is invalid.
+         *
+         * @since 4.9.7
+
+         */
+        public function test_function_should_error_when_invalid_email_address() {
+                $this->_setRole( 'administrator' );
+                $request_id = wp_create_user_request( 'invalid-requester-email@example.com', 'export_personal_data' );
+                wp_update_post(
+                        array(
+                                'ID'         => $request_id,
+                                'post_title' => '', // Invalid requester's email address.
+                        )
+                );
+
+                $this->_make_ajax_call(
+                        array(
+                                'security' => wp_create_nonce( 'wp-privacy-export-personal-data-' . $request_id ),
+                                'id'       => $request_id,
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'A valid email address must be given.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the exporter index is missing.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_missing_exporter_index() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'exporter' => null, // Missing exporter index.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Missing exporter index.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the page index is missing.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_missing_page_index() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'page' => null, // Missing page index.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Missing page index.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when an exporter has improperly used the `wp_privacy_personal_data_exporters` filter.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_has_improperly_used_exporters_filter() {
+                $this->_setRole( 'administrator' );
+
+                // Improper filter usage: returns false instead of an expected array.
+                add_filter( 'wp_privacy_personal_data_exporters', '__return_false', 999 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', '__return_false', 999 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'An exporter has improperly used the registration filter.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the exporter index is negative.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_negative_exporter_index() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'exporter' => -1, // Negative exporter index.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Exporter index cannot be negative.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the exporter index is out of range.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_index_out_of_range() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'exporter' => PHP_INT_MAX, // Out of range exporter index.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Exporter index out of range.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when the page index is less than one.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_page_index_less_than_one() {
+                $this->_setRole( 'administrator' );
+
+                $this->_make_ajax_call(
+                        array(
+                                'page' => 0, // Page index less than one.
+                        )
+                );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame( 'Page index cannot be less than one.', $this->_last_response_parsed['data'] );
+        }
+
+        /**
+         * The function should send error when an exporter is not an array.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_not_array() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_not_array' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_not_array' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Expected an array describing the exporter at index %s.',
+                                self::$exporter_key
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_not_array( $exporters ) {
+                $exporters[ self::$exporter_key ] = false;
+                return $exporters;
+        }
+
+        /**
+         * The function should send error when an exporter is missing a friendly name.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_missing_friendly_name() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_friendly_name' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_friendly_name' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Exporter array at index %s does not include a friendly name.',
+                                self::$exporter_key
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_missing_friendly_name( $exporters ) {
+                unset( $exporters[ self::$exporter_key ]['exporter_friendly_name'] ); // Missing an exporter's friendly name.
+                return $exporters;
+        }
+
+        /**
+         * The function should send error when an exporter is missing a callback.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_missing_callback() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_callback' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_callback' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Exporter does not include a callback: %s.',
+                                self::$exporter_friendly_name
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_missing_callback( $exporters ) {
+                unset( $exporters[ self::$exporter_key ]['callback'] ); // Missing an exporter's callback.
+                return $exporters;
+        }
+
+        /**
+         * The function should send error when an exporter, at a given index, has an invalid callback.
+         *
+         * @since 4.9.7
+
+         */
+        public function test_function_should_error_when_exporter_index_invalid_callback() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_index_invalid_callback' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_index_invalid_callback' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Exporter callback is not a valid callback: %s.',
+                                self::$exporter_friendly_name
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_index_invalid_callback( $exporters ) {
+                $exporters[ self::$exporter_key ]['callback'] = false; // The exporter's callback isn't callable here.
+                return $exporters;
+        }
+
+        /**
+         * The function should send error when an exporter, at a given index, is missing an array response.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_index_invalid_response() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_index_invalid_response' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_index_invalid_response' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Expected response as an array from exporter: %s.',
+                                self::$exporter_friendly_name
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_index_invalid_response( $exporters ) {
+                $exporters[ self::$exporter_key ]['callback'] = '__return_null'; // A callback with an invalid response.
+                return $exporters;
+        }
+
+        /**
+         * The function should send error when an exporter is missing data in array response.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_missing_data_response() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_data_response' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_data_response' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Expected data in response array from exporter: %s.',
+                                self::$exporter_friendly_name
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_missing_data_response( $exporters ) {
+                $exporters[ self::$exporter_key ]['callback'] = array( $this, 'callback_missing_data_response' );
+                return $exporters;
+        }
+
+        /**
+         * Callback for exporter's response.
+         *
+         * @since 4.9.7
+         *
+         * @param  string $email_address The requester's email address.
+         * @param  int    $page          Page number.
+         *
+         * @return array $return Export data.
+         */
+        public function callback_missing_data_response( $email_address, $page = 1 ) {
+                $response = $this->callback_custom_personal_data_exporter( $email_address, $page );
+                unset( $response['data'] ); // Missing data part of response.
+                return $response;
+        }
+
+        /**
+         * The function should send error when an exporter is missing 'done' in array response.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_missing_done_response() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_done_response' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_done_response' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Expected done (boolean) in response array from exporter: %s.',
+                                self::$exporter_friendly_name
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_missing_done_response( $exporters ) {
+                $exporters[ self::$exporter_key ]['callback'] = array( $this, 'callback_missing_done_response' );
+                return $exporters;
+        }
+
+        /**
+         * Callback for exporter's response.
+         *
+         * @since 4.9.7
+         *
+         * @param  string $email_address The requester's email address.
+         * @param  int    $page          Page number.
+         *
+         * @return array $return Export data.
+         */
+        public function callback_missing_done_response( $email_address, $page = 1 ) {
+                $response = $this->callback_custom_personal_data_exporter( $email_address, $page );
+                unset( $response['done'] ); // Missing done part of response.
+                return $response;
+        }
+
+        /**
+         * The function should send error when an exporter is missing 'data' array in array response.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_should_error_when_exporter_missing_data_array_response() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_data_array_response' ), 20 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_exporter_missing_data_array_response' ), 20 );
+
+                $this->assertFalse( $this->_last_response_parsed['success'] );
+                $this->assertSame(
+                        sprintf(
+                                'Expected data array in response array from exporter: %s.',
+                                self::$exporter_friendly_name
+                        ),
+                        $this->_last_response_parsed['data']
+                );
+        }
+
+        /**
+         * Filter the array of exporters.
+         *
+         * @since 4.9.7
+         *
+         * @param  array $exporters Exporters.
+         *
+         * @return array $exporters Exporters.
+         */
+        public function filter_exporter_missing_data_array_response( $exporters ) {
+                $exporters[ self::$exporter_key ]['callback'] = array( $this, 'callback_missing_data_array_response' );
+                return $exporters;
+        }
+
+        /**
+         * Callback for exporter's response.
+         *
+         * @since 4.9.7
+         *
+         * @param  string $email_address The requester's email address.
+         * @param  int    $page          Page number.
+         *
+         * @return array $return Export data.
+         */
+        public function callback_missing_data_array_response( $email_address, $page = 1 ) {
+                $response         = $this->callback_custom_personal_data_exporter( $email_address, $page );
+                $response['data'] = false; // Not an array.
+                return $response;
+        }
+
+        /**
+         * The function's output should be filterable with the `wp_privacy_personal_data_export_page` filter.
+         *
+         * @since 4.9.7
+         */
+        public function test_function_output_should_be_filterable() {
+                $this->_setRole( 'administrator' );
+
+                add_filter( 'wp_privacy_personal_data_export_page', array( $this, 'filter_exporter_data_response' ), 20, 7 );
+                $this->_make_ajax_call();
+                remove_filter( 'wp_privacy_personal_data_export_page', array( $this, 'filter_exporter_data_response' ), 20 );
+
+                $expected_group_label = sprintf(
+                        '%s-%s-%s-%s-%s-%s',
+                        self::$exporter,
+                        self::$page,
+                        self::$request_email,
+                        self::$request_id,
+                        self::$send_as_email,
+                        self::$exporter_key
+                );
+
+                $this->assertTrue( $this->_last_response_parsed['success'] );
+                $this->assertSame( $expected_group_label, $this->_last_response_parsed['data']['group_label'] );
+                $this->assertSame( 'filtered_group_id', $this->_last_response_parsed['data']['group_id'] );
+                $this->assertSame( 'filtered_item_id', $this->_last_response_parsed['data']['item_id'] );
+                $this->assertSame( 'filtered_name', $this->_last_response_parsed['data']['data'][0]['name'] );
+                $this->assertSame( 'filtered_value', $this->_last_response_parsed['data']['data'][0]['value'] );
+        }
+
+        /**
+         * Filter exporter's data response.
+         *
+         * @since 4.9.7
+         *
+         * @param array  $response        The personal data for the given exporter and page.
+         * @param int    $exporter_index  The index of the exporter that provided this data.
+         * @param string $email_address   The email address associated with this personal data.
+         * @param int    $page            The page for this response.
+         * @param int    $request_id      The privacy request post ID associated with this request.
+         * @param bool   $send_as_email   Whether the final results of the export should be emailed to the user.
+         * @param string $exporter_key    The key (slug) of the exporter that provided this data.
+         *
+         * @return array $response The personal data for the given exporter and page.
+         */
+        public function filter_exporter_data_response( $response, $exporter_index, $email_address, $page, $request_id, $send_as_email, $exporter_key ) {
+                $group_label                  = sprintf(
+                        '%s-%s-%s-%s-%s-%s',
+                        $exporter_index,
+                        $page,
+                        $email_address,
+                        $request_id,
+                        $send_as_email,
+                        $exporter_key
+                );
+                $response['group_label']      = $group_label;
+                $response['group_id']         = 'filtered_group_id';
+                $response['item_id']          = 'filtered_item_id';
+                $response['data'][0]['name']  = 'filtered_name';
+                $response['data'][0]['value'] = 'filtered_value';
+
+                return $response;
+        }
+
+        /**
+        * Filter to register a custom personal data exporter.
+        *
+        * @since 4.9.7
+        *
+        * @param  array $exporters An array of personal data exporters.
+        *
+        * @return array $exporters An array of personal data exporters.
+        */
+        public function filter_register_custom_personal_data_exporter( $exporters ) {
+                $exporters[ self::$exporter_key ] = array(
+                        'exporter_friendly_name' => self::$exporter_friendly_name,
+                        'callback'               => array( $this, 'callback_custom_personal_data_exporter' ),
+                );
+                return $exporters;
+        }
+
+        /**
+        * Callback for a custom personal data exporter.
+        *
+        * @since 4.9.7
+        *
+        * @param  string $email_address The requester's email address.
+        * @param  int    $page          Page number.
+        *
+        * @return array $response Export data response.
+        */
+        public function callback_custom_personal_data_exporter( $email_address, $page = 1 ) {
+                $data_to_export = array();
+
+                if ( 1 === $page ) {
+                        $data_to_export = array(
+                                'group_id'    => self::$exporter_key . '-group-id',
+                                'group_label' => self::$exporter_key . '-group-label',
+                                'item_id'     => self::$exporter_key . '-item-id',
+                                'data'        => array(
+                                        array(
+                                                'name'  => 'Email',
+                                                'value' => $email_address,
+                                        ),
+                                ),
+                        );
+                }
+
+                return array(
+                        'data' => $data_to_export,
+                        'done' => true,
+                );
+        }
+
+        /**
+         * Helper function for ajax handler.
+         *
+         * @since 4.9.7
+         *
+         * @param array $args Ajax request arguments.
+         */
+        protected function _make_ajax_call( $args = array() ) {
+                $this->_last_response_parsed = null;
+                $this->_last_response        = '';
+
+                $defaults = array(
+                        'action'      => self::$action,
+                        'security'    => wp_create_nonce( self::$action . '-' . self::$request_id ),
+                        'exporter'    => self::$exporter,
+                        'page'        => self::$page,
+                        'sendAsEmail' => self::$send_as_email,
+                        'id'          => self::$request_id,
+                );
+
+                $_POST = wp_parse_args( $args, $defaults );
+
+                try {
+                        $this->_handleAjax( self::$action );
+                } catch ( WPAjaxDieContinueException $e ) {
+                        unset( $e );
+                }
+                if ( $this->_last_response ) {
+                        $this->_last_response_parsed = json_decode( $this->_last_response, true );
+                }
+        }
+
+}