wordpress-develop/tests/phpunit/tests/formatting/SanitizeFileName.php

<?php

/**
 * @group formatting
 */
class Tests_Formatting_SanitizeFileName extends WP_UnitTestCase {
	function test_munges_extensions() {
		# r17990
		$file_name = sanitize_file_name( 'test.phtml.txt' );
		$this->assertEquals( 'test.phtml_.txt', $file_name );
	}

	function test_removes_special_chars() {
		$special_chars = array( '?', '[', ']', '/', '\\', '=', '<', '>', ':', ';', ',', "'", '"', '&', '$', '#', '*', '(', ')', '|', '~', '`', '!', '{', '}', '%', '+', chr( 0 ) );
		$string        = 'test';
		foreach ( $special_chars as $char ) {
			$string .= $char;
		}
		$string .= 'test';
		$this->assertEquals( 'testtest', sanitize_file_name( $string ) );
	}

	/**
	 * Test that spaces are correctly replaced with dashes.
	 *
	 * @ticket 16330
	 */
	function test_replace_spaces() {
		$urls = array(
			'unencoded space.png'  => 'unencoded-space.png',
			'encoded-space.jpg'    => 'encoded-space.jpg',
			'plus+space.jpg'       => 'plusspace.jpg',
			'multi %20 +space.png' => 'multi-20-space.png',
		);

		foreach ( $urls as $test => $expected ) {
			$this->assertEquals( $expected, sanitize_file_name( $test ) );
		}
	}

	function test_replaces_any_number_of_hyphens_with_one_hyphen() {
		$this->assertEquals( 'a-t-t', sanitize_file_name( 'a----t----t' ) );
	}

	function test_trims_trailing_hyphens() {
		$this->assertEquals( 'a-t-t', sanitize_file_name( 'a----t----t----' ) );
	}

	function test_replaces_any_amount_of_whitespace_with_one_hyphen() {
		$this->assertEquals( 'a-t', sanitize_file_name( 'a          t' ) );
		$this->assertEquals( 'a-t', sanitize_file_name( "a    \n\n\nt" ) );
	}

	/**
	 * @ticket 16226
	 */
	function test_replaces_percent_sign() {
		$this->assertEquals( 'a22b.jpg', sanitize_file_name( 'a%22b.jpg' ) );
	}

	function test_replaces_unnammed_file_extensions() {
		// Test filenames with both supported and unsupported extensions.
		$this->assertEquals( 'unnamed-file.exe', sanitize_file_name( '_.exe' ) );
		$this->assertEquals( 'unnamed-file.jpg', sanitize_file_name( '_.jpg' ) );
	}

	function test_replaces_unnammed_file_extensionless() {
		// Test a filenames that becomes extensionless.
		$this->assertEquals( 'no-extension', sanitize_file_name( '_.no-extension' ) );
	}

	/**
	 * @dataProvider data_wp_filenames
	 */
	function test_replaces_invalid_utf8_characters( $input, $expected ) {
		$this->assertEquals( $expected, sanitize_file_name( $input ) );
	}

	function data_wp_filenames() {
		return array(
			array( urldecode( '%B1myfile.png' ), 'myfile.png' ),
			array( urldecode( '%B1myfile' ), 'myfile' ),
			array( 'demo bar.png', 'demo-bar.png' ),
			array( 'demo' . json_decode( '"\u00a0"' ) . 'bar.png', 'demo-bar.png' ),
		);
	}
}