tests/phpunit/includes/CdbTest.php

   1 <?php
   2
   3 /**
   4  * Test the CDB reader/writer
   5  * @covers CdbWriter_PHP
   6  * @covers CdbWriter_DBA
   7  */
   8 class CdbTest extends MediaWikiTestCase {
   9
  10         protected function setUp() {
  11                 parent::setUp();
  12                 if ( !CdbReader::haveExtension() ) {
  13                         $this->markTestSkipped( 'Native CDB support is not available' );
  14                 }
  15         }
  16
  17         /**
  18          * @group medium
  19          */
  20         public function testCdb() {
  21                 $dir = wfTempDir();
  22                 if ( !is_writable( $dir ) ) {
  23                         $this->markTestSkipped( "Temp dir isn't writable" );
  24                 }
  25
  26                 $phpcdbfile = $this->getNewTempFile();
  27                 $dbacdbfile = $this->getNewTempFile();
  28
  29                 $w1 = new CdbWriter_PHP( $phpcdbfile );
  30                 $w2 = new CdbWriter_DBA( $dbacdbfile );
  31
  32                 $data = array();
  33                 for ( $i = 0; $i < 1000; $i++ ) {
  34                         $key = $this->randomString();
  35                         $value = $this->randomString();
  36                         $w1->set( $key, $value );
  37                         $w2->set( $key, $value );
  38
  39                         if ( !isset( $data[$key] ) ) {
  40                                 $data[$key] = $value;
  41                         }
  42                 }
  43
  44                 $w1->close();
  45                 $w2->close();
  46
  47                 $this->assertEquals(
  48                         md5_file( $phpcdbfile ),
  49                         md5_file( $dbacdbfile ),
  50                         'same hash'
  51                 );
  52
  53                 $r1 = new CdbReader_PHP( $phpcdbfile );
  54                 $r2 = new CdbReader_DBA( $dbacdbfile );
  55
  56                 foreach ( $data as $key => $value ) {
  57                         if ( $key === '' ) {
  58                                 // Known bug
  59                                 continue;
  60                         }
  61                         $v1 = $r1->get( $key );
  62                         $v2 = $r2->get( $key );
  63
  64                         $v1 = $v1 === false ? '(not found)' : $v1;
  65                         $v2 = $v2 === false ? '(not found)' : $v2;
  66
  67                         # cdbAssert( 'Mismatch', $key, $v1, $v2 );
  68                         $this->cdbAssert( "PHP error", $key, $v1, $value );
  69                         $this->cdbAssert( "DBA error", $key, $v2, $value );
  70                 }
  71         }
  72
  73         private function randomString() {
  74                 $len = mt_rand( 0, 10 );
  75                 $s = '';
  76                 for ( $j = 0; $j < $len; $j++ ) {
  77                         $s .= chr( mt_rand( 0, 255 ) );
  78                 }
  79
  80                 return $s;
  81         }
  82
  83         private function cdbAssert( $msg, $key, $v1, $v2 ) {
  84                 $this->assertEquals(
  85                         $v2,
  86                         $v1,
  87                         $msg . ', k=' . bin2hex( $key )
  88                 );
  89         }
  90 }