Skip to content

Instantly share code, notes, and snippets.

@alexkingorg
Created September 20, 2012 16:33
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save alexkingorg/3756939 to your computer and use it in GitHub Desktop.
Save alexkingorg/3756939 to your computer and use it in GitHub Desktop.
WordPress unit test to verify that slashed data is saved properly for posts
<?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 );
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment