tests/HTMLPurifier/URISchemeRegistryTest.php

   1 <?php
   2
   3 require_once 'HTMLPurifier/URISchemeRegistry.php';
   4
   5 class HTMLPurifier_URISchemeRegistryTest extends HTMLPurifier_Harness
   6 {
   7
   8     function test() {
   9
  10         generate_mock_once('HTMLPurifier_URIScheme');
  11
  12         $config = HTMLPurifier_Config::create(array(
  13             'URI.AllowedSchemes' => 'http, telnet',
  14             'URI.OverrideAllowedSchemes' => true
  15         ));
  16         $context = new HTMLPurifier_Context();
  17
  18         $registry = new HTMLPurifier_URISchemeRegistry();
  19         $this->assertIsA($registry->getScheme('http', $config, $context), 'HTMLPurifier_URIScheme_http');
  20
  21         $scheme_http = new HTMLPurifier_URISchemeMock();
  22         $scheme_telnet = new HTMLPurifier_URISchemeMock();
  23         $scheme_foobar = new HTMLPurifier_URISchemeMock();
  24
  25         // register a new scheme
  26         $registry->register('telnet', $scheme_telnet);
  27         $this->assertIdentical($registry->getScheme('telnet', $config, $context), $scheme_telnet);
  28
  29         // overload a scheme, this is FINAL (forget about defaults)
  30         $registry->register('http', $scheme_http);
  31         $this->assertIdentical($registry->getScheme('http', $config, $context), $scheme_http);
  32
  33         // when we register a scheme, it's automatically allowed
  34         $registry->register('foobar', $scheme_foobar);
  35         $this->assertIdentical($registry->getScheme('foobar', $config, $context), $scheme_foobar);
  36
  37         // now, test when overriding is not allowed
  38         $config = HTMLPurifier_Config::create(array(
  39             'URI.AllowedSchemes' => 'http, telnet',
  40             'URI.OverrideAllowedSchemes' => false
  41         ));
  42         $this->assertNull($registry->getScheme('foobar', $config, $context));
  43
  44         // scheme not allowed and never registered
  45         $this->assertNull($registry->getScheme('ftp', $config, $context));
  46
  47     }
  48
  49 }
  50