tests/HTMLPurifier/HTMLModuleTest.php

   1 <?php
   2
   3 class HTMLPurifier_HTMLModuleTest extends HTMLPurifier_Harness
   4 {
   5
   6     function test_addElementToContentSet() {
   7
   8         $module = new HTMLPurifier_HTMLModule();
   9
  10         $module->addElementToContentSet('b', 'Inline');
  11         $this->assertIdentical($module->content_sets, array('Inline' => 'b'));
  12
  13         $module->addElementToContentSet('i', 'Inline');
  14         $this->assertIdentical($module->content_sets, array('Inline' => 'b | i'));
  15
  16     }
  17
  18     function test_addElement() {
  19
  20         $module = new HTMLPurifier_HTMLModule();
  21         $def = $module->addElement(
  22             'a', 'Inline', 'Optional: #PCDATA', array('Common'),
  23             array(
  24                 'href' => 'URI'
  25             )
  26         );
  27
  28         $module2 = new HTMLPurifier_HTMLModule();
  29         $def2 = new HTMLPurifier_ElementDef();
  30         $def2->content_model = '#PCDATA';
  31         $def2->content_model_type = 'optional';
  32         $def2->attr = array(
  33             'href' => 'URI',
  34             0 => array('Common')
  35         );
  36         $module2->info['a'] = $def2;
  37         $module2->elements = array('a');
  38         $module2->content_sets['Inline'] = 'a';
  39
  40         $this->assertIdentical($module, $module2);
  41         $this->assertIdentical($def, $def2);
  42         $this->assertReference($def, $module->info['a']);
  43
  44     }
  45
  46     function test_parseContents() {
  47
  48         $module = new HTMLPurifier_HTMLModule();
  49
  50         // pre-defined templates
  51         $this->assertIdentical(
  52             $module->parseContents('Inline'),
  53             array('optional', 'Inline | #PCDATA')
  54         );
  55         $this->assertIdentical(
  56             $module->parseContents('Flow'),
  57             array('optional', 'Flow | #PCDATA')
  58         );
  59         $this->assertIdentical(
  60             $module->parseContents('Empty'),
  61             array('empty', '')
  62         );
  63
  64         // normalization procedures
  65         $this->assertIdentical(
  66             $module->parseContents('optional: a'),
  67             array('optional', 'a')
  68         );
  69         $this->assertIdentical(
  70             $module->parseContents('OPTIONAL :a'),
  71             array('optional', 'a')
  72         );
  73         $this->assertIdentical(
  74             $module->parseContents('Optional: a'),
  75             array('optional', 'a')
  76         );
  77
  78         // others
  79         $this->assertIdentical(
  80             $module->parseContents('Optional: a | b | c'),
  81             array('optional', 'a | b | c')
  82         );
  83
  84         // object pass-through
  85         generate_mock_once('HTMLPurifier_AttrDef');
  86         $this->assertIdentical(
  87             $module->parseContents(new HTMLPurifier_AttrDefMock()),
  88             array(null, null)
  89         );
  90
  91     }
  92
  93     function test_mergeInAttrIncludes() {
  94
  95         $module = new HTMLPurifier_HTMLModule();
  96
  97         $attr = array();
  98         $module->mergeInAttrIncludes($attr, 'Common');
  99         $this->assertIdentical($attr, array(0 => array('Common')));
 100
 101         $attr = array('a' => 'b');
 102         $module->mergeInAttrIncludes($attr, array('Common', 'Good'));
 103         $this->assertIdentical($attr, array('a' => 'b', 0 => array('Common', 'Good')));
 104
 105     }
 106
 107     function test_addBlankElement() {
 108
 109         $module = new HTMLPurifier_HTMLModule();
 110         $def = $module->addBlankElement('a');
 111
 112         $def2 = new HTMLPurifier_ElementDef();
 113         $def2->standalone = false;
 114
 115         $this->assertReference($module->info['a'], $def);
 116         $this->assertIdentical($def, $def2);
 117
 118     }
 119
 120     function test_makeLookup() {
 121
 122         $module = new HTMLPurifier_HTMLModule();
 123
 124         $this->assertIdentical(
 125             $module->makeLookup('foo'),
 126             array('foo' => true)
 127         );
 128         $this->assertIdentical(
 129             $module->makeLookup(array('foo')),
 130             array('foo' => true)
 131         );
 132
 133         $this->assertIdentical(
 134             $module->makeLookup('foo', 'two'),
 135             array('foo' => true, 'two' => true)
 136         );
 137         $this->assertIdentical(
 138             $module->makeLookup(array('foo', 'two')),
 139             array('foo' => true, 'two' => true)
 140         );
 141
 142     }
 143
 144 }
 145