ExternalStoreTest.php

Go to the documentation of this file.

<?php
class ExternalStoreTest extends MediaWikiTestCase {

        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', array( '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 = array(
                'cluster1' => array(
                        '200' => 'Hello',
                        '300' => array(
                                '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];
        }
}