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                 $w1 = new CdbWriter_PHP( 'php.cdb' );
  17                 $w2 = new CdbWriter_DBA( 'dba.cdb' );
  18
  19                 $data = array();
  20                 for ( $i = 0; $i < 1000; $i++ ) {
  21                         $key = $this->randomString();
  22                         $value = $this->randomString();
  23                         $w1->set( $key, $value );
  24                         $w2->set( $key, $value );
  25
  26                         if ( !isset( $data[$key] ) ) {
  27                                 $data[$key] = $value;
  28                         }
  29                 }
  30
  31                 $w1->close();
  32                 $w2->close();
  33
  34                 $this->assertEquals(
  35                         md5_file( 'dba.cdb' ),
  36                         md5_file( 'php.cdb' ),
  37                         'same hash'
  38                 );
  39
  40                 $r1 = new CdbReader_PHP( 'php.cdb' );
  41                 $r2 = new CdbReader_DBA( 'dba.cdb' );
  42
  43                 foreach ( $data as $key => $value ) {
  44                         if ( $key === '' ) {
  45                                 // Known bug
  46                                 continue;
  47                         }
  48                         $v1 = $r1->get( $key );
  49                         $v2 = $r2->get( $key );
  50
  51                         $v1 = $v1 === false ? '(not found)' : $v1;
  52                         $v2 = $v2 === false ? '(not found)' : $v2;
  53
  54                         #cdbAssert( 'Mismatch', $key, $v1, $v2 );
  55                         $this->cdbAssert( "PHP error", $key, $v1, $value );
  56                         $this->cdbAssert( "DBA error", $key, $v2, $value );
  57                 }
  58
  59                 unlink( 'dba.cdb' );
  60                 unlink( 'php.cdb' );
  61         }
  62
  63         private function randomString() {
  64                 $len = mt_rand( 0, 10 );
  65                 $s = '';
  66                 for ( $j = 0; $j < $len; $j++ ) {
  67                         $s .= chr( mt_rand( 0, 255 ) );
  68                 }
  69                 return $s;
  70         }
  71
  72         private function cdbAssert( $msg, $key, $v1, $v2 ) {
  73                 $this->assertEquals(
  74                         $v2,
  75                         $v1,
  76                         $msg . ', k=' . bin2hex( $key )
  77                 );
  78         }
  79 }