tests/HTMLPurifier/HTMLModuleManagerTest.php

   1 <?php
   2
   3 class HTMLPurifier_HTMLModuleManagerTest extends HTMLPurifier_Harness
   4 {
   5
   6     protected function createManager() {
   7         $manager = new HTMLPurifier_HTMLModuleManager();
   8
   9         $this->config->set('HTML', 'CustomDoctype', 'Blank');
  10         $manager->doctypes->register('Blank');
  11
  12         $attrdef_nmtokens = new HTMLPurifier_AttrDef_HTML_Nmtokens();
  13
  14         generate_mock_once('HTMLPurifier_AttrDef');
  15         $attrdef = new HTMLPurifier_AttrDefMock();
  16         $attrdef->setReturnValue('make', $attrdef_nmtokens);
  17         $manager->attrTypes->set('NMTOKENS', $attrdef);
  18         return $manager;
  19     }
  20
  21     function test_addModule() {
  22
  23         $manager = $this->createManager();
  24
  25         // ...but we add user modules
  26
  27         $common_module = new HTMLPurifier_HTMLModule();
  28         $common_module->name = 'Common';
  29         $common_module->attr_collections['Common'] = array('class' => 'NMTOKENS');
  30         $common_module->content_sets['Flow'] = 'Block | Inline';
  31         $manager->addModule($common_module);
  32
  33         $structural_module = new HTMLPurifier_HTMLModule();
  34         $structural_module->name = 'Structural';
  35         $structural_module->addElement('p', 'Block', 'Inline', 'Common');
  36         $manager->addModule($structural_module);
  37
  38         $formatting_module = new HTMLPurifier_HTMLModule();
  39         $formatting_module->name = 'Formatting';
  40         $formatting_module->addElement('em', 'Inline', 'Inline', 'Common');
  41         $manager->addModule($formatting_module);
  42
  43         $unsafe_module = new HTMLPurifier_HTMLModule();
  44         $unsafe_module->name = 'Unsafe';
  45         $unsafe_module->safe = false;
  46         $unsafe_module->addElement('div', 'Block', 'Flow');
  47         $manager->addModule($unsafe_module);
  48
  49         $config = HTMLPurifier_Config::createDefault();
  50         $config->set('HTML', 'Trusted', false);
  51         $config->set('HTML', 'CustomDoctype', 'Blank');
  52
  53         $manager->setup($config);
  54
  55         $attrdef_nmtokens = new HTMLPurifier_AttrDef_HTML_Nmtokens();
  56
  57         $p = new HTMLPurifier_ElementDef();
  58         $p->attr['class'] = $attrdef_nmtokens;
  59         $p->child = new HTMLPurifier_ChildDef_Optional(array('em', '#PCDATA'));
  60         $p->content_model = 'em | #PCDATA';
  61         $p->content_model_type = 'optional';
  62         $p->descendants_are_inline = true;
  63
  64         $em = new HTMLPurifier_ElementDef();
  65         $em->attr['class'] = $attrdef_nmtokens;
  66         $em->child = new HTMLPurifier_ChildDef_Optional(array('em', '#PCDATA'));
  67         $em->content_model = 'em | #PCDATA';
  68         $em->content_model_type = 'optional';
  69         $em->descendants_are_inline = true;
  70
  71         $this->assertEqual(
  72             array('p' => $p, 'em' => $em),
  73             $manager->getElements()
  74         );
  75
  76         // test trusted parameter override
  77
  78         $div = new HTMLPurifier_ElementDef();
  79         $div->child = new HTMLPurifier_ChildDef_Optional(array('p', 'div', 'em', '#PCDATA'));
  80         $div->content_model = 'p | div | em | #PCDATA';
  81         $div->content_model_type = 'optional';
  82         $div->descendants_are_inline = false;
  83
  84         $this->assertEqual($div, $manager->getElement('div', true));
  85
  86     }
  87
  88     function testAllowedModules() {
  89
  90         $manager = new HTMLPurifier_HTMLModuleManager();
  91         $manager->doctypes->register(
  92             'Fantasy Inventory 1.0', true,
  93             array('Weapons', 'Magic')
  94         );
  95
  96         // register these modules so it doesn't blow up
  97         $weapons_module = new HTMLPurifier_HTMLModule();
  98         $weapons_module->name = 'Weapons';
  99         $manager->registerModule($weapons_module);
 100
 101         $magic_module = new HTMLPurifier_HTMLModule();
 102         $magic_module->name = 'Magic';
 103         $manager->registerModule($magic_module);
 104
 105         $config = HTMLPurifier_Config::create(array(
 106             'HTML.CustomDoctype' => 'Fantasy Inventory 1.0',
 107             'HTML.AllowedModules' => 'Weapons'
 108         ));
 109         $manager->setup($config);
 110
 111         $this->assertTrue( isset($manager->modules['Weapons']));
 112         $this->assertFalse(isset($manager->modules['Magic']));
 113
 114     }
 115
 116
 117
 118 }
 119