Ticket #43546: 43546.wpPrivacyGeneratePersonalDataExportFile.php

Index: tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php
===================================================================
--- tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php     (nonexistent)
+++ tests/phpunit/tests/privacy/wpPrivacyGeneratePersonalDataExportFile.php     (working copy)
@@ -0,0 +1,118 @@
+<?php
+/**
+ * Define a class to test `wp_privacy_generate_personal_data_export_file()`.
+ *
+ * @package WordPress
+ * @subpackage UnitTests
+ * @since 4.9.6
+ */
+
+/**
+ * Test cases for `wp_privacy_generate_personal_data_export_file()`.
+ *
+ * @group privacy
+ * @covers wp_privacy_generate_personal_data_export_file
+ *
+ * @since 4.9.6
+ */
+class Tests_Privacy_wpPrivacyGeneratePersonalDataExportFile extends WP_UnitTestCase {
+       /**
+        * An Export Request ID
+        *
+        * @since 4.9.6
+        *
+        * @var int $export_request_id
+        */
+       protected static $export_request_id;
+
+       /**
+        * Create fixtures that are shared by multiple test cases.
+        *
+        * @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' );
+       }
+
+       /**
+        * Set up the test fixture.
+        * Override wp_die(), pretend to be ajax, and suppres E_WARNINGs
+        */
+       public function setUp() {
+               parent::setUp();
+
+               add_filter( 'wp_die_ajax_handler', array( $this, 'getDieHandler' ), 1, 1 );
+               add_filter( 'wp_doing_ajax', '__return_true' );
+
+               // 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
+        */
+       public function tearDown() {
+               remove_filter( 'wp_die_ajax_handler', array( $this, 'getDieHandler' ), 1, 1 );
+               error_reporting( $this->_error_level );
+               parent::tearDown();
+       }
+
+       /**
+        * Return our callback handler
+        *
+        * @return callback
+        */
+       public function getDieHandler() {
+               return array( $this, 'dieHandler' );
+       }
+
+       /**
+        * Handler for wp_die()
+        * Don't die, just continue on.
+        *
+        * @param string $message
+        */
+       public function dieHandler( $message ) {
+       }
+
+       // TODO test ziparchive not available
+
+       public function test_wp_privacy_generate_personal_data_export_file_rejects_remove_requests() {
+               $request_id = wp_create_user_request( 'removal-requester@example.com', 'remove_personal_data' );
+
+               ob_start();
+               wp_privacy_generate_personal_data_export_file( $request_id );
+
+               $response = json_decode( ob_get_clean() );
+               $this->assertFalse( $response->success );
+               $this->assertEquals( 'Invalid request ID when generating export file.', $response->data );
+       }
+
+       public function test_wp_privacy_generate_personal_data_export_file_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'
+                       )
+               );
+
+               ob_start();
+               wp_privacy_generate_personal_data_export_file( $request_id );
+
+               $response = json_decode( ob_get_clean() );
+               $this->assertFalse( $response->success );
+               $this->assertEquals( 'Invalid email address when generating export file.', $response->data );
+       }
+
+       // TODO mkdir_p fails
+       // TODO index.html created in export folder - silence is golden
+       // TODO unable to protect error if can't create index.html file
+       // TODO unable to open export file error if can't write
+       // TODO report file exists, has starts with <html> and ends with </html>
+       // TODO unable to add data error if can't add file to export
+       // TODO error if can't open archive file
+       // TODO report file removed, even if errors
+}