public

alexkingorg / wp-test-post-slash.php

Created 2012-09-20

<?php
 
/**
 * @group post
 * @group slashes
 */
class Tests_Post_Slashes extends WP_UnitTestCase {
	function setUp() {
		parent::setUp();
		$this->author_id = $this->factory->user->create( array( 'role' => 'editor' ) );
		$this->old_current_user = get_current_user_id();
		wp_set_current_user( $this->author_id );
		
		// it is important to test with both even and odd numbered slashes as
		// kses does a strip-then-add slashes in some of it's function calls
		$this->slash_1 = 'String with 1 slash \\';
		$this->slash_2 = 'String with 2 slashes \\\\';
		$this->slash_3 = 'String with 3 slashes \\\\\\';
		$this->slash_4 = 'String with 4 slashes \\\\\\\\';
		$this->slash_5 = 'String with 5 slashes \\\\\\\\\\';
		$this->slash_6 = 'String with 6 slashes \\\\\\\\\\\\';
		$this->slash_7 = 'String with 7 slashes \\\\\\\\\\\\\\';
	}
 
	function tearDown() {
		wp_set_current_user( $this->old_current_user );
		parent::tearDown();
	}
 
	/**
	 * Tests the controller function that expects slashed data
	 *
	 */
	function test_edit_post() {
		$id = $this->factory->post->create();
 
		$_POST = array();
		$_POST['post_ID'] = $id;
		$_POST['post_title'] = $this->slash_1;
		$_POST['content'] = $this->slash_5;
		$_POST['excerpt'] = $this->slash_7;
		$_POST = add_magic_quotes( $_POST ); // the edit_post() function will strip slashes
 
		$post_id = edit_post();
		$post = get_post( $post_id );
 
		$this->assertEquals( $this->slash_1, $post->post_title );
		$this->assertEquals( $this->slash_5, $post->post_content );
		$this->assertEquals( $this->slash_7, $post->post_excerpt );
 
		$_POST = array();
		$_POST['post_ID'] = $id;
		$_POST['post_title'] = $this->slash_2;
		$_POST['content'] = $this->slash_4;
		$_POST['excerpt'] = $this->slash_6;
		$_POST = add_magic_quotes( $_POST );
 
		$post_id = edit_post();
		$post = get_post( $post_id );
 
		$this->assertEquals( $this->slash_2, $post->post_title );
		$this->assertEquals( $this->slash_4, $post->post_content );
		$this->assertEquals( $this->slash_6, $post->post_excerpt );
	}
 
	/**
	 * Tests the model function that expects un-slashed data
	 *
	 */
	function test_wp_insert_post() {
		$id = wp_insert_post(array(
			'post_status' => 'publish',
			'post_title' => $this->slash_1,
			'post_content' => $this->slash_3,
			'post_excerpt' => $this->slash_5,
			'post_type' => 'post'
		));
		$post = get_post( $id );
 
		$this->assertEquals( $this->slash_1, $post->post_title );
		$this->assertEquals( $this->slash_3, $post->post_content );
		$this->assertEquals( $this->slash_5, $post->post_excerpt );
 
		$id = wp_insert_post(array(
			'post_status' => 'publish',
			'post_title' => $this->slash_2,
			'post_content' => $this->slash_4,
			'post_excerpt' => $this->slash_6,
			'post_type' => 'post'
		));
		$post = get_post( $id );
 
		$this->assertEquals( $this->slash_2, $post->post_title );
		$this->assertEquals( $this->slash_4, $post->post_content );
		$this->assertEquals( $this->slash_6, $post->post_excerpt );
	}
 
	/**
	 * Tests the model function that expects un-slashed data
	 *
	 */
	function test_wp_update_post() {
		$id = $this->factory->post->create();
	
		wp_update_post(array(
			'ID' => $id,
			'post_title' => $this->slash_1,
			'post_content' => $this->slash_3,
			'post_excerpt' => $this->slash_5,
		));
		$post = get_post( $id );
 
		$this->assertEquals( $this->slash_1, $post->post_title );
		$this->assertEquals( $this->slash_3, $post->post_content );
		$this->assertEquals( $this->slash_5, $post->post_excerpt );
 
		wp_update_post(array(
			'ID' => $id,
			'post_title' => $this->slash_2,
			'post_content' => $this->slash_4,
			'post_excerpt' => $this->slash_6,
		));
		$post = get_post( $id );
 
		$this->assertEquals( $this->slash_2, $post->post_title );
		$this->assertEquals( $this->slash_4, $post->post_content );
		$this->assertEquals( $this->slash_6, $post->post_excerpt );
	}
 
}