MultiWriteBagOStuffTest.php

Go to the documentation of this file.

<?php
 
class MultiWriteBagOStuffTest extends MediaWikiTestCase {
    private $cache1;
    private $cache2;
    private $cache;
 
    protected function setUp() {
        parent::setUp();
 
        $this->cache1 = new HashBagOStuff();
        $this->cache2 = new HashBagOStuff();
        $this->cache = new MultiWriteBagOStuff( [
            'caches' => [ $this->cache1, $this->cache2 ],
            'replication' => 'async',
            'asyncHandler' => 'DeferredUpdates::addCallableUpdate'
        ] );
    }
 
    public function testSetImmediate() {
        $key = wfRandomString();
        $value = wfRandomString();
        $this->cache->set( $key, $value );
 
        // Set in tier 1
        $this->assertEquals( $value, $this->cache1->get( $key ), 'Written to tier 1' );
        // Set in tier 2
        $this->assertEquals( $value, $this->cache2->get( $key ), 'Written to tier 2' );
    }
 
    public function testSyncMerge() {
        $key = wfRandomString();
        $value = wfRandomString();
        $func = function () use ( $value ) {
            return $value;
        };
 
        // XXX: DeferredUpdates bound to transactions in CLI mode
        $dbw = wfGetDB( DB_MASTER );
        $dbw->begin();
        $this->cache->merge( $key, $func );
 
        // Set in tier 1
        $this->assertEquals( $value, $this->cache1->get( $key ), 'Written to tier 1' );
        // Not yet set in tier 2
        $this->assertEquals( false, $this->cache2->get( $key ), 'Not written to tier 2' );
 
        $dbw->commit();
 
        // Set in tier 2
        $this->assertEquals( $value, $this->cache2->get( $key ), 'Written to tier 2' );
 
        $key = wfRandomString();
 
        $dbw->begin();
        $this->cache->merge( $key, $func, 0, 1, BagOStuff::WRITE_SYNC );
 
        // Set in tier 1
        $this->assertEquals( $value, $this->cache1->get( $key ), 'Written to tier 1' );
        // Also set in tier 2
        $this->assertEquals( $value, $this->cache2->get( $key ), 'Written to tier 2' );
 
        $dbw->commit();
    }
 
    public function testSetDelayed() {
        $key = wfRandomString();
        $value = (object)[ 'v' => wfRandomString() ];
        $expectValue = clone $value;
 
        // XXX: DeferredUpdates bound to transactions in CLI mode
        $dbw = wfGetDB( DB_MASTER );
        $dbw->begin();
        $this->cache->set( $key, $value );
 
        // Test that later changes to $value don't affect the saved value (e.g. T168040)
        $value->v = 'bogus';
 
        // Set in tier 1
        $this->assertEquals( $expectValue, $this->cache1->get( $key ), 'Written to tier 1' );
        // Not yet set in tier 2
        $this->assertEquals( false, $this->cache2->get( $key ), 'Not written to tier 2' );
 
        $dbw->commit();
 
        // Set in tier 2
        $this->assertEquals( $expectValue, $this->cache2->get( $key ), 'Written to tier 2' );
    }
 
    public function testMakeKey() {
        $cache1 = $this->getMockBuilder( HashBagOStuff::class )
            ->setMethods( [ 'makeKey' ] )->getMock();
        $cache1->expects( $this->once() )->method( 'makeKey' )
            ->willReturn( 'special' );
 
        $cache2 = $this->getMockBuilder( HashBagOStuff::class )
            ->setMethods( [ 'makeKey' ] )->getMock();
        $cache2->expects( $this->never() )->method( 'makeKey' );
 
        $cache = new MultiWriteBagOStuff( [ 'caches' => [ $cache1, $cache2 ] ] );
        $this->assertSame( 'special', $cache->makeKey( 'a', 'b' ) );
    }
 
    public function testMakeGlobalKey() {
        $cache1 = $this->getMockBuilder( HashBagOStuff::class )
            ->setMethods( [ 'makeGlobalKey' ] )->getMock();
        $cache1->expects( $this->once() )->method( 'makeGlobalKey' )
            ->willReturn( 'special' );
 
        $cache2 = $this->getMockBuilder( HashBagOStuff::class )
            ->setMethods( [ 'makeGlobalKey' ] )->getMock();
        $cache2->expects( $this->never() )->method( 'makeGlobalKey' );
 
        $cache = new MultiWriteBagOStuff( [ 'caches' => [ $cache1, $cache2 ] ] );
 
        $this->assertSame( 'special', $cache->makeGlobalKey( 'a', 'b' ) );
    }
 
    public function testDuplicateStoreAdd() {
        $bag = new HashBagOStuff();
        $cache = new MultiWriteBagOStuff( [
            'caches' => [ $bag, $bag ],
        ] );
 
        $this->assertTrue( $cache->add( 'key', 1, 30 ) );
    }
}