htmlpurifier/tests/HTMLPurifier/HTMLModuleManagerTest.php

<?php

require_once 'HTMLPurifier/HTMLModuleManager.php';

class HTMLPurifier_HTMLModuleManagerTest extends UnitTestCase
{
    
    function test_addModule() {
        $manager = new HTMLPurifier_HTMLModuleManager();
        $manager->doctypes->register('Blank'); // doctype normally is blank...
        
        $attrdef_nmtokens = new HTMLPurifier_AttrDef();
        $attrdef_nmtokens->_name = 'nmtokens'; // for testing only
        
        generate_mock_once('HTMLPurifier_AttrDef');
        $attrdef = new HTMLPurifier_AttrDefMock();
        $attrdef->setReturnValue('make', $attrdef_nmtokens);
        $manager->attrTypes->info['NMTOKENS'] =& $attrdef;
        
        // ...but we add user modules
        
        $common_module = new HTMLPurifier_HTMLModule();
        $common_module->name = 'Common';
        $common_module->attr_collections['Common'] = array('class' => 'NMTOKENS');
        $common_module->content_sets['Flow'] = 'Block | Inline';
        $manager->addModule($common_module);
        
        $structural_module = new HTMLPurifier_HTMLModule();
        $structural_module->name = 'Structural';
        $structural_module->addElement('p', true, 'Block', 'Inline', 'Common');
        $structural_module->addElement('div', false, 'Block', 'Flow');
        $manager->addModule($structural_module);
        
        $formatting_module = new HTMLPurifier_HTMLModule();
        $formatting_module->name = 'Formatting';
        $formatting_module->addElement('em', true, 'Inline', 'Inline', 'Common');
        $manager->addModule($formatting_module);
        
        $config = HTMLPurifier_Config::createDefault();
        $config->set('HTML', 'Trusted', false);
        $config->set('HTML', 'CustomDoctype', 'Blank');
        
        $manager->setup($config);
        
        $p = new HTMLPurifier_ElementDef();
        $p->attr['class'] = $attrdef_nmtokens;
        $p->child = new HTMLPurifier_ChildDef_Optional(array('em', '#PCDATA'));
        $p->content_model = 'em | #PCDATA';
        $p->content_model_type = 'optional';
        $p->descendants_are_inline = true;
        $p->safe = true;
        
        $em = new HTMLPurifier_ElementDef();
        $em->attr['class'] = $attrdef_nmtokens;
        $em->child = new HTMLPurifier_ChildDef_Optional(array('em', '#PCDATA'));
        $em->content_model = 'em | #PCDATA';
        $em->content_model_type = 'optional';
        $em->descendants_are_inline = true;
        $em->safe = true;
        
        $this->assertEqual(
            array('p' => $p, 'em' => $em),
            $manager->getElements()
        );
        
        // test trusted parameter override
        
        $div = new HTMLPurifier_ElementDef();
        $div->child = new HTMLPurifier_ChildDef_Optional(array('p', 'div', 'em', '#PCDATA'));
        $div->content_model = 'p | div | em | #PCDATA';
        $div->content_model_type = 'optional';
        $div->descendants_are_inline = false;
        $div->safe = false;
        
        $this->assertEqual($div, $manager->getElement('div', true));
        
    }
    
    function testAllowedModules() {
        
        $manager = new HTMLPurifier_HTMLModuleManager();
        $manager->doctypes->register(
            'Fantasy Inventory 1.0', true,
            array('Weapons', 'Magic')
        );
        
        // register these modules so it doesn't blow up
        $weapons_module = new HTMLPurifier_HTMLModule();
        $weapons_module->name = 'Weapons';
        $manager->registerModule($weapons_module);
        
        $magic_module = new HTMLPurifier_HTMLModule();
        $magic_module->name = 'Magic';
        $manager->registerModule($magic_module);
        
        $config = HTMLPurifier_Config::create(array(
            'HTML.CustomDoctype' => 'Fantasy Inventory 1.0',
            'HTML.AllowedModules' => 'Weapons'
        ));
        $manager->setup($config);
        
        $this->assertTrue( isset($manager->modules['Weapons']));
        $this->assertFalse(isset($manager->modules['Magic']));
        
    }
    
}