backupPrefetchTest.php

Go to the documentation of this file.

<?php

require_once __DIR__ . "/../../../maintenance/backupPrefetch.inc";

class BaseDumpTest extends MediaWikiTestCase {

        private $dump = null;

        protected function tearDown() {
                if ( $this->dump !== null ) {
                        $this->dump->close();
                }

                // Bug 37458, parent teardown need to be done after closing the
                // dump or it might cause some permissions errors.
                parent::tearDown();
        }

        private function assertPrefetchEquals( $expected, $page, $revision ) {
                $this->assertEquals( $expected, $this->dump->prefetch( $page, $revision ),
                        "Prefetch of page $page revision $revision" );

        }

        function testSequential() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( "BackupDumperTestP1Text1", 1, 1 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text4 some additional Text", 2, 5 );
                $this->assertPrefetchEquals( "Talk about BackupDumperTestP1 Text1", 4, 8 );
        }

        function testSynchronizeRevisionMissToRevision() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( null, 2, 3 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text4 some additional Text", 2, 5 );
        }

        function testSynchronizeRevisionMissToPage() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( null, 2, 40 );
                $this->assertPrefetchEquals( "Talk about BackupDumperTestP1 Text1", 4, 8 );
        }

        function testSynchronizePageMiss() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( null, 3, 40 );
                $this->assertPrefetchEquals( "Talk about BackupDumperTestP1 Text1", 4, 8 );
        }

        function testPageMissAtEnd() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( null, 6, 40 );
        }

        function testRevisionMissAtEnd() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( null, 4, 40 );
        }

        function testSynchronizePageMissAtStart() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( null, 0, 2 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
        }

        function testSynchronizeRevisionMissAtStart() {
                $fname = $this->setUpPrefetch();
                $this->dump = new BaseDump( $fname );

                $this->assertPrefetchEquals( null, 1, -2 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
        }

        function testSequentialAcrossFiles() {
                $fname1 = $this->setUpPrefetch( array( 1 ) );
                $fname2 = $this->setUpPrefetch( array( 2, 4 ) );
                $this->dump = new BaseDump( $fname1 . ";" . $fname2 );

                $this->assertPrefetchEquals( "BackupDumperTestP1Text1", 1, 1 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
                $this->assertPrefetchEquals( "BackupDumperTestP2Text4 some additional Text", 2, 5 );
                $this->assertPrefetchEquals( "Talk about BackupDumperTestP1 Text1", 4, 8 );
        }

        function testSynchronizeSkipAcrossFile() {
                $fname1 = $this->setUpPrefetch( array( 1 ) );
                $fname2 = $this->setUpPrefetch( array( 2 ) );
                $fname3 = $this->setUpPrefetch( array( 4 ) );
                $this->dump = new BaseDump( $fname1 . ";" . $fname2 . ";" . $fname3 );

                $this->assertPrefetchEquals( "BackupDumperTestP1Text1", 1, 1 );
                $this->assertPrefetchEquals( "Talk about BackupDumperTestP1 Text1", 4, 8 );
        }

        function testSynchronizeMissInWholeFirstFile() {
                $fname1 = $this->setUpPrefetch( array( 1 ) );
                $fname2 = $this->setUpPrefetch( array( 2 ) );
                $this->dump = new BaseDump( $fname1 . ";" . $fname2 );

                $this->assertPrefetchEquals( "BackupDumperTestP2Text1", 2, 2 );
        }


        private function setUpPrefetch( $requested_pages = array( 1, 2, 4 ) ) {
                // The file name, where we store the prepared prefetch file
                $fname = $this->getNewTempFile();

                // The header of every prefetch file
                $header = '<mediawiki xmlns="http://www.mediawiki.org/xml/export-0.7/" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.mediawiki.org/xml/export-0.7/ http://www.mediawiki.org/xml/export-0.7.xsd" version="0.7" xml:lang="en">
  <siteinfo>
    <sitename>wikisvn</sitename>
    <base>http://localhost/wiki-svn/index.php/Main_Page</base>
    <generator>MediaWiki 1.21alpha</generator>
    <case>first-letter</case>
    <namespaces>
      <namespace key="-2" case="first-letter">Media</namespace>
      <namespace key="-1" case="first-letter">Special</namespace>
      <namespace key="0" case="first-letter" />
      <namespace key="1" case="first-letter">Talk</namespace>
      <namespace key="2" case="first-letter">User</namespace>
      <namespace key="3" case="first-letter">User talk</namespace>
      <namespace key="4" case="first-letter">Wikisvn</namespace>
      <namespace key="5" case="first-letter">Wikisvn talk</namespace>
      <namespace key="6" case="first-letter">File</namespace>
      <namespace key="7" case="first-letter">File talk</namespace>
      <namespace key="8" case="first-letter">MediaWiki</namespace>
      <namespace key="9" case="first-letter">MediaWiki talk</namespace>
      <namespace key="10" case="first-letter">Template</namespace>
      <namespace key="11" case="first-letter">Template talk</namespace>
      <namespace key="12" case="first-letter">Help</namespace>
      <namespace key="13" case="first-letter">Help talk</namespace>
      <namespace key="14" case="first-letter">Category</namespace>
      <namespace key="15" case="first-letter">Category talk</namespace>
    </namespaces>
  </siteinfo>
';


                // An array holding the pages that are available for prefetch
                $available_pages = array();

                // Simple plain page
                $available_pages[1] = '  <page>
    <title>BackupDumperTestP1</title>
    <ns>0</ns>
    <id>1</id>
    <revision>
      <id>1</id>
      <timestamp>2012-04-01T16:46:05Z</timestamp>
      <contributor>
        <ip>127.0.0.1</ip>
      </contributor>
      <comment>BackupDumperTestP1Summary1</comment>
      <sha1>0bolhl6ol7i6x0e7yq91gxgaan39j87</sha1>
      <text xml:space="preserve">BackupDumperTestP1Text1</text>
      <model name="wikitext">1</model>
      <format mime="text/x-wiki">1</format>
    </revision>
  </page>
';
                // Page with more than one revisions. Hole in rev ids.
                $available_pages[2] = '  <page>
    <title>BackupDumperTestP2</title>
    <ns>0</ns>
    <id>2</id>
    <revision>
      <id>2</id>
      <timestamp>2012-04-01T16:46:05Z</timestamp>
      <contributor>
        <ip>127.0.0.1</ip>
      </contributor>
      <comment>BackupDumperTestP2Summary1</comment>
      <sha1>jprywrymfhysqllua29tj3sc7z39dl2</sha1>
      <text xml:space="preserve">BackupDumperTestP2Text1</text>
      <model name="wikitext">1</model>
      <format mime="text/x-wiki">1</format>
    </revision>
    <revision>
      <id>5</id>
      <parentid>2</parentid>
      <timestamp>2012-04-01T16:46:05Z</timestamp>
      <contributor>
        <ip>127.0.0.1</ip>
      </contributor>
      <comment>BackupDumperTestP2Summary4 extra</comment>
      <sha1>6o1ciaxa6pybnqprmungwofc4lv00wv</sha1>
      <text xml:space="preserve">BackupDumperTestP2Text4 some additional Text</text>
      <model name="wikitext">1</model>
      <format mime="text/x-wiki">1</format>
    </revision>
  </page>
';
                // Page with id higher than previous id + 1
                $available_pages[4] = '  <page>
    <title>Talk:BackupDumperTestP1</title>
    <ns>1</ns>
    <id>4</id>
    <revision>
      <id>8</id>
      <timestamp>2012-04-01T16:46:05Z</timestamp>
      <contributor>
        <ip>127.0.0.1</ip>
      </contributor>
      <comment>Talk BackupDumperTestP1 Summary1</comment>
      <sha1>nktofwzd0tl192k3zfepmlzxoax1lpe</sha1>
      <model name="wikitext">1</model>
      <format mime="text/x-wiki">1</format>
      <text xml:space="preserve">Talk about BackupDumperTestP1 Text1</text>
    </revision>
  </page>
';

                // The common ending for all files
                $tail = '</mediawiki>
';

                // Putting together the content of the prefetch files
                $content = $header;
                foreach ( $requested_pages as $i ) {
                        $this->assertTrue( array_key_exists( $i, $available_pages ),
                                "Check for availability of requested page " . $i );
                        $content .= $available_pages[$i];
                }
                $content .= $tail;

                $this->assertEquals( strlen( $content ), file_put_contents(
                        $fname, $content ), "Length of prepared prefetch" );

                return $fname;
        }

}