ExternalStoreTest.php

Go to the documentation of this file.

<?php
class ExternalStoreTest extends MediaWikiTestCase {
 
    public function testExternalFetchFromURL() {
        $this->setMwGlobals( 'wgExternalStores', false );
 
        $this->assertFalse(
            ExternalStore::fetchFromURL( 'FOO://cluster1/200' ),
            'Deny if wgExternalStores is not set to a non-empty array'
        );
 
        $this->setMwGlobals( 'wgExternalStores', [ 'FOO' ] );
 
        $this->assertEquals(
            ExternalStore::fetchFromURL( 'FOO://cluster1/200' ),
            'Hello',
            'Allow FOO://cluster1/200'
        );
        $this->assertEquals(
            ExternalStore::fetchFromURL( 'FOO://cluster1/300/0' ),
            'Hello',
            'Allow FOO://cluster1/300/0'
        );
        # Assertions for r68900
        $this->assertFalse(
            ExternalStore::fetchFromURL( 'ftp.example.org' ),
            'Deny domain ftp.example.org'
        );
        $this->assertFalse(
            ExternalStore::fetchFromURL( '/example.txt' ),
            'Deny path /example.txt'
        );
        $this->assertFalse(
            ExternalStore::fetchFromURL( 'http://' ),
            'Deny protocol http://'
        );
    }
}
 
class ExternalStoreFOO {
 
    protected $data = [
        'cluster1' => [
            '200' => 'Hello',
            '300' => [
                'Hello', 'World',
            ],
        ],
    ];
 
    function fetchFromURL( $url ) {
        // Based on ExternalStoreDB
        $path = explode( '/', $url );
        $cluster = $path[2];
        $id = $path[3];
        if ( isset( $path[4] ) ) {
            $itemID = $path[4];
        } else {
            $itemID = false;
        }
 
        if ( !isset( $this->data[$cluster][$id] ) ) {
            return null;
        }
 
        if ( $itemID !== false
            && is_array( $this->data[$cluster][$id] )
            && isset( $this->data[$cluster][$id][$itemID] )
        ) {
            return $this->data[$cluster][$id][$itemID];
        }
 
        return $this->data[$cluster][$id];
    }
}