tests/HTMLPurifier/HTMLModule/TidyTest.php

   1 <?php
   2
   3 Mock::generatePartial(
   4     'HTMLPurifier_HTMLModule_Tidy',
   5     'HTMLPurifier_HTMLModule_Tidy_TestForConstruct',
   6     array('makeFixes', 'makeFixesForLevel', 'populate')
   7 );
   8
   9 class HTMLPurifier_HTMLModule_TidyTest extends HTMLPurifier_Harness
  10 {
  11
  12     public function test_getFixesForLevel()
  13     {
  14         $module = new HTMLPurifier_HTMLModule_Tidy();
  15         $module->fixesForLevel['light'][]  = 'light-fix';
  16         $module->fixesForLevel['medium'][] = 'medium-fix';
  17         $module->fixesForLevel['heavy'][]  = 'heavy-fix';
  18
  19         $this->assertIdentical(
  20             array(),
  21             $module->getFixesForLevel('none')
  22         );
  23         $this->assertIdentical(
  24             array('light-fix' => true),
  25             $module->getFixesForLevel('light')
  26         );
  27         $this->assertIdentical(
  28             array('light-fix' => true, 'medium-fix' => true),
  29             $module->getFixesForLevel('medium')
  30         );
  31         $this->assertIdentical(
  32             array('light-fix' => true, 'medium-fix' => true, 'heavy-fix' => true),
  33             $module->getFixesForLevel('heavy')
  34         );
  35
  36         $this->expectError('Tidy level turbo not recognized');
  37         $module->getFixesForLevel('turbo');
  38
  39     }
  40
  41     public function test_setup()
  42     {
  43         $i = 0; // counter, helps us isolate expectations
  44
  45         // initialize partial mock
  46         $module = new HTMLPurifier_HTMLModule_Tidy_TestForConstruct();
  47         $module->fixesForLevel['light']  = array('light-fix-1', 'light-fix-2');
  48         $module->fixesForLevel['medium'] = array('medium-fix-1', 'medium-fix-2');
  49         $module->fixesForLevel['heavy']  = array('heavy-fix-1', 'heavy-fix-2');
  50
  51         $j = 0;
  52         $fixes = array(
  53             'light-fix-1'  => $lf1 = $j++,
  54             'light-fix-2'  => $lf2 = $j++,
  55             'medium-fix-1' => $mf1 = $j++,
  56             'medium-fix-2' => $mf2 = $j++,
  57             'heavy-fix-1'  => $hf1 = $j++,
  58             'heavy-fix-2'  => $hf2 = $j++
  59         );
  60         $module->returns('makeFixes', $fixes);
  61
  62         $config = HTMLPurifier_Config::create(array(
  63             'HTML.TidyLevel' => 'none'
  64         ));
  65         $module->expectAt($i++, 'populate', array(array()));
  66         $module->setup($config);
  67
  68         // basic levels
  69
  70         $config = HTMLPurifier_Config::create(array(
  71             'HTML.TidyLevel' => 'light'
  72         ));
  73         $module->expectAt($i++, 'populate', array(array(
  74             'light-fix-1' => $lf1,
  75             'light-fix-2' => $lf2
  76         )));
  77         $module->setup($config);
  78
  79         $config = HTMLPurifier_Config::create(array(
  80             'HTML.TidyLevel' => 'heavy'
  81         ));
  82         $module->expectAt($i++, 'populate', array(array(
  83             'light-fix-1'  => $lf1,
  84             'light-fix-2'  => $lf2,
  85             'medium-fix-1' => $mf1,
  86             'medium-fix-2' => $mf2,
  87             'heavy-fix-1'  => $hf1,
  88             'heavy-fix-2'  => $hf2
  89         )));
  90         $module->setup($config);
  91
  92         // fine grained tuning
  93
  94         $config = HTMLPurifier_Config::create(array(
  95             'HTML.TidyLevel' => 'none',
  96             'HTML.TidyAdd'   => array('light-fix-1', 'medium-fix-1')
  97         ));
  98         $module->expectAt($i++, 'populate', array(array(
  99             'light-fix-1' => $lf1,
 100             'medium-fix-1' => $mf1
 101         )));
 102         $module->setup($config);
 103
 104         $config = HTMLPurifier_Config::create(array(
 105             'HTML.TidyLevel' => 'medium',
 106             'HTML.TidyRemove'   => array('light-fix-1', 'medium-fix-1')
 107         ));
 108         $module->expectAt($i++, 'populate', array(array(
 109             'light-fix-2' => $lf2,
 110             'medium-fix-2' => $mf2
 111         )));
 112         $module->setup($config);
 113
 114     }
 115
 116     public function test_makeFixesForLevel()
 117     {
 118         $module = new HTMLPurifier_HTMLModule_Tidy();
 119         $module->defaultLevel = 'heavy';
 120
 121         $module->makeFixesForLevel(array(
 122             'fix-1' => 0,
 123             'fix-2' => 1,
 124             'fix-3' => 2
 125         ));
 126
 127         $this->assertIdentical($module->fixesForLevel['heavy'], array('fix-1', 'fix-2', 'fix-3'));
 128         $this->assertIdentical($module->fixesForLevel['medium'], array());
 129         $this->assertIdentical($module->fixesForLevel['light'], array());
 130
 131     }
 132     public function test_makeFixesForLevel_undefinedLevel()
 133     {
 134         $module = new HTMLPurifier_HTMLModule_Tidy();
 135         $module->defaultLevel = 'bananas';
 136
 137         $this->expectError('Default level bananas does not exist');
 138
 139         $module->makeFixesForLevel(array(
 140             'fix-1' => 0
 141         ));
 142
 143     }
 144
 145     public function test_getFixType()
 146     {
 147         // syntax needs documenting
 148
 149         $module = new HTMLPurifier_HTMLModule_Tidy();
 150
 151         $this->assertIdentical(
 152             $module->getFixType('a'),
 153             array('tag_transform', array('element' => 'a'))
 154         );
 155
 156         $this->assertIdentical(
 157             $module->getFixType('a@href'),
 158             $reuse = array('attr_transform_pre', array('element' => 'a', 'attr' => 'href'))
 159         );
 160
 161         $this->assertIdentical(
 162             $module->getFixType('a@href#pre'),
 163             $reuse
 164         );
 165
 166         $this->assertIdentical(
 167             $module->getFixType('a@href#post'),
 168             array('attr_transform_post', array('element' => 'a', 'attr' => 'href'))
 169         );
 170
 171         $this->assertIdentical(
 172             $module->getFixType('xml:foo@xml:bar'),
 173             array('attr_transform_pre', array('element' => 'xml:foo', 'attr' => 'xml:bar'))
 174         );
 175
 176         $this->assertIdentical(
 177             $module->getFixType('blockquote#child'),
 178             array('child', array('element' => 'blockquote'))
 179         );
 180
 181         $this->assertIdentical(
 182             $module->getFixType('@lang'),
 183             array('attr_transform_pre', array('attr' => 'lang'))
 184         );
 185
 186         $this->assertIdentical(
 187             $module->getFixType('@lang#post'),
 188             array('attr_transform_post', array('attr' => 'lang'))
 189         );
 190
 191     }
 192
 193     public function test_populate()
 194     {
 195         $i = 0;
 196
 197         $module = new HTMLPurifier_HTMLModule_Tidy();
 198         $module->populate(array(
 199             'element' => $element = $i++,
 200             'element@attr' => $attr = $i++,
 201             'element@attr#post' => $attr_post = $i++,
 202             'element#child' => $child = $i++,
 203             'element#content_model_type' => $content_model_type = $i++,
 204             '@attr' => $global_attr = $i++,
 205             '@attr#post' => $global_attr_post = $i++
 206         ));
 207
 208         $module2 = new HTMLPurifier_HTMLModule_Tidy();
 209         $e = $module2->addBlankElement('element');
 210         $e->attr_transform_pre['attr'] = $attr;
 211         $e->attr_transform_post['attr'] = $attr_post;
 212         $e->child = $child;
 213         $e->content_model_type = $content_model_type;
 214         $module2->info_tag_transform['element'] = $element;
 215         $module2->info_attr_transform_pre['attr'] = $global_attr;
 216         $module2->info_attr_transform_post['attr'] = $global_attr_post;
 217
 218         $this->assertEqual($module, $module2);
 219
 220     }
 221
 222 }
 223
 224 // vim: et sw=4 sts=4