mirror of
https://github.com/ezyang/htmlpurifier.git
synced 2024-12-23 08:51:53 +00:00
b267b0c202
git-svn-id: http://htmlpurifier.org/svnroot/htmlpurifier/trunk@158 48356398-32a2-884e-a903-53898d9a118a
253 lines
9.8 KiB
PHP
253 lines
9.8 KiB
PHP
<?php
|
|
|
|
require_once 'HTMLPurifier/Lexer/DirectLex.php';
|
|
require_once 'HTMLPurifier/Lexer/PEARSax3.php';
|
|
|
|
class HTMLPurifier_LexerTest extends UnitTestCase
|
|
{
|
|
|
|
var $Lexer;
|
|
var $DirectLex, $PEARSax3, $DOMLex;
|
|
var $_entity_lookup;
|
|
var $_has_dom;
|
|
|
|
function setUp() {
|
|
$this->Lexer = new HTMLPurifier_Lexer();
|
|
|
|
$this->DirectLex = new HTMLPurifier_Lexer_DirectLex();
|
|
$this->PEARSax3 = new HTMLPurifier_Lexer_PEARSax3();
|
|
|
|
$this->_has_dom = version_compare(PHP_VERSION, '5', '>=');
|
|
if ($this->_has_dom) {
|
|
require_once 'HTMLPurifier/Lexer/DOMLex.php';
|
|
$this->DOMLex = new HTMLPurifier_Lexer_DOMLex();
|
|
}
|
|
|
|
$this->_entity_lookup = HTMLPurifier_EntityLookup::instance();
|
|
|
|
}
|
|
|
|
function test_substituteNonSpecialEntities() {
|
|
$char_theta = $this->_entity_lookup->table['theta'];
|
|
$this->assertIdentical($char_theta,
|
|
$this->Lexer->substituteNonSpecialEntities('θ') );
|
|
$this->assertIdentical('"',
|
|
$this->Lexer->substituteNonSpecialEntities('"') );
|
|
}
|
|
|
|
function test_tokenizeHTML() {
|
|
|
|
$input = array();
|
|
$expect = array();
|
|
$sax_expect = array();
|
|
|
|
$input[0] = '';
|
|
$expect[0] = array();
|
|
|
|
$input[1] = 'This is regular text.';
|
|
$expect[1] = array(
|
|
new HTMLPurifier_Token_Text('This is regular text.')
|
|
);
|
|
|
|
$input[2] = 'This is <b>bold</b> text';
|
|
$expect[2] = array(
|
|
new HTMLPurifier_Token_Text('This is ')
|
|
,new HTMLPurifier_Token_Start('b', array())
|
|
,new HTMLPurifier_Token_Text('bold')
|
|
,new HTMLPurifier_Token_End('b')
|
|
,new HTMLPurifier_Token_Text(' text')
|
|
);
|
|
|
|
$input[3] = '<DIV>Totally rad dude. <b>asdf</b></div>';
|
|
$expect[3] = array(
|
|
new HTMLPurifier_Token_Start('DIV', array())
|
|
,new HTMLPurifier_Token_Text('Totally rad dude. ')
|
|
,new HTMLPurifier_Token_Start('b', array())
|
|
,new HTMLPurifier_Token_Text('asdf')
|
|
,new HTMLPurifier_Token_End('b')
|
|
,new HTMLPurifier_Token_End('div')
|
|
);
|
|
|
|
// [XML-INVALID]
|
|
$input[4] = '<asdf></asdf><d></d><poOloka><poolasdf><ds></asdf></ASDF>';
|
|
$expect[4] = array(
|
|
new HTMLPurifier_Token_Start('asdf')
|
|
,new HTMLPurifier_Token_End('asdf')
|
|
,new HTMLPurifier_Token_Start('d')
|
|
,new HTMLPurifier_Token_End('d')
|
|
,new HTMLPurifier_Token_Start('poOloka')
|
|
,new HTMLPurifier_Token_Start('poolasdf')
|
|
,new HTMLPurifier_Token_Start('ds')
|
|
,new HTMLPurifier_Token_End('asdf')
|
|
,new HTMLPurifier_Token_End('ASDF')
|
|
);
|
|
// DOM is different because it condenses empty tags into REAL empty ones
|
|
// as well as makes it well-formed
|
|
$dom_expect[4] = array(
|
|
new HTMLPurifier_Token_Empty('asdf')
|
|
,new HTMLPurifier_Token_Empty('d')
|
|
,new HTMLPurifier_Token_Start('pooloka')
|
|
,new HTMLPurifier_Token_Start('poolasdf')
|
|
,new HTMLPurifier_Token_Empty('ds')
|
|
,new HTMLPurifier_Token_End('poolasdf')
|
|
,new HTMLPurifier_Token_End('pooloka')
|
|
);
|
|
|
|
$input[5] = '<a'."\t".'href="foobar.php"'."\n".'title="foo!">Link to <b id="asdf">foobar</b></a>';
|
|
$expect[5] = array(
|
|
new HTMLPurifier_Token_Start('a',array('href'=>'foobar.php','title'=>'foo!'))
|
|
,new HTMLPurifier_Token_Text('Link to ')
|
|
,new HTMLPurifier_Token_Start('b',array('id'=>'asdf'))
|
|
,new HTMLPurifier_Token_Text('foobar')
|
|
,new HTMLPurifier_Token_End('b')
|
|
,new HTMLPurifier_Token_End('a')
|
|
);
|
|
|
|
$input[6] = '<br />';
|
|
$expect[6] = array(
|
|
new HTMLPurifier_Token_Empty('br')
|
|
);
|
|
|
|
// [SGML-INVALID] [RECOVERABLE]
|
|
$input[7] = '<!-- Comment --> <!-- not so well formed --->';
|
|
$expect[7] = array(
|
|
new HTMLPurifier_Token_Comment(' Comment ')
|
|
,new HTMLPurifier_Token_Text(' ')
|
|
,new HTMLPurifier_Token_Comment(' not so well formed -')
|
|
);
|
|
$sax_expect[7] = false; // we need to figure out proper comment output
|
|
|
|
// [SGML-INVALID]
|
|
$input[8] = '<a href=""';
|
|
$expect[8] = array(
|
|
new HTMLPurifier_Token_Text('<a href=""')
|
|
);
|
|
// SAX parses it into a tag
|
|
$sax_expect[8] = array(
|
|
new HTMLPurifier_Token_Start('a', array('href'=>''))
|
|
);
|
|
// DOM parses it into an empty tag
|
|
$dom_expect[8] = array(
|
|
new HTMLPurifier_Token_Empty('a', array('href'=>''))
|
|
);
|
|
|
|
$input[9] = '<b>';
|
|
$expect[9] = array(
|
|
new HTMLPurifier_Token_Text('<b>')
|
|
);
|
|
$sax_expect[9] = array(
|
|
new HTMLPurifier_Token_Text('<')
|
|
,new HTMLPurifier_Token_Text('b')
|
|
,new HTMLPurifier_Token_Text('>')
|
|
);
|
|
// note that SAX can clump text nodes together. We won't be
|
|
// too picky though
|
|
|
|
// [SGML-INVALID]
|
|
$input[10] = '<a "=>';
|
|
// We barf on this, aim for no attributes
|
|
$expect[10] = array(
|
|
new HTMLPurifier_Token_Start('a', array('"' => ''))
|
|
);
|
|
// DOM correctly has no attributes, but also closes the tag
|
|
$dom_expect[10] = array(
|
|
new HTMLPurifier_Token_Empty('a')
|
|
);
|
|
// SAX barfs on this
|
|
$sax_expect[10] = array(
|
|
new HTMLPurifier_Token_Start('a', array('"' => ''))
|
|
);
|
|
|
|
// [INVALID] [RECOVERABLE]
|
|
$input[11] = '"';
|
|
$expect[11] = array( new HTMLPurifier_Token_Text('"') );
|
|
|
|
// compare with this valid one:
|
|
$input[12] = '"';
|
|
$expect[12] = array( new HTMLPurifier_Token_Text('"') );
|
|
$sax_expect[12] = false; // choked!
|
|
|
|
// CDATA sections!
|
|
$input[13] = '<![CDATA[You <b>can't</b> get me!]]>';
|
|
$expect[13] = array( new HTMLPurifier_Token_Text(
|
|
'You <b>can't</b> get me!' // raw
|
|
) );
|
|
$sax_expect[13] = array( // SAX has a seperate call for each entity
|
|
new HTMLPurifier_Token_Text('You '),
|
|
new HTMLPurifier_Token_Text('<'),
|
|
new HTMLPurifier_Token_Text('b'),
|
|
new HTMLPurifier_Token_Text('>'),
|
|
new HTMLPurifier_Token_Text('can'),
|
|
new HTMLPurifier_Token_Text('&'),
|
|
new HTMLPurifier_Token_Text('#39;t'),
|
|
new HTMLPurifier_Token_Text('<'),
|
|
new HTMLPurifier_Token_Text('/b'),
|
|
new HTMLPurifier_Token_Text('>'),
|
|
new HTMLPurifier_Token_Text(' get me!')
|
|
);
|
|
|
|
$char_theta = $this->_entity_lookup->table['theta'];
|
|
$char_rarr = $this->_entity_lookup->table['rarr'];
|
|
|
|
// test entity replacement
|
|
$input[14] = 'θ';
|
|
$expect[14] = array( new HTMLPurifier_Token_Text($char_theta) );
|
|
|
|
// test that entities aren't replaced in CDATA sections
|
|
$input[15] = 'θ <![CDATA[→]]>';
|
|
$expect[15] = array( new HTMLPurifier_Token_Text($char_theta . ' →') );
|
|
$sax_expect[15] = array(
|
|
new HTMLPurifier_Token_Text($char_theta . ' '),
|
|
new HTMLPurifier_Token_Text('&'),
|
|
new HTMLPurifier_Token_Text('rarr;')
|
|
);
|
|
|
|
// test entity resolution in attributes
|
|
$input[16] = '<a href="index.php?title=foo&id=bar">Link</a>';
|
|
$expect[16] = array(
|
|
new HTMLPurifier_Token_Start('a',array('href' => 'index.php?title=foo&id=bar'))
|
|
,new HTMLPurifier_Token_Text('Link')
|
|
,new HTMLPurifier_Token_End('a')
|
|
);
|
|
$sax_expect[16] = false; // PEARSax doesn't support it!
|
|
|
|
foreach($input as $i => $discard) {
|
|
$result = $this->DirectLex->tokenizeHTML($input[$i]);
|
|
$this->assertEqual($expect[$i], $result, 'DirectLexTest '.$i.': %s');
|
|
paintIf($result, $expect[$i] != $result);
|
|
|
|
// assert unless I say otherwise
|
|
$sax_result = $this->PEARSax3->tokenizeHTML($input[$i]);
|
|
if (!isset($sax_expect[$i])) {
|
|
// by default, assert with normal result
|
|
$this->assertEqual($expect[$i], $sax_result, 'PEARSax3Test '.$i.': %s');
|
|
paintIf($sax_result, $expect[$i] != $sax_result);
|
|
} elseif ($sax_expect[$i] === false) {
|
|
// assertions were turned off, optionally dump
|
|
// paintIf($sax_expect, $i == NUMBER);
|
|
} else {
|
|
// match with a custom SAX result array
|
|
$this->assertEqual($sax_expect[$i], $sax_result, 'PEARSax3Test (custom) '.$i.': %s');
|
|
paintIf($sax_result, $sax_expect[$i] != $sax_result);
|
|
}
|
|
if ($this->_has_dom) {
|
|
$dom_result = $this->DOMLex->tokenizeHTML($input[$i]);
|
|
// same structure as SAX
|
|
if (!isset($dom_expect[$i])) {
|
|
$this->assertEqual($expect[$i], $dom_result, 'DOMLexTest '.$i.': %s');
|
|
paintIf($dom_result, $expect[$i] != $dom_result);
|
|
} elseif ($dom_expect[$i] === false) {
|
|
// paintIf($dom_result, $i == NUMBER);
|
|
} else {
|
|
$this->assertEqual($dom_expect[$i], $dom_result, 'DOMLexTest (custom) '.$i.': %s');
|
|
paintIf($dom_result, $dom_expect[$i] != $dom_result);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
?>
|