<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN"
"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head><meta http-equiv="content-type" content="text/html; charset=utf-8" />
<title>[44786] trunk/tests/phpunit/tests/privacy: Privacy: Add missing unit tests for exporting personal data.</title>
</head>
<body>

<style type="text/css"><!--
#msg dl.meta { border: 1px #006 solid; background: #369; padding: 6px; color: #fff; }
#msg dl.meta dt { float: left; width: 6em; font-weight: bold; }
#msg dt:after { content:':';}
#msg dl, #msg dt, #msg ul, #msg li, #header, #footer, #logmsg { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt;  }
#msg dl a { font-weight: bold}
#msg dl a:link    { color:#fc3; }
#msg dl a:active  { color:#ff0; }
#msg dl a:visited { color:#cc6; }
h3 { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt; font-weight: bold; }
#msg pre { white-space: pre-line; overflow: auto; background: #ffc; border: 1px #fa0 solid; padding: 6px; }
#logmsg { background: #ffc; border: 1px #fa0 solid; padding: 1em 1em 0 1em; }
#logmsg p, #logmsg pre, #logmsg blockquote { margin: 0 0 1em 0; }
#logmsg p, #logmsg li, #logmsg dt, #logmsg dd { line-height: 14pt; }
#logmsg h1, #logmsg h2, #logmsg h3, #logmsg h4, #logmsg h5, #logmsg h6 { margin: .5em 0; }
#logmsg h1:first-child, #logmsg h2:first-child, #logmsg h3:first-child, #logmsg h4:first-child, #logmsg h5:first-child, #logmsg h6:first-child { margin-top: 0; }
#logmsg ul, #logmsg ol { padding: 0; list-style-position: inside; margin: 0 0 0 1em; }
#logmsg ul { text-indent: -1em; padding-left: 1em; }#logmsg ol { text-indent: -1.5em; padding-left: 1.5em; }
#logmsg > ul, #logmsg > ol { margin: 0 0 1em 0; }
#logmsg pre { background: #eee; padding: 1em; }
#logmsg blockquote { border: 1px solid #fa0; border-left-width: 10px; padding: 1em 1em 0 1em; background: white;}
#logmsg dl { margin: 0; }
#logmsg dt { font-weight: bold; }
#logmsg dd { margin: 0; padding: 0 0 0.5em 0; }
#logmsg dd:before { content:'\00bb';}
#logmsg table { border-spacing: 0px; border-collapse: collapse; border-top: 4px solid #fa0; border-bottom: 1px solid #fa0; background: #fff; }
#logmsg table th { text-align: left; font-weight: normal; padding: 0.2em 0.5em; border-top: 1px dotted #fa0; }
#logmsg table td { text-align: right; border-top: 1px dotted #fa0; padding: 0.2em 0.5em; }
#logmsg table thead th { text-align: center; border-bottom: 1px solid #fa0; }
#logmsg table th.Corner { text-align: left; }
#logmsg hr { border: none 0; border-top: 2px dashed #fa0; height: 1px; }
#header, #footer { color: #fff; background: #636; border: 1px #300 solid; padding: 6px; }
#patch { width: 100%; }
#patch h4 {font-family: verdana,arial,helvetica,sans-serif;font-size:10pt;padding:8px;background:#369;color:#fff;margin:0;}
#patch .propset h4, #patch .binary h4 {margin:0;}
#patch pre {padding:0;line-height:1.2em;margin:0;}
#patch .diff {width:100%;background:#eee;padding: 0 0 10px 0;overflow:auto;}
#patch .propset .diff, #patch .binary .diff  {padding:10px 0;}
#patch span {display:block;padding:0 10px;}
#patch .modfile, #patch .addfile, #patch .delfile, #patch .propset, #patch .binary, #patch .copfile {border:1px solid #ccc;margin:10px 0;}
#patch ins {background:#dfd;text-decoration:none;display:block;padding:0 10px;}
#patch del {background:#fdd;text-decoration:none;display:block;padding:0 10px;}
#patch .lines, .info {color:#888;background:#fff;}
--></style>
<div id="msg">
<dl class="meta" style="font-size: 105%">
<dt style="float: left; width: 6em; font-weight: bold">Revision</dt> <dd><a style="font-weight: bold" href="https://core.trac.wordpress.org/changeset/44786">44786</a><script type="application/ld+json">{"@context":"http://schema.org","@type":"EmailMessage","description":"Review this Commit","action":{"@type":"ViewAction","url":"https://core.trac.wordpress.org/changeset/44786","name":"Review Commit"}}</script></dd>
<dt style="float: left; width: 6em; font-weight: bold">Author</dt> <dd>desrosj</dd>
<dt style="float: left; width: 6em; font-weight: bold">Date</dt> <dd>2019-03-04 19:35:30 +0000 (Mon, 04 Mar 2019)</dd>
</dl>

<pre style='padding-left: 1em; margin: 2em 0; border-left: 2px solid #ccc; line-height: 1.25; font-size: 105%; font-family: sans-serif'>Privacy: Add missing unit tests for exporting personal data.

Props allendav, birgire, iandunn, desrosj, garrett-eclipse.
Fixes 44233.</pre>

<h3>Added Paths</h3>
<ul>
<li><a href="#trunktestsphpunittestsprivacywpPrivacyGeneratePersonalDataExportFilephp">trunk/tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php</a></li>
<li><a href="#trunktestsphpunittestsprivacywpPrivacyProcessPersonalDataExportPagephp">trunk/tests/phpunit/tests/privacy/wpPrivacyProcessPersonalDataExportPage.php</a></li>
</ul>

</div>
<div id="patch">
<h3>Diff</h3>
<a id="trunktestsphpunittestsprivacywpPrivacyGeneratePersonalDataExportFilephp"></a>
<div class="addfile"><h4 style="background-color: #eee; color: inherit; margin: 1em 0; padding: 1.3em; font-size: 115%">Added: trunk/tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php</h4>
<pre class="diff"><span>
<span class="info" style="display: block; padding: 0 10px; color: #888">--- trunk/tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php                             (rev 0)
+++ trunk/tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php       2019-03-04 19:35:30 UTC (rev 44786)
</span><span class="lines" style="display: block; padding: 0 10px; color: #888">@@ -0,0 +1,308 @@
</span><ins style="background-color: #dfd; text-decoration:none; display:block; padding: 0 10px">+<?php
+/**
+ * Define a class to test `wp_privacy_generate_personal_data_export_file()`.
+ *
+ * @package WordPress
+ * @subpackage UnitTests
+ * @since 5.2.0
+ */
+
+/**
+ * Test cases for `wp_privacy_generate_personal_data_export_file()`.
+ *
+ * @group privacy
+ * @covers ::wp_privacy_generate_personal_data_export_file
+ *
+ * @since 5.2.0
+ */
+class Tests_Privacy_WpPrivacyGeneratePersonalDataExportFile extends WP_UnitTestCase {
+       /**
+        * An Export Request ID
+        *
+        * @since 5.2.0
+        *
+        * @var int $export_request_id
+        */
+       protected static $export_request_id;
+
+       /**
+        * The full path to the export file for the current test method.
+        *
+        * @since 5.2.0
+        *
+        * @var string $export_file_name
+        */
+       public $export_file_name = '';
+
+       /**
+        * The full path to the exports directory.
+        *
+        * @since 5.2.0
+        *
+        * @var string $exports_dir
+        */
+       public static $exports_dir;
+
+       /**
+        * Create fixtures that are shared by multiple test cases.
+        *
+        * @since 5.2.0
+        *
+        * @param WP_UnitTest_Factory $factory The base factory object.
+        */
+       public static function wpSetUpBeforeClass( $factory ) {
+               self::$export_request_id = wp_create_user_request( 'export-requester@example.com', 'export_personal_data' );
+               update_post_meta( self::$export_request_id, '_export_data_grouped', array() );
+               self::$exports_dir = wp_privacy_exports_dir();
+       }
+
+       /**
+        * Set up the test fixture.
+        *
+        * Override `wp_die()`, pretend to be Ajax, and suppress `E_WARNING`s.
+        *
+        * @since 5.2.0
+        */
+       public function setUp() {
+               parent::setUp();
+
+               $this->export_file_name = '';
+
+               if ( ! class_exists( 'ZipArchive' ) ) {
+                       $this->markTestSkipped( 'The ZipArchive class is missing.' );
+               }
+
+               if ( ! $this->remove_exports_dir() ) {
+                       $this->markTestSkipped( 'Existing exports directory could not be removed. Skipping test.' );
+               }
+
+               // We need to override the die handler. Otherwise, the unit tests will die too.
+               add_filter( 'wp_die_ajax_handler', array( $this, 'get_wp_die_handler' ), 1, 1 );
+               add_filter( 'wp_doing_ajax', '__return_true' );
+               add_action( 'wp_privacy_personal_data_export_file_created', array( $this, 'action_wp_privacy_personal_data_export_file_created' ) );
+
+               // Suppress warnings from "Cannot modify header information - headers already sent by".
+               $this->_error_level = error_reporting();
+               error_reporting( $this->_error_level & ~E_WARNING );
+       }
+
+       /**
+        * Tear down the test fixture.
+        *
+        * Remove the `wp_die()` override, restore error reporting.
+        *
+        * @since 5.2.0
+        */
+       public function tearDown() {
+               $this->remove_exports_dir();
+               error_reporting( $this->_error_level );
+               parent::tearDown();
+       }
+
+       /**
+        * Stores the name of the export zip file to check the file is actually created.
+        *
+        * @since 5.2.0
+        *
+        * @param string $archive_name Created export zip file path.
+        */
+       public function action_wp_privacy_personal_data_export_file_created( $archive_name ) {
+               $this->export_file_name = $archive_name;
+       }
+
+       /**
+        * Removes the privacy exports directory, including files and subdirectories.
+        *
+        * Ignores hidden files and has upper limit of nested levels, because of `list_files()`.
+        *
+        * @since 5.2.0
+        *
+        * @return bool Whether the privacy exports directory was removed.
+        */
+       private function remove_exports_dir() {
+               /**
+                * The `$exports_dir` will be a file after the `test_detect_cannot_create_folder()` test method, or,
+                * if an incorrect value is returned to the `wp_privacy_exports_dir` filter.
+                */
+               if ( is_file( untrailingslashit( self::$exports_dir ) ) ) {
+                       wp_delete_file( untrailingslashit( self::$exports_dir ) );
+                       return ! is_file( untrailingslashit( self::$exports_dir ) );
+               }
+
+               if ( ! is_dir( self::$exports_dir ) ) {
+                       return true;
+               }
+
+               chmod( self::$exports_dir, 0755 );
+
+               $files = list_files( self::$exports_dir );
+
+               // Delete files first, then delete subdirectories.
+               foreach ( $files as $file ) {
+                       if ( is_file( $file ) ) {
+                               wp_delete_file( $file );
+                       }
+               }
+
+               foreach ( $files as $file ) {
+                       if ( is_dir( $file ) ) {
+                               rmdir( $file );
+                       }
+               }
+
+               rmdir( self::$exports_dir );
+
+               return ! is_dir( self::$exports_dir );
+       }
+
+       /**
+        * When a remove request ID is passed to the export function an error should be displayed.
+        *
+        * @ticket 44233
+        */
+       public function test_rejects_remove_requests() {
+               $request_id = wp_create_user_request( 'removal-requester@example.com', 'remove_personal_data' );
+
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( '{"success":false,"data":"Invalid request ID when generating export file."}' );
+               wp_privacy_generate_personal_data_export_file( $request_id );
+       }
+
+       /**
+        * When an invalid request ID is passed an error should be displayed.
+        *
+        * @ticket 44233
+        */
+       public function test_invalid_request_id() {
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( '{"success":false,"data":"Invalid request ID when generating export file."}' );
+               wp_privacy_generate_personal_data_export_file( 123456789 );
+       }
+
+       /**
+        * When the request post title is not a valid email an error should be displayed.
+        *
+        * @ticket 44233
+        */
+       public function test_rejects_requests_with_bad_email_addresses() {
+               $request_id = wp_create_user_request( 'bad-email-requester@example.com', 'export_personal_data' );
+
+               wp_update_post(
+                       array(
+                               'ID'         => $request_id,
+                               'post_title' => 'not-a-valid-email-address',
+                       )
+               );
+
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( '{"success":false,"data":"Invalid email address when generating export file."}' );
+               wp_privacy_generate_personal_data_export_file( $request_id );
+       }
+
+       /**
+        * When the export directory fails to be created an error should be displayed.
+        *
+        * @ticket 44233
+        */
+       public function test_detect_cannot_create_folder() {
+               // Create a file with the folder name to ensure the function cannot create a folder.
+               touch( untrailingslashit( self::$exports_dir ) );
+
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( '{"success":false,"data":"Unable to create export folder."}' );
+               wp_privacy_generate_personal_data_export_file( self::$export_request_id );
+       }
+
+       /**
+        * When the index.html file cannot be created an error should be displayed.
+        *
+        * @ticket 44233
+        */
+       public function test_detects_cannot_create_index() {
+               // Make the export directory read only so the index.html file can't be created.
+               mkdir( self::$exports_dir );
+               chmod( self::$exports_dir, 0444 );
+
+               if ( '444' !== substr( decoct( fileperms( self::$exports_dir ) ), -3 ) ) {
+                       $this->markTestSkipped( 'Data export directory permissions were not changed correctly.' );
+               }
+
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( '{"success":false,"data":"Unable to protect export folder from browsing."}' );
+               wp_privacy_generate_personal_data_export_file( self::$export_request_id );
+       }
+
+       /**
+        * Test that an index.html file can be added to the export directory.
+        *
+        * @ticket 44233
+        */
+       public function test_creates_index_in_export_folder() {
+               $this->expectOutputString( '' );
+               wp_privacy_generate_personal_data_export_file( self::$export_request_id );
+
+               $this->assertTrue( file_exists( self::$exports_dir . 'index.html' ) );
+       }
+
+       /**
+        * When the export directory is not writable the report should fail to write.
+        *
+        * @ticket 44233
+        */
+       public function test_detects_cannot_write_html() {
+               // Make the folder read only so HTML writing will fail.
+               mkdir( self::$exports_dir );
+               touch( self::$exports_dir . 'index.html' );
+               chmod( self::$exports_dir, 0555 );
+
+               if ( '555' !== substr( decoct( fileperms( self::$exports_dir ) ), -3 ) ) {
+                       $this->markTestSkipped( 'Data export directory permissions were not changed correctly.' );
+               }
+
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( '{"success":false,"data":"Unable to open export file (HTML report) for writing."}' );
+               wp_privacy_generate_personal_data_export_file( self::$export_request_id );
+
+               $this->assertEmpty( $this->export_file_name );
+       }
+
+       /**
+        * Test that an export file is successfully created.
+        *
+        * @ticket 44233
+        */
+       public function test_can_succeed() {
+               wp_privacy_generate_personal_data_export_file( self::$export_request_id );
+
+               $this->assertTrue( file_exists( $this->export_file_name ) );
+       }
+
+       /**
+        * Test the export file has all the expected parts.
+        *
+        * @ticket 44233
+        */
+       public function test_contents() {
+               $this->expectOutputString( '' );
+               wp_privacy_generate_personal_data_export_file( self::$export_request_id );
+               $this->assertTrue( file_exists( $this->export_file_name ) );
+
+               $report_dir = trailingslashit( self::$exports_dir . 'test_contents' );
+               mkdir( $report_dir );
+
+               $zip        = new ZipArchive();
+               $opened_zip = $zip->open( $this->export_file_name );
+               $this->assertTrue( $opened_zip );
+
+               $zip->extractTo( $report_dir );
+               $zip->close();
+               $this->assertTrue( file_exists( $report_dir . 'index.html' ) );
+
+               $report_contents = file_get_contents( $report_dir . 'index.html' );
+               $request         = wp_get_user_request_data( self::$export_request_id );
+
+               $this->assertContains( '<h1>Personal Data Export</h1>', $report_contents );
+               $this->assertContains( '<h2>About</h2>', $report_contents );
+               $this->assertContains( $request->email, $report_contents );
+       }
+}
</ins><span class="cx" style="display: block; padding: 0 10px">Property changes on: trunk/tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php
</span><span class="cx" style="display: block; padding: 0 10px">___________________________________________________________________
</span></span></pre></div>
<a id="svnexecutable"></a>
<div class="addfile"><h4 style="background-color: #eee; color: inherit; margin: 1em 0; padding: 1.3em; font-size: 115%">Added: svn:executable</h4></div>
<ins style="background-color: #dfd; text-decoration:none; display:block; padding: 0 10px">+*
</ins><span class="cx" style="display: block; padding: 0 10px">\ No newline at end of property
</span><a id="trunktestsphpunittestsprivacywpPrivacyProcessPersonalDataExportPagephp"></a>
<div class="addfile"><h4 style="background-color: #eee; color: inherit; margin: 1em 0; padding: 1.3em; font-size: 115%">Added: trunk/tests/phpunit/tests/privacy/wpPrivacyProcessPersonalDataExportPage.php</h4>
<pre class="diff"><span>
<span class="info" style="display: block; padding: 0 10px; color: #888">--- trunk/tests/phpunit/tests/privacy/wpPrivacyProcessPersonalDataExportPage.php                              (rev 0)
+++ trunk/tests/phpunit/tests/privacy/wpPrivacyProcessPersonalDataExportPage.php        2019-03-04 19:35:30 UTC (rev 44786)
</span><span class="lines" style="display: block; padding: 0 10px; color: #888">@@ -0,0 +1,700 @@
</span><ins style="background-color: #dfd; text-decoration:none; display:block; padding: 0 10px">+<?php
+/**
+ * Test cases for the `wp_privacy_process_personal_data_export_page()` function.
+ *
+ * @package WordPress
+ * @subpackage UnitTests
+ * @since 5.2.0
+ */
+
+/**
+ * Tests_Privacy_WpPrivacyProcessPersonalDataExportPage class.
+ *
+ * @group privacy
+ * @covers ::wp_privacy_process_personal_data_export_page
+ *
+ * @since 5.2.0
+ */
+class Tests_Privacy_WpPrivacyProcessPersonalDataExportPage extends WP_UnitTestCase {
+       /**
+        * Request ID.
+        *
+        * @since 5.2.0
+        *
+        * @var int $request_id
+        */
+       protected static $request_id;
+
+       /**
+        * Response for the First Page.
+        *
+        * @since 5.2.0
+        *
+        * @var array $response
+        */
+       protected static $response_first_page;
+
+       /**
+        * Response for the Last Page.
+        *
+        * @since 5.2.0
+        *
+        * @var array $response_last_page
+        */
+       protected static $response_last_page;
+
+       /**
+        * Export File Url.
+        *
+        * @since 5.2.0
+        *
+        * @var string $export_file_url
+        */
+       protected static $export_file_url;
+
+       /**
+        * Requester Email.
+        *
+        * @since 5.2.0
+        *
+        * @var string $requester_email
+        */
+       protected static $requester_email;
+
+       /**
+        * Index Of The First Page.
+        *
+        * @since 5.2.0
+        *
+        * @var int $page
+        */
+       protected static $page_index_first;
+
+       /**
+        * Index Of The Last Page.
+        *
+        * @since 5.2.0
+        *
+        * @var int $page_index_last
+        */
+       protected static $page_index_last;
+
+       /**
+        * Index of the First Exporter.
+        *
+        * @since 5.2.0
+        *
+        * @var int $exporter_index_first
+        */
+       protected static $exporter_index_first;
+
+       /**
+        * Index of the Last Exporter.
+        *
+        * @since 5.2.0
+        *
+        * @var int $exporter_index_last
+        */
+       protected static $exporter_index_last;
+
+       /**
+        * Key of the First Exporter.
+        *
+        * @since 5.2.0
+        *
+        * @var int $exporter_key_first
+        */
+       protected static $exporter_key_first;
+
+       /**
+        * Key of the Last Exporter.
+        *
+        * @since 5.2.0
+        *
+        * @var int $exporter_key_last
+        */
+       protected static $exporter_key_last;
+
+       /**
+        * Export data stored on the `wp_privacy_personal_data_export_file` action hook.
+        *
+        * @var string $_export_data_grouped_fetched_within_callback
+        */
+       public $_export_data_grouped_fetched_within_callback;
+
+       /**
+        * Create user request fixtures shared by test methods.
+        *
+        * @since 5.2.0
+        *
+        * @param WP_UnitTest_Factory $factory Factory.
+        */
+       public static function wpSetUpBeforeClass( $factory ) {
+               self::$requester_email      = 'requester@example.com';
+               self::$export_file_url      = wp_privacy_exports_url() . 'wp-personal-data-file-requester-at-example-com-Wv0RfMnGIkl4CFEDEEkSeIdfLmaUrLsl.zip';
+               self::$request_id           = wp_create_user_request( self::$requester_email, 'export_personal_data' );
+               self::$page_index_first     = 1;
+               self::$page_index_last      = 2;
+               self::$exporter_index_first = 1;
+               self::$exporter_index_last  = 2;
+               self::$exporter_key_first   = 'custom-exporter-first';
+               self::$exporter_key_last    = 'custom-exporter-last';
+
+               $data = array(
+                       array(
+                               'group_id'    => 'custom-exporter-group-id',
+                               'group_label' => 'custom-exporter-group-label',
+                               'item_id'     => 'custom-exporter-item-id',
+                               'data'        => array(
+                                       array(
+                                               'name'  => 'Email',
+                                               'value' => self::$requester_email,
+                                       ),
+                               ),
+                       ),
+               );
+
+               self::$response_first_page = array(
+                       'done' => false,
+                       'data' => $data,
+               );
+
+               self::$response_last_page = array(
+                       'done' => true,
+                       'data' => $data,
+               );
+       }
+
+       /**
+        * Setup before each test method.
+        *
+        * @since 5.2.0
+        */
+       public function setUp() {
+               parent::setUp();
+
+               // Avoid writing export files to disk. Using `WP_Filesystem_MockFS` is blocked by #44204.
+               remove_action( 'wp_privacy_personal_data_export_file', 'wp_privacy_generate_personal_data_export_file', 10 );
+
+               // Register our custom data exporters, very late, so we can override other unrelated exporters.
+               add_filter( 'wp_privacy_personal_data_exporters', array( $this, 'filter_register_custom_personal_data_exporters' ), 9999 );
+
+               // Set Ajax context for `wp_send_json()` and `wp_die()`.
+               add_filter( 'wp_doing_ajax', '__return_true' );
+
+               // Set up a `wp_die()` ajax handler that throws an exception, to be able to get
+               // the error message from `wp_send_json_error( 'some message here' )`,
+               // called by `wp_privacy_process_personal_data_export_page()`.
+               add_filter( 'wp_die_ajax_handler', array( $this, 'get_wp_die_handler' ), 1, 1 );
+
+               // Suppress warnings from "Cannot modify header information - headers already sent by".
+               $this->_error_level = error_reporting();
+               error_reporting( $this->_error_level & ~E_WARNING );
+       }
+
+       /**
+        * Clean up after each test method.
+        *
+        * @since 5.2.0
+        */
+       public function tearDown() {
+               error_reporting( $this->_error_level );
+
+               parent::tearDown();
+       }
+
+       /**
+        * Filter to register custom personal data exporters.
+        *
+        * @since 5.2.0
+        *
+        * @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_exporters( $exporters ) {
+               // Let's override other unrelated exporters.
+               $exporters = array();
+
+               $exporters[ self::$exporter_key_first ] = array(
+                       'exporter_friendly_name' => __( 'Custom Exporter #1' ),
+                       'callback'               => null,
+               );
+               $exporters[ self::$exporter_key_last ]  = array(
+                       'exporter_friendly_name' => __( 'Custom Exporter #2' ),
+                       'callback'               => null,
+               );
+
+               return $exporters;
+       }
+
+       /**
+        * Set up a test method to properly assert an exception.
+        *
+        * @since 5.2.0
+        *
+        * @param string $expected_output The expected string exception output.
+        */
+       private function _setup_expected_failure( $expected_output ) {
+               $this->setExpectedException( 'WPDieException' );
+               $this->expectOutputString( $expected_output );
+       }
+
+       /**
+        * Ensure the correct errors are returned when exporter responses are incorrect.
+        *
+        * @ticket 44233
+        *
+        * @dataProvider data_wp_privacy_process_personal_data_export_page
+        *
+        * @param string|array $expected_response The response from the personal data exporter for the given test.
+        */
+       public function test_wp_privacy_process_personal_data_export_page( $expected_response ) {
+               $actual_response = wp_privacy_process_personal_data_export_page(
+                       $expected_response,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       self::$request_id,
+                       true,
+                       self::$exporter_key_last
+               );
+
+               $this->assertSame( $expected_response, $actual_response );
+       }
+
+       /**
+        * Provide test cases for `test_wp_privacy_process_personal_data_export_page()`.
+        *
+        * @since 5.2.0
+        *
+        * @return array {
+        *     @type array {
+        *         @type string|array $response The response from the personal data exporter to test. Can be a string or an array.
+        *     }
+        * }
+        */
+       public function data_wp_privacy_process_personal_data_export_page() {
+               return array(
+                       // Response is not an array.
+                       array(
+                               'not-an-array',
+                       ),
+                       // Missing `done` array key.
+                       array(
+                               array(
+                                       'missing-done-array-key' => true,
+                               ),
+                       ),
+                       // Missing `data` array key.
+                       array(
+                               array(
+                                       'done' => true,
+                               ),
+                       ),
+                       // `data` key is not an array.
+                       array(
+                               array(
+                                       'done' => true,
+                                       'data' => 'not-an-array',
+                               ),
+                       ),
+                       array(
+                               array(
+                                       'done' => true,
+                                       'data' => array(),
+                               ),
+                       ),
+               );
+       }
+
+       /**
+        * Provide test scenarios for both sending and not sending an email.
+        *
+        * @since 5.2.0
+        *
+        * @return array {
+        *     @type array {
+        *         @type bool $send_as_email Whether the final results of the export should be emailed to the user.
+        *     }
+        * }
+        */
+       public function data_send_as_email_options() {
+               return array(
+                       array(
+                               true,
+                       ),
+                       array(
+                               false,
+                       ),
+               );
+       }
+
+       /**
+        * The function should send a JSON error when receiving an invalid request ID.
+        *
+        * @ticket 44233
+        *
+        * @dataProvider data_send_as_email_options
+        *
+        * @param bool Whether the final results of the export should be emailed to the user.
+        */
+       public function test_send_error_when_invalid_request_id( $send_as_email ) {
+               $response           = array(
+                       'done' => true,
+                       'data' => array(),
+               );
+               $invalid_request_id = 0;
+
+               // Process data, given the last exporter, on the last page and send as email.
+               $this->_setup_expected_failure( '{"success":false,"data":"Invalid request ID when merging exporter data."}' );
+
+               wp_privacy_process_personal_data_export_page(
+                       $response,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       $invalid_request_id,
+                       $send_as_email,
+                       self::$exporter_key_last
+               );
+       }
+
+       /**
+        * The function should send a JSON error when the request has an invalid action name.
+        *
+        * @ticket 44233
+        *
+        * @dataProvider data_send_as_email_options
+        *
+        * @param bool Whether the final results of the export should be emailed to the user.
+        */
+       public function test_send_error_when_invalid_request_action_name( $send_as_email ) {
+               $response = array(
+                       'done' => true,
+                       'data' => array(),
+               );
+
+               // Create a valid request ID, but for a different action than the function expects.
+               $request_id = wp_create_user_request( self::$requester_email, 'remove_personal_data' );
+
+               // Process data, given the last exporter, on the last page and send as email.
+               $this->_setup_expected_failure( '{"success":false,"data":"Invalid request ID when merging exporter data."}' );
+
+               wp_privacy_process_personal_data_export_page(
+                       $response,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       $request_id,
+                       $send_as_email,
+                       self::$exporter_key_last
+               );
+       }
+
+       /**
+        * The function should store export raw data until the export finishes. Then the meta key should be deleted.
+        *
+        * @ticket 44233
+        *
+        * @dataProvider data_send_as_email_options
+        *
+        * @param bool Whether the final results of the export should be emailed to the user.
+        *
+        */
+       public function test_raw_data_post_meta( $send_as_email ) {
+               $this->assertEmpty( get_post_meta( self::$request_id, '_export_data_raw', true ) );
+
+               // Adds post meta when processing data, given the first exporter on the first page and send as email.
+               wp_privacy_process_personal_data_export_page(
+                       self::$response_first_page,
+                       self::$exporter_index_first,
+                       self::$requester_email,
+                       self::$page_index_first,
+                       self::$request_id,
+                       $send_as_email,
+                       self::$exporter_key_first
+               );
+
+               $this->assertNotEmpty( get_post_meta( self::$request_id, '_export_data_raw', true ) );
+
+               // Deletes post meta when processing data, given the last exporter on the last page and send as email.
+               wp_privacy_process_personal_data_export_page(
+                       self::$response_last_page,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       self::$request_id,
+                       $send_as_email,
+                       self::$exporter_key_last
+               );
+
+               $this->assertEmpty( get_post_meta( self::$request_id, '_export_data_raw', true ) );
+       }
+
+       /**
+        * The function should add `_export_data_grouped` post meta for the request, only available
+        * when personal data export file is generated.
+        *
+        * @ticket 44233
+        *
+        * @dataProvider data_send_as_email_options
+        *
+        * @param bool Whether the final results of the export should be emailed to the user.
+        */
+       public function test_add_post_meta_with_groups_data_only_available_when_export_file_generated( $send_as_email ) {
+               // Adds post meta when processing data, given the first exporter on the first page and send as email.
+               wp_privacy_process_personal_data_export_page(
+                       self::$response_first_page,
+                       self::$exporter_index_first,
+                       self::$requester_email,
+                       self::$page_index_first,
+                       self::$request_id,
+                       true,
+                       self::$exporter_key_first
+               );
+               $this->assertEmpty( get_post_meta( self::$request_id, '_export_data_grouped', true ) );
+
+               add_action( 'wp_privacy_personal_data_export_file', array( $this, 'action_callback_to_get_export_groups_data' ) );
+
+               // Process data, given the last exporter on the last page and send as email.
+               wp_privacy_process_personal_data_export_page(
+                       self::$response_last_page,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       self::$request_id,
+                       true,
+                       self::$exporter_key_last
+               );
+
+               $this->assertNotEmpty( $this->_export_data_grouped_fetched_within_callback );
+               $this->assertEmpty( get_post_meta( self::$request_id, '_export_data_grouped', true ) );
+       }
+
+       /**
+        * When mail delivery fails, the function should send a JSON error on the last page of the last exporter.
+        *
+        * @ticket 44233
+        */
+       public function test_send_error_on_last_page_of_last_exporter_when_mail_delivery_fails() {
+               // Cause `wp_mail()` to return false, to simulate mail delivery failure. Filter removed in tearDown.
+               add_filter( 'wp_mail_from', '__return_empty_string' );
+
+               // Process data, given the last exporter, on the last page and send as email.
+               $this->_setup_expected_failure( '{"success":false,"data":"Unable to send personal data export email."}' );
+
+               wp_privacy_process_personal_data_export_page(
+                       self::$response_last_page,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       self::$request_id,
+                       true,
+                       self::$exporter_key_last
+               );
+       }
+
+       /**
+        * The function should return the response, containing the export file URL, when not sent as email
+        * for the last exporter on the last page.
+        *
+        * @ticket 44233
+        */
+       public function test_return_response_with_export_file_url_when_not_sent_as_email_for_last_exporter_on_last_page() {
+               update_post_meta( self::$request_id, '_export_file_url', self::$export_file_url );
+
+               // Process data, given the last exporter, on the last page and not send as email.
+               $actual_response = wp_privacy_process_personal_data_export_page(
+                       self::$response_last_page,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       self::$request_id,
+                       false,
+                       self::$exporter_key_last
+               );
+
+               $this->assertArrayHasKey( 'url', $actual_response );
+               $this->assertSame( self::$export_file_url, $actual_response['url'] );
+               $this->assertSame( self::$response_last_page['done'], $actual_response['done'] );
+               $this->assertSame( self::$response_last_page['data'], $actual_response['data'] );
+       }
+
+       /**
+        * The function should return the response, not containing the export file URL, when sent as email
+        * for the last exporter on the last page.
+        *
+        * @ticket 44233
+        */
+       public function test_return_response_without_export_file_url_when_sent_as_email_for_last_exporter_on_last_page() {
+               update_post_meta( self::$request_id, '_export_file_url', self::$export_file_url );
+
+               // Process data, given the last exporter, on the last page and send as email.
+               $actual_response = wp_privacy_process_personal_data_export_page(
+                       self::$response_last_page,
+                       self::$exporter_index_last,
+                       self::$requester_email,
+                       self::$page_index_last,
+                       self::$request_id,
+                       true,
+                       self::$exporter_key_last
+               );
+
+               $this->assertArrayNotHasKey( 'url', $actual_response );
+               $this->assertSame( self::$response_last_page['done'], $actual_response['done'] );
+               $this->assertSame( self::$response_last_page['data'], $actual_response['data'] );
+       }
+
+       /**
+        * Test that request statuses are properly transitioned.
+        *
+        * @ticket 44233
+        *
+        * @dataProvider data_export_page_status_transitions
+        *
+        * @param string $expected_status The expected post status after calling the function.
+        * @param string $response_page   The exporter page to pass. Options are 'first' and 'last'. Default 'first'.
+        * @param string $exporter_index  The exporter index to pass. Options are 'first' and 'last'. Default 'first'.
+        * @param string $page_index      The page index to pass. Options are 'first' and 'last'. Default 'first'.
+        * @param bool   $send_as_email   If the response should be sent as an email.
+        * @param string $exporter_key    The slug (key) of the exporter to pass.
+        */
+       public function test_request_status_transitions_correctly( $expected_status, $response_page, $exporter_index, $page_index, $send_as_email, $exporter_key ) {
+               if ( 'first' === $response_page ) {
+                       $response_page = self::$response_first_page;
+               } else {
+                       $response_page = self::$response_last_page;
+               }
+
+               if ( 'first' === $exporter_index ) {
+                       $exporter_index = self::$exporter_index_first;
+               } else {
+                       $exporter_index = self::$exporter_index_last;
+               }
+
+               if ( 'first' === $page_index ) {
+                       $page_index = self::$page_index_first;
+               } else {
+                       $page_index = self::$page_index_last;
+               }
+
+               if ( 'first' === $exporter_key ) {
+                       $exporter_key = self::$exporter_key_first;
+               } else {
+                       $exporter_key = self::$exporter_key_last;
+               }
+
+               wp_privacy_process_personal_data_export_page(
+                       $response_page,
+                       $exporter_index,
+                       self::$requester_email,
+                       $page_index,
+                       self::$request_id,
+                       $send_as_email,
+                       $exporter_key
+               );
+
+               $this->assertSame( $expected_status, get_post_status( self::$request_id ) );
+       }
+
+       /**
+        * Provide test cases for `test_wp_privacy_process_personal_data_export_page()`.
+        *
+        * @since 5.2.0
+        *
+        * @return array {
+        *     @type array {
+        *         @string string $expected_status The expected post status after calling the function.
+        *         @string string $response_page   The exporter page to pass. Options are 'first' and 'last'. Default 'first'.
+        *         @string string $exporter_index  The exporter index to pass. Options are 'first' and 'last'. Default 'first'.
+        *         @string string $page_index      The page index to pass. Options are 'first' and 'last'. Default 'first'.
+        *         @bool   bool   $send_as_email   If the response should be sent as an email.
+        *         @string string $exporter_key    The slug (key) of the exporter to pass.
+        *     }
+        * }
+        */
+       public function data_export_page_status_transitions() {
+               return array(
+                       // Mark the request as completed for the last exporter on the last page, with and without email.
+                       array(
+                               'request-completed',
+                               'last',
+                               'last',
+                               'last',
+                               true,
+                               'last',
+                       ),
+                       array(
+                               'request-completed',
+                               'last',
+                               'last',
+                               'last',
+                               false,
+                               'last',
+                       ),
+                       // Leave the request as pending when not the last exporter and not on the last page.
+                       array(
+                               'request-pending',
+                               'first',
+                               'first',
+                               'first',
+                               true,
+                               'first',
+                       ),
+                       array(
+                               'request-pending',
+                               'first',
+                               'first',
+                               'first',
+                               false,
+                               'first',
+                       ),
+                       // Leave the request as pending when last exporter and not on the last page.
+                       array(
+                               'request-pending',
+                               'first',
+                               'last',
+                               'first',
+                               true,
+                               'last',
+                       ),
+                       array(
+                               'request-pending',
+                               'first',
+                               'last',
+                               'first',
+                               false,
+                               'last',
+                       ),
+                       // Leave the request as pending when not last exporter on the last page.
+                       array(
+                               'request-pending',
+                               'last',
+                               'first',
+                               'last',
+                               true,
+                               'last',
+                       ),
+                       array(
+                               'request-pending',
+                               'last',
+                               'first',
+                               'last',
+                               false,
+                               'first',
+                       ),
+               );
+       }
+
+       /**
+        * A callback for the `wp_privacy_personal_data_export_file` action that stores the
+        * `_export_data_grouped` meta data locally for testing.
+        *
+        * @since 5.2.0
+        *
+        * @param int $request_id Request ID.
+        */
+       public function action_callback_to_get_export_groups_data( $request_id ) {
+               $this->_export_data_grouped_fetched_within_callback = get_post_meta( $request_id, '_export_data_grouped', true );
+       }
+}
</ins><span class="cx" style="display: block; padding: 0 10px">Property changes on: trunk/tests/phpunit/tests/privacy/wpPrivacyProcessPersonalDataExportPage.php
</span><span class="cx" style="display: block; padding: 0 10px">___________________________________________________________________
</span></span></pre></div>
<a id="svnexecutable"></a>
<div class="addfile"><h4 style="background-color: #eee; color: inherit; margin: 1em 0; padding: 1.3em; font-size: 115%">Added: svn:executable</h4></div>
<ins style="background-color: #dfd; text-decoration:none; display:block; padding: 0 10px">+*
</ins><span class="cx" style="display: block; padding: 0 10px">\ No newline at end of property
</span></div>

</body>
</html>