PhabricatorEnvTestCase.php

<?php

final class PhabricatorEnvTestCase extends PhabricatorTestCase {

  public function testLocalURIForLink() {
    $map = array(
      '/'                     => true,
      '/D123'                 => true,
      '/path/to/something/'   => true,
      "/path/to/\nHeader: x"  => false,
      'http://evil.com/'      => false,
      '//evil.com/evil/'      => false,
      'javascript:lol'        => false,
      ''                      => false,
      null                    => false,
      '/\\evil.com'           => false,
    );

    foreach ($map as $uri => $expect) {
      $this->assertEqual(
        $expect,
        PhabricatorEnv::isValidLocalURIForLink($uri),
        pht('Valid local resource: %s', $uri));
    }
  }

  public function testRemoteURIForLink() {
    $map = array(
      'http://example.com/' => true,
      'derp://example.com/' => false,
      'javascript:alert(1)' => false,
      'http://127.0.0.1/' => true,
      'http://169.254.169.254/latest/meta-data/hostname' => true,
    );

    foreach ($map as $uri => $expect) {
      $this->assertEqual(
        $expect,
        PhabricatorEnv::isValidRemoteURIForLink($uri),
        pht('Valid linkable remote URI: %s', $uri));
    }
  }

  public function testRemoteURIForFetch() {
    $map = array(
      'http://example.com/' => true,

      // No domain or protocol.
      '' => false,

      // No domain.
      'http://' => false,

      // No protocol.
      'evil.com' => false,

      // No protocol.
      '//evil.com' => false,

      // Bad protocol.
      'javascript://evil.com/' => false,
      'file:///etc/shadow' => false,

      // Unresolvable hostname.
      'http://u1VcxwUp368SIFzl7rkWWg23KM5JPB2kTHHngxjXCQc.zzz/' => false,

      // Domains explicitly in blacklisted IP space.
      'http://127.0.0.1/' => false,
      'http://169.254.169.254/latest/meta-data/hostname' => false,

      // Domain resolves into blacklisted IP space.
      'http://localhost/' => false,
    );

    $protocols = array('http', 'https');

    foreach ($map as $uri => $expect) {
      $this->assertEqual(
        $expect,
        PhabricatorEnv::isValidRemoteURIForFetch($uri, $protocols),
        pht('Valid fetchable remote URI: %s', $uri));
    }
  }

  public function testDictionarySource() {
    $source = new PhabricatorConfigDictionarySource(array('x' => 1));

    $this->assertEqual(
      array(
        'x' => 1,
      ),
      $source->getKeys(array('x', 'z')));

    $source->setKeys(array('z' => 2));

    $this->assertEqual(
      array(
        'x' => 1,
        'z' => 2,
      ),
      $source->getKeys(array('x', 'z')));

    $source->setKeys(array('x' => 3));

    $this->assertEqual(
      array(
        'x' => 3,
        'z' => 2,
      ),
      $source->getKeys(array('x', 'z')));

    $source->deleteKeys(array('x'));

    $this->assertEqual(
      array(
        'z' => 2,
      ),
      $source->getKeys(array('x', 'z')));
  }

  public function testStackSource() {
    $s1 = new PhabricatorConfigDictionarySource(array('x' => 1));
    $s2 = new PhabricatorConfigDictionarySource(array('x' => 2));

    $stack = new PhabricatorConfigStackSource();

    $this->assertEqual(array(), $stack->getKeys(array('x')));

    $stack->pushSource($s1);
    $this->assertEqual(array('x' => 1), $stack->getKeys(array('x')));

    $stack->pushSource($s2);
    $this->assertEqual(array('x' => 2), $stack->getKeys(array('x')));

    $stack->setKeys(array('x' => 3));
    $this->assertEqual(array('x' => 3), $stack->getKeys(array('x')));

    $stack->popSource();
    $this->assertEqual(array('x' => 1), $stack->getKeys(array('x')));

    $stack->popSource();
    $this->assertEqual(array(), $stack->getKeys(array('x')));

    $caught = null;
    try {
      $stack->popSource();
    } catch (Exception $ex) {
      $caught = $ex;
    }

    $this->assertTrue($caught instanceof Exception);
  }

  public function testOverrides() {
    $outer = PhabricatorEnv::beginScopedEnv();

      $outer->overrideEnvConfig('test.value', 1);
      $this->assertEqual(1, PhabricatorEnv::getEnvConfig('test.value'));

      $inner = PhabricatorEnv::beginScopedEnv();
        $inner->overrideEnvConfig('test.value', 2);
        $this->assertEqual(2, PhabricatorEnv::getEnvConfig('test.value'));
      if (phutil_is_hiphop_runtime()) {
        $inner->__destruct();
      }
      unset($inner);

      $this->assertEqual(1, PhabricatorEnv::getEnvConfig('test.value'));
    if (phutil_is_hiphop_runtime()) {
      $outer->__destruct();
    }
    unset($outer);
  }

  public function testOverrideOrder() {
    $outer = PhabricatorEnv::beginScopedEnv();
    $inner = PhabricatorEnv::beginScopedEnv();

    $caught = null;
    try {
      $outer->__destruct();
    } catch (Exception $ex) {
      $caught = $ex;
    }

    $this->assertTrue(
      $caught instanceof Exception,
      pht(
        'Destroying a scoped environment which is not on the top of the '.
        'stack should throw.'));

    if (phutil_is_hiphop_runtime()) {
      $inner->__destruct();
    }
    unset($inner);

    if (phutil_is_hiphop_runtime()) {
      $outer->__destruct();
    }
    unset($outer);
  }

  public function testGetEnvExceptions() {
    $caught = null;
    try {
      PhabricatorEnv::getEnvConfig('not.a.real.config.option');
    } catch (Exception $ex) {
      $caught = $ex;
    }
    $this->assertTrue($caught instanceof Exception);

    $caught = null;
    try {
      PhabricatorEnv::getEnvConfig('test.value');
    } catch (Exception $ex) {
      $caught = $ex;
    }
    $this->assertFalse($caught instanceof Exception);
  }

  public function testSelfURI() {
    $base_uri = 'https://allowed.example.com/';

    $allowed_uris = array(
      'https://old.example.com/',
    );

    $env = PhabricatorEnv::beginScopedEnv();
    $env->overrideEnvConfig('phabricator.base-uri', $base_uri);
    $env->overrideEnvConfig('phabricator.allowed-uris', $allowed_uris);

    $map = array(
      'https://allowed.example.com/' => true,
      'https://allowed.example.com' => true,
      'https://allowed.EXAMPLE.com' => true,
      'http://allowed.example.com/' => true,
      'https://allowed.example.com/path/to/resource.png' => true,

      'https://old.example.com/' => true,
      'https://old.example.com' => true,
      'https://old.EXAMPLE.com' => true,
      'http://old.example.com/' => true,
      'https://old.example.com/path/to/resource.png' => true,

      'https://other.example.com/' => false,
    );

    foreach ($map as $input => $expect) {
      $this->assertEqual(
        $expect,
        PhabricatorEnv::isSelfURI($input),
        pht('Is self URI? %s', $input));
    }
  }

}