<?php
class HTMLPurifier_URITest extends HTMLPurifier_URIHarness {
	protected function createURI($uri) {
		$parser = new HTMLPurifier_URIParser ();
		return $parser->parse ( $uri );
	}
	public function test_construct() {
		$uri1 = new HTMLPurifier_URI ( 'HTTP', 'bob', 'example.com', '23', '/foo', 'bar=2', 'slash' );
		$uri2 = new HTMLPurifier_URI ( 'http', 'bob', 'example.com', 23, '/foo', 'bar=2', 'slash' );
		$this->assertIdentical ( $uri1, $uri2 );
	}
	protected $oldRegistry;
	protected function &setUpSchemeRegistryMock() {
		$this->oldRegistry = HTMLPurifier_URISchemeRegistry::instance ();
		generate_mock_once ( 'HTMLPurifier_URIScheme' );
		generate_mock_once ( 'HTMLPurifier_URISchemeRegistry' );
		$registry = HTMLPurifier_URISchemeRegistry::instance ( new HTMLPurifier_URISchemeRegistryMock () );
		return $registry;
	}
	protected function setUpSchemeMock($name) {
		$registry = $this->setUpSchemeRegistryMock ();
		$scheme_mock = new HTMLPurifier_URISchemeMock ();
		$registry->setReturnValue ( 'getScheme', $scheme_mock, array (
				$name,
				'*',
				'*' 
		) );
		return $scheme_mock;
	}
	protected function setUpNoValidSchemes() {
		$registry = $this->setUpSchemeRegistryMock ();
		$registry->setReturnValue ( 'getScheme', false, array (
				'*',
				'*',
				'*' 
		) );
	}
	protected function tearDownSchemeRegistryMock() {
		HTMLPurifier_URISchemeRegistry::instance ( $this->oldRegistry );
	}
	public function test_getSchemeObj() {
		$scheme_mock = $this->setUpSchemeMock ( 'http' );
		
		$uri = $this->createURI ( 'http:' );
		$scheme_obj = $uri->getSchemeObj ( $this->config, $this->context );
		$this->assertIdentical ( $scheme_obj, $scheme_mock );
		
		$this->tearDownSchemeRegistryMock ();
	}
	public function test_getSchemeObj_invalidScheme() {
		$this->setUpNoValidSchemes ();
		
		$uri = $this->createURI ( 'http:' );
		$result = $uri->getSchemeObj ( $this->config, $this->context );
		$this->assertIdentical ( $result, false );
		
		$this->tearDownSchemeRegistryMock ();
	}
	public function test_getSchemaObj_defaultScheme() {
		$scheme = 'foobar';
		
		$scheme_mock = $this->setUpSchemeMock ( $scheme );
		$this->config->set ( 'URI.DefaultScheme', $scheme );
		
		$uri = $this->createURI ( 'hmm' );
		$scheme_obj = $uri->getSchemeObj ( $this->config, $this->context );
		$this->assertIdentical ( $scheme_obj, $scheme_mock );
		
		$this->tearDownSchemeRegistryMock ();
	}
	public function test_getSchemaObj_invalidDefaultScheme() {
		$this->setUpNoValidSchemes ();
		$this->config->set ( 'URI.DefaultScheme', 'foobar' );
		
		$uri = $this->createURI ( 'hmm' );
		
		$this->expectError ( 'Default scheme object "foobar" was not readable' );
		$result = $uri->getSchemeObj ( $this->config, $this->context );
		$this->assertIdentical ( $result, false );
		
		$this->tearDownSchemeRegistryMock ();
	}
	protected function assertToString($expect_uri, $scheme, $userinfo, $host, $port, $path, $query, $fragment) {
		$uri = new HTMLPurifier_URI ( $scheme, $userinfo, $host, $port, $path, $query, $fragment );
		$string = $uri->toString ();
		$this->assertIdentical ( $string, $expect_uri );
	}
	public function test_toString_full() {
		$this->assertToString ( 'http://bob@example.com:300/foo?bar=baz#fragment', 'http', 'bob', 'example.com', 300, '/foo', 'bar=baz', 'fragment' );
	}
	public function test_toString_scheme() {
		$this->assertToString ( 'http:', 'http', null, null, null, '', null, null );
	}
	public function test_toString_authority() {
		$this->assertToString ( '//bob@example.com:8080', null, 'bob', 'example.com', 8080, '', null, null );
	}
	public function test_toString_path() {
		$this->assertToString ( '/path/to', null, null, null, null, '/path/to', null, null );
	}
	public function test_toString_query() {
		$this->assertToString ( '?q=string', null, null, null, null, '', 'q=string', null );
	}
	public function test_toString_fragment() {
		$this->assertToString ( '#fragment', null, null, null, null, '', null, 'fragment' );
	}
	protected function assertValidation($uri, $expect_uri = true) {
		if ($expect_uri === true)
			$expect_uri = $uri;
		$uri = $this->createURI ( $uri );
		$result = $uri->validate ( $this->config, $this->context );
		if ($expect_uri === false) {
			$this->assertFalse ( $result );
		} else {
			$this->assertTrue ( $result );
			$this->assertIdentical ( $uri->toString (), $expect_uri );
		}
	}
	public function test_validate_overlongPort() {
		$this->assertValidation ( 'http://example.com:65536', 'http://example.com' );
	}
	public function test_validate_zeroPort() {
		$this->assertValidation ( 'http://example.com:00', 'http://example.com' );
	}
	public function test_validate_invalidHostThatLooksLikeIPv6() {
		$this->assertValidation ( 'http://[2001:0db8:85z3:08d3:1319:8a2e:0370:7334]', '' );
	}
	public function test_validate_removeRedundantScheme() {
		$this->assertValidation ( 'http:foo:/:', 'foo%3A/:' );
	}
	public function test_validate_username() {
		$this->assertValidation ( "http://user\xE3\x91\x94:@foo.com", 'http://user%E3%91%94:@foo.com' );
	}
	public function test_validate_path_abempty() {
		$this->assertValidation ( "http://host/\xE3\x91\x94:", 'http://host/%E3%91%94:' );
	}
	public function test_validate_path_absolute() {
		$this->assertValidation ( "/\xE3\x91\x94:", '/%E3%91%94:' );
	}
	public function test_validate_path_rootless() {
		$this->assertValidation ( "mailto:\xE3\x91\x94:", 'mailto:%E3%91%94:' );
	}
	public function test_validate_path_noscheme() {
		$this->assertValidation ( "\xE3\x91\x94", '%E3%91%94' );
	}
	public function test_validate_query() {
		$this->assertValidation ( "?/\xE3\x91\x94", '?/%E3%91%94' );
	}
	public function test_validate_fragment() {
		$this->assertValidation ( "#/\xE3\x91\x94", '#/%E3%91%94' );
	}
	public function test_validate_path_empty() {
		$this->assertValidation ( 'http://google.com' );
	}
}

// vim: et sw=4 sts=4