tests/HTMLPurifier/AttrCollectionsTest.php

   1 <?php
   2
   3 Mock::generatePartial(
   4     'HTMLPurifier_AttrCollections',
   5     'HTMLPurifier_AttrCollections_TestForConstruct',
   6     array('performInclusions', 'expandIdentifiers')
   7 );
   8
   9 class HTMLPurifier_AttrCollectionsTest extends HTMLPurifier_Harness
  10 {
  11
  12     function testConstruction() {
  13
  14         generate_mock_once('HTMLPurifier_AttrTypes');
  15
  16         $collections = new HTMLPurifier_AttrCollections_TestForConstruct();
  17
  18         $types = new HTMLPurifier_AttrTypesMock();
  19
  20         $modules = array();
  21
  22         $modules['Module1'] = new HTMLPurifier_HTMLModule();
  23         $modules['Module1']->attr_collections = array(
  24             'Core' => array(
  25                 0 => array('Soup', 'Undefined'),
  26                 'attribute' => 'Type',
  27                 'attribute-2' => 'Type2',
  28             ),
  29             'Soup' => array(
  30                 'attribute-3' => 'Type3-old' // overwritten
  31             )
  32         );
  33
  34         $modules['Module2'] = new HTMLPurifier_HTMLModule();
  35         $modules['Module2']->attr_collections = array(
  36             'Core' => array(
  37                 0 => array('Brocolli')
  38             ),
  39             'Soup' => array(
  40                 'attribute-3' => 'Type3'
  41             ),
  42             'Brocolli' => array()
  43         );
  44
  45         $collections->__construct($types, $modules);
  46         // this is without identifier expansion or inclusions
  47         $this->assertIdentical(
  48             $collections->info,
  49             array(
  50                 'Core' => array(
  51                     0 => array('Soup', 'Undefined', 'Brocolli'),
  52                     'attribute' => 'Type',
  53                     'attribute-2' => 'Type2'
  54                 ),
  55                 'Soup' => array(
  56                     'attribute-3' => 'Type3'
  57                 ),
  58                 'Brocolli' => array()
  59             )
  60         );
  61
  62     }
  63
  64     function test_performInclusions() {
  65
  66         generate_mock_once('HTMLPurifier_AttrTypes');
  67
  68         $types = new HTMLPurifier_AttrTypesMock();
  69         $collections = new HTMLPurifier_AttrCollections($types, array());
  70         $collections->info = array(
  71             'Core' => array(0 => array('Inclusion', 'Undefined'), 'attr-original' => 'Type'),
  72             'Inclusion' => array(0 => array('SubInclusion'), 'attr' => 'Type'),
  73             'SubInclusion' => array('attr2' => 'Type')
  74         );
  75
  76         $collections->performInclusions($collections->info['Core']);
  77         $this->assertIdentical(
  78             $collections->info['Core'],
  79             array(
  80                 'attr-original' => 'Type',
  81                 'attr' => 'Type',
  82                 'attr2' => 'Type'
  83             )
  84         );
  85
  86         // test recursive
  87         $collections->info = array(
  88             'One' => array(0 => array('Two'), 'one' => 'Type'),
  89             'Two' => array(0 => array('One'), 'two' => 'Type')
  90         );
  91         $collections->performInclusions($collections->info['One']);
  92         $this->assertIdentical(
  93             $collections->info['One'],
  94             array(
  95                 'one' => 'Type',
  96                 'two' => 'Type'
  97             )
  98         );
  99
 100     }
 101
 102     function test_expandIdentifiers() {
 103
 104         generate_mock_once('HTMLPurifier_AttrTypes');
 105
 106         $types = new HTMLPurifier_AttrTypesMock();
 107         $collections = new HTMLPurifier_AttrCollections($types, array());
 108
 109         $attr = array(
 110             'attr1' => 'Color',
 111             'attr2*' => 'URI'
 112         );
 113         $c_object = new HTMLPurifier_AttrDef_HTML_Color();
 114         $u_object = new HTMLPurifier_AttrDef_URI();
 115
 116         $types->setReturnValue('get', $c_object, array('Color'));
 117         $types->setReturnValue('get', $u_object, array('URI'));
 118
 119         $collections->expandIdentifiers($attr, $types);
 120
 121         $u_object->required = true;
 122         $this->assertIdentical(
 123             $attr,
 124             array(
 125                 'attr1' => $c_object,
 126                 'attr2' => $u_object
 127             )
 128         );
 129
 130     }
 131
 132 }
 133