tests/HTMLPurifier/GeneratorTest.php

   1 <?php
   2
   3 require_once 'HTMLPurifier/Generator.php';
   4 require_once 'HTMLPurifier/EntityLookup.php';
   5
   6 class HTMLPurifier_GeneratorTest extends UnitTestCase
   7 {
   8
   9     var $gen;
  10     var $_entity_lookup;
  11
  12     function HTMLPurifier_GeneratorTest() {
  13         $this->UnitTestCase();
  14         $this->gen = new HTMLPurifier_Generator();
  15         $this->_entity_lookup = HTMLPurifier_EntityLookup::instance();
  16     }
  17
  18     function test_generateFromToken() {
  19
  20         $inputs = array();
  21         $expect = array();
  22         $config = array();
  23
  24         $inputs[0] = new HTMLPurifier_Token_Text('Foobar.<>');
  25         $expect[0] = 'Foobar.&lt;&gt;';
  26
  27         $inputs[1] = new HTMLPurifier_Token_Start('a',
  28                             array('href' => 'dyn?a=foo&b=bar')
  29                          );
  30         $expect[1] = '<a href="dyn?a=foo&amp;b=bar">';
  31
  32         $inputs[2] = new HTMLPurifier_Token_End('b');
  33         $expect[2] = '</b>';
  34
  35         $inputs[3] = new HTMLPurifier_Token_Empty('br',
  36                             array('style' => 'font-family:"Courier New";')
  37                          );
  38         $expect[3] = '<br style="font-family:&quot;Courier New&quot;;" />';
  39
  40         $inputs[4] = new HTMLPurifier_Token_Start('asdf');
  41         $expect[4] = '<asdf>';
  42
  43         $inputs[5] = new HTMLPurifier_Token_Empty('br');
  44         $expect[5] = '<br />';
  45
  46         // test fault tolerance
  47         $inputs[6] = null;
  48         $expect[6] = '';
  49
  50         // don't convert non-special characters
  51         $theta_char = $this->_entity_lookup->table['theta'];
  52         $inputs[7] = new HTMLPurifier_Token_Text($theta_char);
  53         $expect[7] = $theta_char;
  54
  55         $default_config = HTMLPurifier_Config::createDefault();
  56         foreach ($inputs as $i => $input) {
  57             if (!isset($config[$i])) $config[$i] = $default_config;
  58             $result = $this->gen->generateFromToken($input, $config[$i]);
  59             $this->assertEqual($result, $expect[$i]);
  60             paintIf($result, $result != $expect[$i]);
  61         }
  62
  63     }
  64
  65     function test_generateAttributes() {
  66
  67         $inputs = array();
  68         $expect = array();
  69         $config = array();
  70
  71         $inputs[0] = array();
  72         $expect[0] = '';
  73
  74         $inputs[1] = array('href' => 'dyn?a=foo&b=bar');
  75         $expect[1] = 'href="dyn?a=foo&amp;b=bar"';
  76
  77         $inputs[2] = array('style' => 'font-family:"Courier New";');
  78         $expect[2] = 'style="font-family:&quot;Courier New&quot;;"';
  79
  80         $inputs[3] = array('src' => 'picture.jpg', 'alt' => 'Short & interesting');
  81         $expect[3] = 'src="picture.jpg" alt="Short &amp; interesting"';
  82
  83         // don't escape nonspecial characters
  84         $theta_char = $this->_entity_lookup->table['theta'];
  85         $inputs[4] = array('title' => 'Theta is ' . $theta_char);
  86         $expect[4] = 'title="Theta is ' . $theta_char . '"';
  87
  88         $default_config = HTMLPurifier_Config::createDefault();
  89         foreach ($inputs as $i => $input) {
  90             if (!isset($config[$i])) $config[$i] = $default_config;
  91             $result = $this->gen->generateAttributes($input, $config[$i]);
  92             $this->assertEqual($result, $expect[$i]);
  93             paintIf($result, $result != $expect[$i]);
  94         }
  95
  96     }
  97
  98     function test_generateFromTokens() {
  99
 100         $inputs = array();
 101         $expect = array();
 102         $config = array();
 103
 104         $inputs[0] = array(
 105             new HTMLPurifier_Token_Start('b'),
 106             new HTMLPurifier_Token_Text('Foobar!'),
 107             new HTMLPurifier_Token_End('b')
 108             );
 109         $expect[0] = '<b>Foobar!</b>';
 110
 111         $inputs[1] = array();
 112         $expect[1] = '';
 113
 114         $default_config = HTMLPurifier_Config::createDefault();
 115         foreach ($inputs as $i => $input) {
 116             if (!isset($config[$i])) $config[$i] = $default_config;
 117             $result = $this->gen->generateFromTokens($input, $config[$i]);
 118             $this->assertEqual($expect[$i], $result);
 119             paintIf($result, $result != $expect[$i]);
 120         }
 121
 122
 123     }
 124
 125 }
 126
 127 ?>