60 lines
3.3 KiB
PHP
60 lines
3.3 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Tests\Unit;
|
||
|
|
|
||
|
|
use BookStack\Exceptions\HttpFetchException;
|
||
|
|
use BookStack\Util\SsrUrlValidator;
|
||
|
|
use Tests\TestCase;
|
||
|
|
|
||
|
|
class SsrUrlValidatorTest extends TestCase
|
||
|
|
{
|
||
|
|
public function test_allowed()
|
||
|
|
{
|
||
|
|
$testMap = [
|
||
|
|
// Single values
|
||
|
|
['config' => '', 'url' => '', 'result' => false],
|
||
|
|
['config' => '', 'url' => 'https://example.com', 'result' => false],
|
||
|
|
['config' => ' ', 'url' => 'https://example.com', 'result' => false],
|
||
|
|
['config' => '*', 'url' => '', 'result' => false],
|
||
|
|
['config' => '*', 'url' => 'https://example.com', 'result' => true],
|
||
|
|
['config' => 'https://*', 'url' => 'https://example.com', 'result' => true],
|
||
|
|
['config' => 'http://*', 'url' => 'https://example.com', 'result' => false],
|
||
|
|
['config' => 'https://*example.com', 'url' => 'https://example.com', 'result' => true],
|
||
|
|
['config' => 'https://*ample.com', 'url' => 'https://example.com', 'result' => true],
|
||
|
|
['config' => 'https://*.example.com', 'url' => 'https://example.com', 'result' => false],
|
||
|
|
['config' => 'https://*.example.com', 'url' => 'https://test.example.com', 'result' => true],
|
||
|
|
['config' => '*//example.com', 'url' => 'https://example.com', 'result' => true],
|
||
|
|
['config' => '*//example.com', 'url' => 'http://example.com', 'result' => true],
|
||
|
|
['config' => 'https://example.com', 'url' => 'https://example.com/a/b/c?test=cat', 'result' => true],
|
||
|
|
['config' => 'https://example.com', 'url' => 'https://example.co.uk', 'result' => false],
|
||
|
|
|
||
|
|
// Escapes
|
||
|
|
['config' => 'https://(.*?).com', 'url' => 'https://example.com', 'result' => false],
|
||
|
|
['config' => 'https://example.com', 'url' => 'https://example.co.uk#https://example.com', 'result' => false],
|
||
|
|
|
||
|
|
// Multi values
|
||
|
|
['config' => '*//example.org *//example.com', 'url' => 'https://example.com', 'result' => true],
|
||
|
|
['config' => '*//example.org *//example.com', 'url' => 'https://example.com/a/b/c?test=cat#hello', 'result' => true],
|
||
|
|
['config' => '*.example.org *.example.com', 'url' => 'https://example.co.uk', 'result' => false],
|
||
|
|
['config' => ' *.example.org *.example.com ', 'url' => 'https://example.co.uk', 'result' => false],
|
||
|
|
['config' => '* *.example.com', 'url' => 'https://example.co.uk', 'result' => true],
|
||
|
|
['config' => '*//example.org *//example.com *//example.co.uk', 'url' => 'https://example.co.uk', 'result' => true],
|
||
|
|
['config' => '*//example.org *//example.com *//example.co.uk', 'url' => 'https://example.net', 'result' => false],
|
||
|
|
];
|
||
|
|
|
||
|
|
foreach ($testMap as $test) {
|
||
|
|
$result = (new SsrUrlValidator($test['config']))->allowed($test['url']);
|
||
|
|
$this->assertEquals($test['result'], $result, "Failed asserting url '{$test['url']}' with config '{$test['config']}' results " . ($test['result'] ? 'true' : 'false'));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public function test_enssure_allowed()
|
||
|
|
{
|
||
|
|
$result = (new SsrUrlValidator('https://example.com'))->ensureAllowed('https://example.com');
|
||
|
|
$this->assertNull($result);
|
||
|
|
|
||
|
|
$this->expectException(HttpFetchException::class);
|
||
|
|
(new SsrUrlValidator('https://example.com'))->ensureAllowed('https://test.example.com');
|
||
|
|
}
|
||
|
|
}
|