tests/HTMLPurifier/URITest.php

   1 <?php
   2
   3 class HTMLPurifier_URITest extends HTMLPurifier_URIHarness
   4 {
   5
   6     protected function createURI($uri)
   7     {
   8         $parser = new HTMLPurifier_URIParser();
   9         return $parser->parse($uri);
  10     }
  11
  12     public function test_construct()
  13     {
  14         $uri1 = new HTMLPurifier_URI('HTTP', 'bob', 'example.com', '23', '/foo', 'bar=2', 'slash');
  15         $uri2 = new HTMLPurifier_URI('http', 'bob', 'example.com',  23,  '/foo', 'bar=2', 'slash');
  16         $this->assertIdentical($uri1, $uri2);
  17     }
  18
  19     protected $oldRegistry;
  20
  21     protected function &setUpSchemeRegistryMock() {
  22         $this->oldRegistry = HTMLPurifier_URISchemeRegistry::instance();
  23         generate_mock_once('HTMLPurifier_URIScheme');
  24         generate_mock_once('HTMLPurifier_URISchemeRegistry');
  25         $registry = HTMLPurifier_URISchemeRegistry::instance(
  26           new HTMLPurifier_URISchemeRegistryMock()
  27         );
  28         return $registry;
  29     }
  30
  31     protected function setUpSchemeMock($name)
  32     {
  33         $registry = $this->setUpSchemeRegistryMock();
  34         $scheme_mock = new HTMLPurifier_URISchemeMock();
  35         $registry->returns('getScheme', $scheme_mock, array($name, '*', '*'));
  36         return $scheme_mock;
  37     }
  38
  39     protected function setUpNoValidSchemes()
  40     {
  41         $registry = $this->setUpSchemeRegistryMock();
  42         $registry->returns('getScheme', false, array('*', '*', '*'));
  43     }
  44
  45     protected function tearDownSchemeRegistryMock()
  46     {
  47         HTMLPurifier_URISchemeRegistry::instance($this->oldRegistry);
  48     }
  49
  50     public function test_getSchemeObj()
  51     {
  52         $scheme_mock = $this->setUpSchemeMock('http');
  53
  54         $uri = $this->createURI('http:');
  55         $scheme_obj = $uri->getSchemeObj($this->config, $this->context);
  56         $this->assertIdentical($scheme_obj, $scheme_mock);
  57
  58         $this->tearDownSchemeRegistryMock();
  59     }
  60
  61     public function test_getSchemeObj_invalidScheme()
  62     {
  63         $this->setUpNoValidSchemes();
  64
  65         $uri = $this->createURI('http:');
  66         $result = $uri->getSchemeObj($this->config, $this->context);
  67         $this->assertIdentical($result, false);
  68
  69         $this->tearDownSchemeRegistryMock();
  70     }
  71
  72     public function test_getSchemaObj_defaultScheme()
  73     {
  74         $scheme = 'foobar';
  75
  76         $scheme_mock = $this->setUpSchemeMock($scheme);
  77         $this->config->set('URI.DefaultScheme', $scheme);
  78
  79         $uri = $this->createURI('hmm');
  80         $scheme_obj = $uri->getSchemeObj($this->config, $this->context);
  81         $this->assertIdentical($scheme_obj, $scheme_mock);
  82
  83         $this->tearDownSchemeRegistryMock();
  84     }
  85
  86     public function test_getSchemaObj_invalidDefaultScheme()
  87     {
  88         $this->setUpNoValidSchemes();
  89         $this->config->set('URI.DefaultScheme', 'foobar');
  90
  91         $uri = $this->createURI('hmm');
  92
  93         $this->expectError('Default scheme object "foobar" was not readable');
  94         $result = $uri->getSchemeObj($this->config, $this->context);
  95         $this->assertIdentical($result, false);
  96
  97         $this->tearDownSchemeRegistryMock();
  98     }
  99
 100     protected function assertToString($expect_uri, $scheme, $userinfo, $host, $port, $path, $query, $fragment)
 101     {
 102         $uri = new HTMLPurifier_URI($scheme, $userinfo, $host, $port, $path, $query, $fragment);
 103         $string = $uri->toString();
 104         $this->assertIdentical($string, $expect_uri);
 105     }
 106
 107     public function test_toString_full()
 108     {
 109         $this->assertToString(
 110             'http://bob@example.com:300/foo?bar=baz#fragment',
 111             'http', 'bob', 'example.com', 300, '/foo', 'bar=baz', 'fragment'
 112         );
 113     }
 114
 115     public function test_toString_scheme()
 116     {
 117         $this->assertToString(
 118             'http:',
 119             'http', null, null, null, '', null, null
 120         );
 121     }
 122
 123     public function test_toString_authority()
 124     {
 125         $this->assertToString(
 126             '//bob@example.com:8080',
 127             null, 'bob', 'example.com', 8080, '', null, null
 128         );
 129     }
 130
 131     public function test_toString_path()
 132     {
 133         $this->assertToString(
 134             '/path/to',
 135             null, null, null, null, '/path/to', null, null
 136         );
 137     }
 138
 139     public function test_toString_query()
 140     {
 141         $this->assertToString(
 142             '?q=string',
 143             null, null, null, null, '', 'q=string', null
 144         );
 145     }
 146
 147     public function test_toString_fragment()
 148     {
 149         $this->assertToString(
 150             '#fragment',
 151             null, null, null, null, '', null, 'fragment'
 152         );
 153     }
 154
 155     protected function assertValidation($uri, $expect_uri = true)
 156     {
 157         if ($expect_uri === true) $expect_uri = $uri;
 158         $uri = $this->createURI($uri);
 159         $result = $uri->validate($this->config, $this->context);
 160         if ($expect_uri === false) {
 161             $this->assertFalse($result);
 162         } else {
 163             $this->assertTrue($result);
 164             $this->assertIdentical($uri->toString(), $expect_uri);
 165         }
 166     }
 167
 168     public function test_validate_overlongPort()
 169     {
 170         $this->assertValidation('http://example.com:65536', 'http://example.com');
 171     }
 172
 173     public function test_validate_zeroPort()
 174     {
 175         $this->assertValidation('http://example.com:00', 'http://example.com');
 176     }
 177
 178     public function test_validate_invalidHostThatLooksLikeIPv6()
 179     {
 180         $this->assertValidation('http://[2001:0db8:85z3:08d3:1319:8a2e:0370:7334]', '');
 181     }
 182
 183     public function test_validate_removeRedundantScheme()
 184     {
 185         $this->assertValidation('http:foo:/:', 'foo%3A/:');
 186     }
 187
 188     public function test_validate_username()
 189     {
 190         $this->assertValidation("http://user\xE3\x91\x94:@foo.com", 'http://user%E3%91%94:@foo.com');
 191     }
 192
 193     public function test_validate_path_abempty()
 194     {
 195         $this->assertValidation("http://host/\xE3\x91\x94:", 'http://host/%E3%91%94:');
 196     }
 197
 198     public function test_validate_path_absolute()
 199     {
 200         $this->assertValidation("/\xE3\x91\x94:", '/%E3%91%94:');
 201     }
 202
 203     public function test_validate_path_rootless()
 204     {
 205         $this->assertValidation("mailto:\xE3\x91\x94:", 'mailto:%E3%91%94:');
 206     }
 207
 208     public function test_validate_path_noscheme()
 209     {
 210         $this->assertValidation("\xE3\x91\x94", '%E3%91%94');
 211     }
 212
 213     public function test_validate_query()
 214     {
 215         $this->assertValidation("?/\xE3\x91\x94", '?/%E3%91%94');
 216     }
 217
 218     public function test_validate_fragment()
 219     {
 220         $this->assertValidation("#/\xE3\x91\x94", '#/%E3%91%94');
 221     }
 222
 223     public function test_validate_path_empty()
 224     {
 225         $this->assertValidation('http://google.com');
 226     }
 227
 228 }
 229
 230 // vim: et sw=4 sts=4