DBSiteStoreTest.php

Go to the documentation of this file.

<?php

class DBSiteStoreTest extends MediaWikiTestCase {

    private function newDBSiteStore() {
        // NOTE: Use the real DB load balancer for now. Eventually, the test framework should
        // provide a LoadBalancer that is safe to use in unit tests.
        return new DBSiteStore( wfGetLB() );
    }

    public function testGetSites() {
        $expectedSites = TestSites::getSites();
        TestSites::insertIntoDb();

        $store = $this->newDBSiteStore();

        $sites = $store->getSites();

        $this->assertInstanceOf( 'SiteList', $sites );

        foreach ( $sites as $site ) {
            $this->assertInstanceOf( 'Site', $site );
        }

        foreach ( $expectedSites as $site ) {
            if ( $site->getGlobalId() !== null ) {
                $this->assertTrue( $sites->hasSite( $site->getGlobalId() ) );
            }
        }
    }

    public function testSaveSites() {
        $store = $this->newDBSiteStore();

        $sites = [];

        $site = new Site();
        $site->setGlobalId( 'ertrywuutr' );
        $site->setLanguageCode( 'en' );
        $sites[] = $site;

        $site = new MediaWikiSite();
        $site->setGlobalId( 'sdfhxujgkfpth' );
        $site->setLanguageCode( 'nl' );
        $sites[] = $site;

        $this->assertTrue( $store->saveSites( $sites ) );

        $site = $store->getSite( 'ertrywuutr' );
        $this->assertInstanceOf( 'Site', $site );
        $this->assertEquals( 'en', $site->getLanguageCode() );
        $this->assertTrue( is_integer( $site->getInternalId() ) );
        $this->assertTrue( $site->getInternalId() >= 0 );

        $site = $store->getSite( 'sdfhxujgkfpth' );
        $this->assertInstanceOf( 'Site', $site );
        $this->assertEquals( 'nl', $site->getLanguageCode() );
        $this->assertTrue( is_integer( $site->getInternalId() ) );
        $this->assertTrue( $site->getInternalId() >= 0 );
    }

    public function testReset() {
        $store1 = $this->newDBSiteStore();
        $store2 = $this->newDBSiteStore();

        // initialize internal cache
        $this->assertGreaterThan( 0, $store1->getSites()->count() );
        $this->assertGreaterThan( 0, $store2->getSites()->count() );

        // Clear actual data. Will purge the external cache and reset the internal
        // cache in $store1, but not the internal cache in store2.
        $this->assertTrue( $store1->clear() );

        // sanity check: $store2 should have a stale cache now
        $this->assertNotNull( $store2->getSite( 'enwiki' ) );

        // purge cache
        $store2->reset();

        // ...now the internal cache of $store2 should be updated and thus empty.
        $site = $store2->getSite( 'enwiki' );
        $this->assertNull( $site );
    }

    public function testClear() {
        $store = $this->newDBSiteStore();
        $this->assertTrue( $store->clear() );

        $site = $store->getSite( 'enwiki' );
        $this->assertNull( $site );

        $sites = $store->getSites();
        $this->assertEquals( 0, $sites->count() );
    }

    public function testGetSitesDefaultOrder() {
        $store = $this->newDBSiteStore();
        $siteB = new Site();
        $siteB->setGlobalId( 'B' );
        $siteA = new Site();
        $siteA->setGlobalId( 'A' );
        $store->saveSites( [ $siteB, $siteA ] );

        $sites = $store->getSites();
        $siteIdentifiers = [];
        foreach ( $sites as $site ) {
            $siteIdentifiers[] = $site->getGlobalId();
        }
        $this->assertSame( [ 'A', 'B' ], $siteIdentifiers );

        // Note: SiteList::getGlobalIdentifiers uses an other internal state. Iteration must be
        // tested separately.
        $this->assertSame( [ 'A', 'B' ], $sites->getGlobalIdentifiers() );
    }
}