LBFactoryTest.php

Go to the documentation of this file.

<?php
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\IMaintainableDatabase;
use Wikimedia\Rdbms\LBFactory;
use Wikimedia\Rdbms\LBFactorySimple;
use Wikimedia\Rdbms\LBFactoryMulti;
use Wikimedia\Rdbms\LoadBalancer;
use Wikimedia\Rdbms\ChronologyProtector;
use Wikimedia\Rdbms\DatabaseMysqli;
use Wikimedia\Rdbms\MySQLMasterPos;
use Wikimedia\Rdbms\DatabaseDomain;
 
class LBFactoryTest extends MediaWikiTestCase {
 
    public function testGetLBFactoryClass( $expected, $deprecated ) {
        $mockDB = $this->getMockBuilder( DatabaseMysqli::class )
            ->disableOriginalConstructor()
            ->getMock();
 
        $config = [
            'class'          => $deprecated,
            'connection'     => $mockDB,
            # Various other parameters required:
            'sectionsByDB'   => [],
            'sectionLoads'   => [],
            'serverTemplate' => [],
        ];
 
        $this->hideDeprecated( '$wgLBFactoryConf must be updated. See RELEASE-NOTES for details' );
        $result = MWLBFactory::getLBFactoryClass( $config );
 
        $this->assertEquals( $expected, $result );
    }
 
    public function getLBFactoryClassProvider() {
        return [
            # Format: new class, old class
            [ Wikimedia\Rdbms\LBFactorySimple::class, 'LBFactory_Simple' ],
            [ Wikimedia\Rdbms\LBFactorySingle::class, 'LBFactory_Single' ],
            [ Wikimedia\Rdbms\LBFactoryMulti::class, 'LBFactory_Multi' ],
            [ Wikimedia\Rdbms\LBFactorySimple::class, 'LBFactorySimple' ],
            [ Wikimedia\Rdbms\LBFactorySingle::class, 'LBFactorySingle' ],
            [ Wikimedia\Rdbms\LBFactoryMulti::class, 'LBFactoryMulti' ],
        ];
    }
 
    public function testLBFactorySimpleServer() {
        global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
 
        $servers = [
            [
                'host'        => $wgDBserver,
                'dbname'      => $wgDBname,
                'user'        => $wgDBuser,
                'password'    => $wgDBpassword,
                'type'        => $wgDBtype,
                'dbDirectory' => $wgSQLiteDataDir,
                'load'        => 0,
                'flags'       => DBO_TRX // REPEATABLE-READ for consistency
            ],
        ];
 
        $factory = new LBFactorySimple( [ 'servers' => $servers ] );
        $lb = $factory->getMainLB();
 
        $dbw = $lb->getConnection( DB_MASTER );
        $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
 
        $dbr = $lb->getConnection( DB_REPLICA );
        $this->assertTrue( $dbr->getLBInfo( 'master' ), 'DB_REPLICA also gets the master' );
 
        $this->assertSame( 'my_test_wiki', $factory->resolveDomainID( 'my_test_wiki' ) );
        $this->assertSame( $factory->getLocalDomainID(), $factory->resolveDomainID( false ) );
 
        $factory->shutdown();
        $lb->closeAll();
    }
 
    public function testLBFactorySimpleServers() {
        global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
 
        $servers = [
            [ // master
                'host'        => $wgDBserver,
                'dbname'      => $wgDBname,
                'user'        => $wgDBuser,
                'password'    => $wgDBpassword,
                'type'        => $wgDBtype,
                'dbDirectory' => $wgSQLiteDataDir,
                'load'        => 0,
                'flags'       => DBO_TRX // REPEATABLE-READ for consistency
            ],
            [ // emulated replica
                'host'        => $wgDBserver,
                'dbname'      => $wgDBname,
                'user'        => $wgDBuser,
                'password'    => $wgDBpassword,
                'type'        => $wgDBtype,
                'dbDirectory' => $wgSQLiteDataDir,
                'load'        => 100,
                'flags'       => DBO_TRX // REPEATABLE-READ for consistency
            ]
        ];
 
        $factory = new LBFactorySimple( [
            'servers' => $servers,
            'loadMonitorClass' => LoadMonitorNull::class
        ] );
        $lb = $factory->getMainLB();
 
        $dbw = $lb->getConnection( DB_MASTER );
        $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
        $this->assertEquals(
            ( $wgDBserver != '' ) ? $wgDBserver : 'localhost',
            $dbw->getLBInfo( 'clusterMasterHost' ),
            'cluster master set' );
 
        $dbr = $lb->getConnection( DB_REPLICA );
        $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'replica shows as replica' );
        $this->assertEquals(
            ( $wgDBserver != '' ) ? $wgDBserver : 'localhost',
            $dbr->getLBInfo( 'clusterMasterHost' ),
            'cluster master set' );
 
        $factory->shutdown();
        $lb->closeAll();
    }
 
    public function testLBFactoryMultiConns() {
        $factory = $this->newLBFactoryMultiLBs();
 
        $dbw = $factory->getMainLB()->getConnection( DB_MASTER );
        $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
 
        $dbr = $factory->getMainLB()->getConnection( DB_REPLICA );
        $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'replica shows as replica' );
 
        // Destructor should trigger without round stage errors
        unset( $factory );
    }
 
    public function testLBFactoryMultiRoundCallbacks() {
        $called = 0;
        $countLBsFunc = function ( LBFactoryMulti $factory ) {
            $count = 0;
            $factory->forEachLB( function () use ( &$count ) {
                ++$count;
            } );
 
            return $count;
        };
 
        $factory = $this->newLBFactoryMultiLBs();
        $this->assertEquals( 0, $countLBsFunc( $factory ) );
        $dbw = $factory->getMainLB()->getConnection( DB_MASTER );
        $this->assertEquals( 1, $countLBsFunc( $factory ) );
        // Test that LoadBalancer instances made during pre-commit callbacks in do not
        // throw DBTransactionError due to transaction ROUND_* stages being mismatched.
        $factory->beginMasterChanges( __METHOD__ );
        $dbw->onTransactionPreCommitOrIdle( function () use ( $factory, &$called ) {
            ++$called;
            // Trigger s1 LoadBalancer instantiation during "finalize" stage.
            // There is no s1wiki DB to select so it is not in getConnection(),
            // but this fools getMainLB() at least.
            $factory->getMainLB( 's1wiki' )->getConnection( DB_MASTER );
        } );
        $factory->commitMasterChanges( __METHOD__ );
        $this->assertEquals( 1, $called );
        $this->assertEquals( 2, $countLBsFunc( $factory ) );
        $factory->shutdown();
        $factory->closeAll();
 
        $called = 0;
        $factory = $this->newLBFactoryMultiLBs();
        $this->assertEquals( 0, $countLBsFunc( $factory ) );
        $dbw = $factory->getMainLB()->getConnection( DB_MASTER );
        $this->assertEquals( 1, $countLBsFunc( $factory ) );
        // Test that LoadBalancer instances made during pre-commit callbacks in do not
        // throw DBTransactionError due to transaction ROUND_* stages being mismatched.hrow
        // DBTransactionError due to transaction ROUND_* stages being mismatched.
        $factory->beginMasterChanges( __METHOD__ );
        $dbw->query( "SELECT 1 as t", __METHOD__ );
        $dbw->onTransactionResolution( function () use ( $factory, &$called ) {
            ++$called;
            // Trigger s1 LoadBalancer instantiation during "finalize" stage.
            // There is no s1wiki DB to select so it is not in getConnection(),
            // but this fools getMainLB() at least.
            $factory->getMainLB( 's1wiki' )->getConnection( DB_MASTER );
        } );
        $factory->commitMasterChanges( __METHOD__ );
        $this->assertEquals( 1, $called );
        $this->assertEquals( 2, $countLBsFunc( $factory ) );
        $factory->shutdown();
        $factory->closeAll();
 
        $factory = $this->newLBFactoryMultiLBs();
        $dbw = $factory->getMainLB()->getConnection( DB_MASTER );
        // DBTransactionError should not be thrown
        $ran = 0;
        $dbw->onTransactionPreCommitOrIdle( function () use ( &$ran ) {
            ++$ran;
        } );
        $factory->commitAll( __METHOD__ );
        $this->assertEquals( 1, $ran );
 
        $factory->shutdown();
        $factory->closeAll();
    }
 
    private function newLBFactoryMultiLBs() {
        global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
 
        return new LBFactoryMulti( [
            'sectionsByDB' => [
                's1wiki' => 's1',
            ],
            'sectionLoads' => [
                's1' => [
                    'test-db3' => 0,
                    'test-db4' => 100,
                ],
                'DEFAULT' => [
                    'test-db1' => 0,
                    'test-db2' => 100,
                ]
            ],
            'serverTemplate' => [
                'dbname' => $wgDBname,
                'user' => $wgDBuser,
                'password' => $wgDBpassword,
                'type' => $wgDBtype,
                'dbDirectory' => $wgSQLiteDataDir,
                'flags' => DBO_DEFAULT
            ],
            'hostsByName' => [
                'test-db1' => $wgDBserver,
                'test-db2' => $wgDBserver,
                'test-db3' => $wgDBserver,
                'test-db4' => $wgDBserver
            ],
            'loadMonitorClass' => LoadMonitorNull::class
        ] );
    }
 
    public function testChronologyProtector() {
        $now = microtime( true );
 
        // (a) First HTTP request
        $m1Pos = new MySQLMasterPos( 'db1034-bin.000976/843431247', $now );
        $m2Pos = new MySQLMasterPos( 'db1064-bin.002400/794074907', $now );
 
        // Master DB 1
        $mockDB1 = $this->getMockBuilder( DatabaseMysqli::class )
            ->disableOriginalConstructor()
            ->getMock();
        $mockDB1->method( 'writesOrCallbacksPending' )->willReturn( true );
        $mockDB1->method( 'lastDoneWrites' )->willReturn( $now );
        $mockDB1->method( 'getMasterPos' )->willReturn( $m1Pos );
        // Load balancer for master DB 1
        $lb1 = $this->getMockBuilder( LoadBalancer::class )
            ->disableOriginalConstructor()
            ->getMock();
        $lb1->method( 'getConnection' )->willReturn( $mockDB1 );
        $lb1->method( 'getServerCount' )->willReturn( 2 );
        $lb1->method( 'getAnyOpenConnection' )->willReturn( $mockDB1 );
        $lb1->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
                function () use ( $mockDB1 ) {
                    $p = 0;
                    $p |= call_user_func( [ $mockDB1, 'writesOrCallbacksPending' ] );
                    $p |= call_user_func( [ $mockDB1, 'lastDoneWrites' ] );
 
                    return (bool)$p;
                }
            ) );
        $lb1->method( 'getMasterPos' )->willReturn( $m1Pos );
        $lb1->method( 'getServerName' )->with( 0 )->willReturn( 'master1' );
        // Master DB 2
        $mockDB2 = $this->getMockBuilder( DatabaseMysqli::class )
            ->disableOriginalConstructor()
            ->getMock();
        $mockDB2->method( 'writesOrCallbacksPending' )->willReturn( true );
        $mockDB2->method( 'lastDoneWrites' )->willReturn( $now );
        $mockDB2->method( 'getMasterPos' )->willReturn( $m2Pos );
        // Load balancer for master DB 2
        $lb2 = $this->getMockBuilder( LoadBalancer::class )
            ->disableOriginalConstructor()
            ->getMock();
        $lb2->method( 'getConnection' )->willReturn( $mockDB2 );
        $lb2->method( 'getServerCount' )->willReturn( 2 );
        $lb2->method( 'getAnyOpenConnection' )->willReturn( $mockDB2 );
        $lb2->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
            function () use ( $mockDB2 ) {
                $p = 0;
                $p |= call_user_func( [ $mockDB2, 'writesOrCallbacksPending' ] );
                $p |= call_user_func( [ $mockDB2, 'lastDoneWrites' ] );
 
                return (bool)$p;
            }
        ) );
        $lb2->method( 'getMasterPos' )->willReturn( $m2Pos );
        $lb2->method( 'getServerName' )->with( 0 )->willReturn( 'master2' );
 
        $bag = new HashBagOStuff();
        $cp = new ChronologyProtector(
            $bag,
            [
                'ip' => '127.0.0.1',
                'agent' => "Totally-Not-FireFox"
            ]
        );
 
        $mockDB1->expects( $this->exactly( 1 ) )->method( 'writesOrCallbacksPending' );
        $mockDB1->expects( $this->exactly( 1 ) )->method( 'lastDoneWrites' );
        $mockDB2->expects( $this->exactly( 1 ) )->method( 'writesOrCallbacksPending' );
        $mockDB2->expects( $this->exactly( 1 ) )->method( 'lastDoneWrites' );
 
        // Nothing to wait for on first HTTP request start
        $cp->initLB( $lb1 );
        $cp->initLB( $lb2 );
        // Record positions in stash on first HTTP request end
        $cp->shutdownLB( $lb1 );
        $cp->shutdownLB( $lb2 );
        $cpIndex = null;
        $cp->shutdown( null, 'sync', $cpIndex );
 
        $this->assertEquals( 1, $cpIndex, "CP write index set" );
 
        // (b) Second HTTP request
 
        // Load balancer for master DB 1
        $lb1 = $this->getMockBuilder( LoadBalancer::class )
            ->disableOriginalConstructor()
            ->getMock();
        $lb1->method( 'getServerCount' )->willReturn( 2 );
        $lb1->method( 'getServerName' )->with( 0 )->willReturn( 'master1' );
        $lb1->expects( $this->once() )
            ->method( 'waitFor' )->with( $this->equalTo( $m1Pos ) );
        // Load balancer for master DB 2
        $lb2 = $this->getMockBuilder( LoadBalancer::class )
            ->disableOriginalConstructor()
            ->getMock();
        $lb2->method( 'getServerCount' )->willReturn( 2 );
        $lb2->method( 'getServerName' )->with( 0 )->willReturn( 'master2' );
        $lb2->expects( $this->once() )
            ->method( 'waitFor' )->with( $this->equalTo( $m2Pos ) );
 
        $cp = new ChronologyProtector(
            $bag,
            [
                'ip' => '127.0.0.1',
                'agent' => "Totally-Not-FireFox"
            ],
            $cpIndex
        );
 
        // Wait for last positions to be reached on second HTTP request start
        $cp->initLB( $lb1 );
        $cp->initLB( $lb2 );
        // Shutdown (nothing to record)
        $cp->shutdownLB( $lb1 );
        $cp->shutdownLB( $lb2 );
        $cpIndex = null;
        $cp->shutdown( null, 'sync', $cpIndex );
 
        $this->assertEquals( null, $cpIndex, "CP write index retained" );
 
        $this->assertEquals( '45e93a9c215c031d38b7c42d8e4700ca', $cp->getClientId() );
    }
 
    private function newLBFactoryMulti( array $baseOverride = [], array $serverOverride = [] ) {
        global $wgDBserver, $wgDBuser, $wgDBpassword, $wgDBname, $wgDBprefix, $wgDBtype;
        global $wgSQLiteDataDir;
 
        return new LBFactoryMulti( $baseOverride + [
            'sectionsByDB' => [],
            'sectionLoads' => [
                'DEFAULT' => [
                    'test-db1' => 1,
                ],
            ],
            'serverTemplate' => $serverOverride + [
                'dbname' => $wgDBname,
                'tablePrefix' => $wgDBprefix,
                'user' => $wgDBuser,
                'password' => $wgDBpassword,
                'type' => $wgDBtype,
                'dbDirectory' => $wgSQLiteDataDir,
                'flags' => DBO_DEFAULT
            ],
            'hostsByName' => [
                'test-db1' => $wgDBserver,
            ],
            'loadMonitorClass' => LoadMonitorNull::class,
            'localDomain' => new DatabaseDomain( $wgDBname, null, $wgDBprefix ),
            'agent' => 'MW-UNIT-TESTS'
        ] );
    }
 
    public function testNiceDomains() {
        global $wgDBname;
 
        if ( wfGetDB( DB_MASTER )->databasesAreIndependent() ) {
            self::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
            return;
        }
 
        $factory = $this->newLBFactoryMulti(
            [],
            []
        );
        $lb = $factory->getMainLB();
 
        $db = $lb->getConnectionRef( DB_MASTER );
        $this->assertEquals(
            wfWikiID(),
            $db->getDomainID()
        );
        unset( $db );
 
        $db = $lb->getConnection( DB_MASTER, [], '' );
 
        $this->assertEquals(
            '',
            $db->getDomainId(),
            'Null domain ID handle used'
        );
        $this->assertEquals(
            '',
            $db->getDBname(),
            'Null domain ID handle used'
        );
        $this->assertEquals(
            '',
            $db->tablePrefix(),
            'Main domain ID handle used; prefix is empty though'
        );
        $this->assertEquals(
            $this->quoteTable( $db, 'page' ),
            $db->tableName( 'page' ),
            "Correct full table name"
        );
        $this->assertEquals(
            $this->quoteTable( $db, $wgDBname ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( "$wgDBname.page" ),
            "Correct full table name"
        );
        $this->assertEquals(
            $this->quoteTable( $db, 'nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( 'nice_db.page' ),
            "Correct full table name"
        );
 
        $lb->reuseConnection( $db ); // don't care
 
        $db = $lb->getConnection( DB_MASTER ); // local domain connection
        $factory->setLocalDomainPrefix( 'my_' );
 
        $this->assertEquals( $wgDBname, $db->getDBname() );
        $this->assertEquals(
            "$wgDBname-my_",
            $db->getDomainID()
        );
        $this->assertEquals(
            $this->quoteTable( $db, 'my_page' ),
            $db->tableName( 'page' ),
            "Correct full table name"
        );
        $this->assertEquals(
            $this->quoteTable( $db, 'other_nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( 'other_nice_db.page' ),
            "Correct full table name"
        );
 
        $factory->closeAll();
        $factory->destroy();
    }
 
    public function testTrickyDomain() {
        global $wgDBname;
 
        if ( wfGetDB( DB_MASTER )->databasesAreIndependent() ) {
            self::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
            return;
        }
 
        $dbname = 'unittest-domain'; // explodes if DB is selected
        $factory = $this->newLBFactoryMulti(
            [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
            [
                'dbname' => 'do_not_select_me' // explodes if DB is selected
            ]
        );
        $lb = $factory->getMainLB();
        $db = $lb->getConnection( DB_MASTER, [], '' );
 
        $this->assertEquals( '', $db->getDomainID(), "Null domain used" );
 
        $this->assertEquals(
            $this->quoteTable( $db, 'page' ),
            $db->tableName( 'page' ),
            "Correct full table name"
        );
 
        $this->assertEquals(
            $this->quoteTable( $db, $dbname ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( "$dbname.page" ),
            "Correct full table name"
        );
 
        $this->assertEquals(
            $this->quoteTable( $db, 'nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( 'nice_db.page' ),
            "Correct full table name"
        );
 
        $lb->reuseConnection( $db ); // don't care
 
        $factory->setLocalDomainPrefix( 'my_' );
        $db = $lb->getConnection( DB_MASTER, [], "$wgDBname-my_" );
 
        $this->assertEquals(
            $this->quoteTable( $db, 'my_page' ),
            $db->tableName( 'page' ),
            "Correct full table name"
        );
        $this->assertEquals(
            $this->quoteTable( $db, 'other_nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( 'other_nice_db.page' ),
            "Correct full table name"
        );
        $this->assertEquals(
            $this->quoteTable( $db, 'garbage-db' ) . '.' . $this->quoteTable( $db, 'page' ),
            $db->tableName( 'garbage-db.page' ),
            "Correct full table name"
        );
 
        $lb->reuseConnection( $db ); // don't care
 
        $factory->closeAll();
        $factory->destroy();
    }
 
    public function testInvalidSelectDB() {
        if ( wfGetDB( DB_MASTER )->databasesAreIndependent() ) {
            $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
        }
 
        $dbname = 'unittest-domain'; // explodes if DB is selected
        $factory = $this->newLBFactoryMulti(
            [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
            [
                'dbname' => 'do_not_select_me' // explodes if DB is selected
            ]
        );
        $lb = $factory->getMainLB();
        $db = $lb->getConnection( DB_MASTER, [], '' );
 
        \Wikimedia\suppressWarnings();
        try {
            $this->assertFalse( $db->selectDB( 'garbage-db' ) );
            $this->fail( "No error thrown." );
        } catch ( \Wikimedia\Rdbms\DBQueryError $e ) {
            $this->assertRegExp( '/[\'"]garbage-db[\'"]/', $e->getMessage() );
        }
        \Wikimedia\restoreWarnings();
    }
 
    public function testInvalidSelectDBIndependant() {
        $dbname = 'unittest-domain'; // explodes if DB is selected
        $factory = $this->newLBFactoryMulti(
            [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
            [
                'dbname' => 'do_not_select_me' // explodes if DB is selected
            ]
        );
        $lb = $factory->getMainLB();
 
        if ( !wfGetDB( DB_MASTER )->databasesAreIndependent() ) {
            $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
        }
 
        $lb->getConnection( DB_MASTER, [], '' );
    }
 
    public function testInvalidSelectDBIndependant2() {
        $dbname = 'unittest-domain'; // explodes if DB is selected
        $factory = $this->newLBFactoryMulti(
            [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
            [
                'dbname' => 'do_not_select_me' // explodes if DB is selected
            ]
        );
        $lb = $factory->getMainLB();
 
        if ( !wfGetDB( DB_MASTER )->databasesAreIndependent() ) {
            $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
        }
 
        $db = $lb->getConnection( DB_MASTER );
        \Wikimedia\suppressWarnings();
        $db->selectDB( 'garbage-db' );
        \Wikimedia\restoreWarnings();
    }
 
    public function testRedefineLocalDomain() {
        global $wgDBname;
 
        if ( wfGetDB( DB_MASTER )->databasesAreIndependent() ) {
            self::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
            return;
        }
 
        $factory = $this->newLBFactoryMulti(
            [],
            []
        );
        $lb = $factory->getMainLB();
 
        $conn1 = $lb->getConnectionRef( DB_MASTER );
        $this->assertEquals(
            wfWikiID(),
            $conn1->getDomainID()
        );
        unset( $conn1 );
 
        $factory->redefineLocalDomain( 'somedb-prefix_' );
        $this->assertEquals( 'somedb-prefix_', $factory->getLocalDomainID() );
 
        $domain = new DatabaseDomain( $wgDBname, null, 'pref_' );
        $factory->redefineLocalDomain( $domain );
 
        $n = 0;
        $lb->forEachOpenConnection( function () use ( &$n ) {
            ++$n;
        } );
        $this->assertEquals( 0, $n, "Connections closed" );
 
        $conn2 = $lb->getConnectionRef( DB_MASTER );
        $this->assertEquals(
            $domain->getId(),
            $conn2->getDomainID()
        );
        unset( $conn2 );
 
        $factory->closeAll();
        $factory->destroy();
    }
 
    private function quoteTable( IDatabase $db, $table ) {
        if ( $db->getType() === 'sqlite' ) {
            return $table;
        } else {
            return $db->addIdentifierQuotes( $table );
        }
    }
 
    public function testCPPosIndexCookieValues() {
        $time = 1526522031;
        $agentId = md5( 'Ramsey\'s Loyal Presa Canario' );
 
        $this->assertEquals(
            '3@542#c47dcfb0566e7d7bc110a6128a45c93a',
            LBFactory::makeCookieValueFromCPIndex( 3, 542, $agentId )
        );
 
        $lbFactory = $this->newLBFactoryMulti();
        $lbFactory->setRequestInfo( [ 'IPAddress' => '10.64.24.52', 'UserAgent' => 'meow' ] );
        $this->assertEquals(
            '1@542#c47dcfb0566e7d7bc110a6128a45c93a',
            LBFactory::makeCookieValueFromCPIndex( 1, 542, $agentId )
        );
 
        $this->assertSame(
            null,
            LBFactory::getCPInfoFromCookieValue( "5#$agentId", $time - 10 )['index'],
            'No time set'
        );
        $this->assertSame(
            null,
            LBFactory::getCPInfoFromCookieValue( "5@$time", $time - 10 )['index'],
            'No agent set'
        );
        $this->assertSame(
            null,
            LBFactory::getCPInfoFromCookieValue( "0@$time#$agentId", $time - 10 )['index'],
            'Bad index'
        );
 
        $this->assertSame(
            2,
            LBFactory::getCPInfoFromCookieValue( "2@$time#$agentId", $time - 10 )['index'],
            'Fresh'
        );
        $this->assertSame(
            2,
            LBFactory::getCPInfoFromCookieValue( "2@$time#$agentId", $time + 9 - 10 )['index'],
            'Almost stale'
        );
        $this->assertSame(
            null,
            LBFactory::getCPInfoFromCookieValue( "0@$time#$agentId", $time + 9 - 10 )['index'],
            'Almost stale; bad index'
        );
        $this->assertSame(
            null,
            LBFactory::getCPInfoFromCookieValue( "2@$time#$agentId", $time + 11 - 10 )['index'],
            'Stale'
        );
 
        $this->assertSame(
            $agentId,
            LBFactory::getCPInfoFromCookieValue( "5@$time#$agentId", $time - 10 )['clientId'],
            'Live (client ID)'
        );
        $this->assertSame(
            null,
            LBFactory::getCPInfoFromCookieValue( "5@$time#$agentId", $time + 11 - 10 )['clientId'],
            'Stale (client ID)'
        );
    }
}