DBSiteStoreTest.php

Go to the documentation of this file.

<?php
 
use MediaWiki\MediaWikiServices;
 
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.
        $lb = MediaWikiServices::getInstance()->getDBLoadBalancer();
        return new DBSiteStore( $lb );
    }
 
    public function testGetSites() {
        $expectedSites = TestSites::getSites();
        TestSites::insertIntoDb();
 
        $store = $this->newDBSiteStore();
 
        $sites = $store->getSites();
 
        $this->assertInstanceOf( SiteList::class, $sites );
 
        foreach ( $sites as $site ) {
            $this->assertInstanceOf( Site::class, $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::class, $site );
        $this->assertEquals( 'en', $site->getLanguageCode() );
        $this->assertTrue( is_int( $site->getInternalId() ) );
        $this->assertTrue( $site->getInternalId() >= 0 );
 
        $site = $store->getSite( 'sdfhxujgkfpth' );
        $this->assertInstanceOf( Site::class, $site );
        $this->assertEquals( 'nl', $site->getLanguageCode() );
        $this->assertTrue( is_int( $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() );
    }
}