Make WANObjectCache sets account for slave lag

[lhc/web/wiklou.git] / tests / phpunit / includes / utils / BatchRowUpdateTest.php

diff --git a/tests/phpunit/includes/utils/BatchRowUpdateTest.php b/tests/phpunit/includes/utils/BatchRowUpdateTest.php
index 082ac82..d224af8 100644 (file)
--- a/tests/phpunit/includes/utils/BatchRowUpdateTest.php
+++ b/tests/phpunit/includes/utils/BatchRowUpdateTest.php
@@ -238,12 +238,16 @@ class BatchRowUpdateTest extends MediaWikiTestCase {
        protected function mockDb() {
                // Cant mock from DatabaseType or DatabaseBase, they dont
                // have the full gamut of methods
+               // FIXME: the constructor normally sets mAtomicLevels and mSrvCache
                $databaseMysql = $this->getMockBuilder( 'DatabaseMysql' )
                        ->disableOriginalConstructor()
                        ->getMock();
                $databaseMysql->expects( $this->any() )
                        ->method( 'isOpen' )
                        ->will( $this->returnValue( true ) );
+               $databaseMysql->expects( $this->any() )
+                       ->method( 'getApproximateLagStatus' )
+                       ->will( $this->returnValue( array( 'lag' => 0, 'since' => 0 ) ) );
                return $databaseMysql;
        }
 }