maintenance/tests/CdbTest.php

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