[p5sagit/p5-mst-13.2.git] / t / lib / db-btree.t

#!./perl -w

BEGIN {
    @INC = '../lib' if -d '../lib' ;
    require Config; import Config;
    if ($Config{'extensions'} !~ /\bDB_File\b/) {
	print "1..0\n";
	exit 0;
    }
}

use DB_File; 
use Fcntl;

print "1..102\n";

sub ok
{
    my $no = shift ;
    my $result = shift ;
 
    print "not " unless $result ;
    print "ok $no\n" ;
}

sub lexical
{
    my(@a) = unpack ("C*", $a) ;
    my(@b) = unpack ("C*", $b) ;

    my $len = (@a > @b ? @b : @a) ;
    my $i = 0 ;

    foreach $i ( 0 .. $len -1) {
        return $a[$i] - $b[$i] if $a[$i] != $b[$i] ;
    }

    return @a - @b ;
}

$Dfile = "dbbtree.tmp";
unlink $Dfile;

umask(0);

# Check the interface to BTREEINFO

my $dbh = new DB_File::BTREEINFO ;
ok(1, ! defined $dbh->{flags}) ;
ok(2, ! defined $dbh->{cachesize}) ;
ok(3, ! defined $dbh->{psize}) ;
ok(4, ! defined $dbh->{lorder}) ;
ok(5, ! defined $dbh->{minkeypage}) ;
ok(6, ! defined $dbh->{maxkeypage}) ;
ok(7, ! defined $dbh->{compare}) ;
ok(8, ! defined $dbh->{prefix}) ;

$dbh->{flags} = 3000 ;
ok(9, $dbh->{flags} == 3000) ;

$dbh->{cachesize} = 9000 ;
ok(10, $dbh->{cachesize} == 9000);

$dbh->{psize} = 400 ;
ok(11, $dbh->{psize} == 400) ;

$dbh->{lorder} = 65 ;
ok(12, $dbh->{lorder} == 65) ;

$dbh->{minkeypage} = 123 ;
ok(13, $dbh->{minkeypage} == 123) ;

$dbh->{maxkeypage} = 1234 ;
ok(14, $dbh->{maxkeypage} == 1234 );

$dbh->{compare} = 1234 ;
ok(15, $dbh->{compare} == 1234) ;

$dbh->{prefix} = 1234 ;
ok(16, $dbh->{prefix} == 1234 );

# Check that an invalid entry is caught both for store & fetch
eval '$dbh->{fred} = 1234' ;
ok(17, $@ =~ /^DB_File::BTREEINFO::STORE - Unknown element 'fred' at/ ) ;
eval '$q = $dbh->{fred}' ;
ok(18, $@ =~ /^DB_File::BTREEINFO::FETCH - Unknown element 'fred' at/ ) ;

# Now check the interface to BTREE

ok(19, $X = tie(%h, 'DB_File',$Dfile, O_RDWR|O_CREAT, 0640, $DB_BTREE )) ;

($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,$atime,$mtime,$ctime,
   $blksize,$blocks) = stat($Dfile);
ok(20, ($mode & 0777) == (($^O eq 'os2' || $^O eq 'MSWin32') ? 0666 : 0640) || $^O eq 'amigaos');

while (($key,$value) = each(%h)) {
    $i++;
}
ok(21, !$i ) ;

$h{'goner1'} = 'snork';

$h{'abc'} = 'ABC';
ok(22, $h{'abc'} eq 'ABC' );
ok(23, ! defined $h{'jimmy'} ) ;
ok(24, ! exists $h{'jimmy'} ) ;
ok(25,  defined $h{'abc'} ) ;

$h{'def'} = 'DEF';
$h{'jkl','mno'} = "JKL\034MNO";
$h{'a',2,3,4,5} = join("\034",'A',2,3,4,5);
$h{'a'} = 'A';

#$h{'b'} = 'B';
$X->STORE('b', 'B') ;

$h{'c'} = 'C';

#$h{'d'} = 'D';
$X->put('d', 'D') ;

$h{'e'} = 'E';
$h{'f'} = 'F';
$h{'g'} = 'X';
$h{'h'} = 'H';
$h{'i'} = 'I';

$h{'goner2'} = 'snork';
delete $h{'goner2'};


# IMPORTANT - $X must be undefined before the untie otherwise the
#             underlying DB close routine will not get called.
undef $X ;
untie(%h);


# tie to the same file again
ok(26, $X = tie(%h,'DB_File',$Dfile, O_RDWR, 0640, $DB_BTREE)) ;

# Modify an entry from the previous tie
$h{'g'} = 'G';

$h{'j'} = 'J';
$h{'k'} = 'K';
$h{'l'} = 'L';
$h{'m'} = 'M';
$h{'n'} = 'N';
$h{'o'} = 'O';
$h{'p'} = 'P';
$h{'q'} = 'Q';
$h{'r'} = 'R';
$h{'s'} = 'S';
$h{'t'} = 'T';
$h{'u'} = 'U';
$h{'v'} = 'V';
$h{'w'} = 'W';
$h{'x'} = 'X';
$h{'y'} = 'Y';
$h{'z'} = 'Z';

$h{'goner3'} = 'snork';

delete $h{'goner1'};
$X->DELETE('goner3');

@keys = keys(%h);
@values = values(%h);

ok(27, $#keys == 29 && $#values == 29) ;

$i = 0 ;
while (($key,$value) = each(%h)) {
    if ($key eq $keys[$i] && $value eq $values[$i] && $key eq lc($value)) {
	$key =~ y/a-z/A-Z/;
	$i++ if $key eq $value;
    }
}

ok(28, $i == 30) ;

@keys = ('blurfl', keys(%h), 'dyick');
ok(29, $#keys == 31) ;

#Check that the keys can be retrieved in order
my @b = keys %h ;
my @c = sort lexical @b ;
ok(30, ArrayCompare(\@b, \@c)) ;

$h{'foo'} = '';
ok(31, $h{'foo'} eq '' ) ;

$h{''} = 'bar';
ok(32, $h{''} eq 'bar' );

# check cache overflow and numeric keys and contents
$ok = 1;
for ($i = 1; $i < 200; $i++) { $h{$i + 0} = $i + 0; }
for ($i = 1; $i < 200; $i++) { $ok = 0 unless $h{$i} == $i; }
ok(33, $ok);

($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,$atime,$mtime,$ctime,
   $blksize,$blocks) = stat($Dfile);
ok(34, $size > 0 );

@h{0..200} = 200..400;
@foo = @h{0..200};
ok(35, join(':',200..400) eq join(':',@foo) );

# Now check all the non-tie specific stuff


# Check R_NOOVERWRITE flag will make put fail when attempting to overwrite
# an existing record.
 
$status = $X->put( 'x', 'newvalue', R_NOOVERWRITE) ;
ok(36, $status == 1 );
 
# check that the value of the key 'x' has not been changed by the 
# previous test
ok(37, $h{'x'} eq 'X' );

# standard put
$status = $X->put('key', 'value') ;
ok(38, $status == 0 );

#check that previous put can be retrieved
$value = 0 ;
$status = $X->get('key', $value) ;
ok(39, $status == 0 );
ok(40, $value eq 'value' );

# Attempting to delete an existing key should work

$status = $X->del('q') ;
ok(41, $status == 0 );
$status = $X->del('') ;
ok(42, $status == 0 );

# Make sure that the key deleted, cannot be retrieved
ok(43, ! defined $h{'q'}) ;
ok(44, ! defined $h{''}) ;

undef $X ;
untie %h ;

ok(45, $X = tie(%h, 'DB_File',$Dfile, O_RDWR, 0640, $DB_BTREE ));

# Attempting to delete a non-existant key should fail

$status = $X->del('joe') ;
ok(46, $status == 1 );

# Check the get interface

# First a non-existing key
$status = $X->get('aaaa', $value) ;
ok(47, $status == 1 );

# Next an existing key
$status = $X->get('a', $value) ;
ok(48, $status == 0 );
ok(49, $value eq 'A' );

# seq
# ###

# use seq to find an approximate match
$key = 'ke' ;
$value = '' ;
$status = $X->seq($key, $value, R_CURSOR) ;
ok(50, $status == 0 );
ok(51, $key eq 'key' );
ok(52, $value eq 'value' );

# seq when the key does not match
$key = 'zzz' ;
$value = '' ;
$status = $X->seq($key, $value, R_CURSOR) ;
ok(53, $status == 1 );


# use seq to set the cursor, then delete the record @ the cursor.

$key = 'x' ;
$value = '' ;
$status = $X->seq($key, $value, R_CURSOR) ;
ok(54, $status == 0 );
ok(55, $key eq 'x' );
ok(56, $value eq 'X' );
$status = $X->del(0, R_CURSOR) ;
ok(57, $status == 0 );
$status = $X->get('x', $value) ;
ok(58, $status == 1 );

# ditto, but use put to replace the key/value pair.
$key = 'y' ;
$value = '' ;
$status = $X->seq($key, $value, R_CURSOR) ;
ok(59, $status == 0 );
ok(60, $key eq 'y' );
ok(61, $value eq 'Y' );

$key = "replace key" ;
$value = "replace value" ;
$status = $X->put($key, $value, R_CURSOR) ;
ok(62, $status == 0 );
ok(63, $key eq 'replace key' );
ok(64, $value eq 'replace value' );
$status = $X->get('y', $value) ;
ok(65, 1) ; # hard-wire to always pass. the previous test ($status == 1)
            # only worked because of a bug in 1.85/6


# use seq to walk forwards through a file 

$status = $X->seq($key, $value, R_FIRST) ;
ok(66, $status == 0 );
$previous = $key ;

$ok = 1 ;
while (($status = $X->seq($key, $value, R_NEXT)) == 0)
{
    ($ok = 0), last if ($previous cmp $key) == 1 ;
}

ok(67, $status == 1 );
ok(68, $ok == 1 );

# use seq to walk backwards through a file 
$status = $X->seq($key, $value, R_LAST) ;
ok(69, $status == 0 );
$previous = $key ;

$ok = 1 ;
while (($status = $X->seq($key, $value, R_PREV)) == 0)
{
    ($ok = 0), last if ($previous cmp $key) == -1 ;
    #print "key = [$key] value = [$value]\n" ;
}

ok(70, $status == 1 );
ok(71, $ok == 1 );


# check seq FIRST/LAST

# sync
# ####

$status = $X->sync ;
ok(72, $status == 0 );


# fd
# ##

$status = $X->fd ;
ok(73, $status != 0 );


undef $X ;
untie %h ;

unlink $Dfile;

# Now try an in memory file
ok(74, $Y = tie(%h, 'DB_File',undef, O_RDWR|O_CREAT, 0640, $DB_BTREE ));

# fd with an in memory file should return failure
$status = $Y->fd ;
ok(75, $status == -1 );


undef $Y ;
untie %h ;

# Duplicate keys
my $bt = new DB_File::BTREEINFO ;
$bt->{flags} = R_DUP ;
ok(76, $YY = tie(%hh, 'DB_File', $Dfile, O_RDWR|O_CREAT, 0640, $bt )) ;

$hh{'Wall'} = 'Larry' ;
$hh{'Wall'} = 'Stone' ; # Note the duplicate key
$hh{'Wall'} = 'Brick' ; # Note the duplicate key
$hh{'Wall'} = 'Brick' ; # Note the duplicate key and value
$hh{'Smith'} = 'John' ;
$hh{'mouse'} = 'mickey' ;

# first work in scalar context
ok(77, scalar $YY->get_dup('Unknown') == 0 );
ok(78, scalar $YY->get_dup('Smith') == 1 );
ok(79, scalar $YY->get_dup('Wall') == 4 );

# now in list context
my @unknown = $YY->get_dup('Unknown') ;
ok(80, "@unknown" eq "" );

my @smith = $YY->get_dup('Smith') ;
ok(81, "@smith" eq "John" );

{
my @wall = $YY->get_dup('Wall') ;
my %wall ;
@wall{@wall} = @wall ;
ok(82, (@wall == 4 && $wall{'Larry'} && $wall{'Stone'} && $wall{'Brick'}) );
}

# hash
my %unknown = $YY->get_dup('Unknown', 1) ;
ok(83, keys %unknown == 0 );

my %smith = $YY->get_dup('Smith', 1) ;
ok(84, keys %smith == 1 && $smith{'John'}) ;

my %wall = $YY->get_dup('Wall', 1) ;
ok(85, keys %wall == 3 && $wall{'Larry'} == 1 && $wall{'Stone'} == 1 
		&& $wall{'Brick'} == 2);

undef $YY ;
untie %hh ;
unlink $Dfile;


# test multiple callbacks
$Dfile1 = "btree1" ;
$Dfile2 = "btree2" ;
$Dfile3 = "btree3" ;
 
$dbh1 = new DB_File::BTREEINFO ;
{ local $^W = 0 ;
  $dbh1->{compare} = sub { $_[0] <=> $_[1] } ; }
 
$dbh2 = new DB_File::BTREEINFO ;
$dbh2->{compare} = sub { $_[0] cmp $_[1] } ;
 
$dbh3 = new DB_File::BTREEINFO ;
$dbh3->{compare} = sub { length $_[0] <=> length $_[1] } ;
 
 
tie(%h, 'DB_File',$Dfile1, O_RDWR|O_CREAT, 0640, $dbh1 ) ;
tie(%g, 'DB_File',$Dfile2, O_RDWR|O_CREAT, 0640, $dbh2 ) ;
tie(%k, 'DB_File',$Dfile3, O_RDWR|O_CREAT, 0640, $dbh3 ) ;
 
@Keys = qw( 0123 12 -1234 9 987654321 def  ) ;
{ local $^W = 0 ;
  @srt_1 = sort { $a <=> $b } @Keys ; }
@srt_2 = sort { $a cmp $b } @Keys ;
@srt_3 = sort { length $a <=> length $b } @Keys ;
 
foreach (@Keys) {
    { local $^W = 0 ; 
      $h{$_} = 1 ; }
    $g{$_} = 1 ;
    $k{$_} = 1 ;
}
 
sub ArrayCompare
{
    my($a, $b) = @_ ;
 
    return 0 if @$a != @$b ;
 
    foreach (1 .. length @$a)
    {
        return 0 unless $$a[$_] eq $$b[$_] ;
    }
 
    1 ;
}
 
ok(86, ArrayCompare (\@srt_1, [keys %h]) );
ok(87, ArrayCompare (\@srt_2, [keys %g]) );
ok(88, ArrayCompare (\@srt_3, [keys %k]) );

untie %h ;
untie %g ;
untie %k ;
unlink $Dfile1, $Dfile2, $Dfile3 ;

# clear
# #####

ok(89, tie(%h, 'DB_File', $Dfile1, O_RDWR|O_CREAT, 0640, $DB_BTREE ) );
foreach (1 .. 10)
  { $h{$_} = $_ * 100 }

# check that there are 10 elements in the hash
$i = 0 ;
while (($key,$value) = each(%h)) {
    $i++;
}
ok(90, $i == 10);

# now clear the hash
%h = () ;

# check it is empty
$i = 0 ;
while (($key,$value) = each(%h)) {
    $i++;
}
ok(91, $i == 0);

untie %h ;
unlink $Dfile1 ;

{
    # check that attempting to tie an array to a DB_BTREE will fail

    my $filename = "xyz" ;
    my @x ;
    eval { tie @x, 'DB_File', $filename, O_RDWR|O_CREAT, 0640, $DB_BTREE ; } ;
    ok(92, $@ =~ /^DB_File can only tie an associative array to a DB_BTREE database/) ;
    unlink $filename ;
}


{
   # sub-class test

   package Another ;

   use strict ;

   open(FILE, ">SubDB.pm") or die "Cannot open SubDB.pm: $!\n" ;
   print FILE <<'EOM' ;

   package SubDB ;

   use strict ;
   use vars qw( @ISA @EXPORT) ;

   require Exporter ;
   use DB_File;
   @ISA=qw(DB_File);
   @EXPORT = @DB_File::EXPORT ;

   sub STORE { 
	my $self = shift ;
        my $key = shift ;
        my $value = shift ;
        $self->SUPER::STORE($key, $value * 2) ;
   }

   sub FETCH { 
	my $self = shift ;
        my $key = shift ;
        $self->SUPER::FETCH($key) - 1 ;
   }

   sub put { 
	my $self = shift ;
        my $key = shift ;
        my $value = shift ;
        $self->SUPER::put($key, $value * 3) ;
   }

   sub get { 
	my $self = shift ;
        $self->SUPER::get($_[0], $_[1]) ;
	$_[1] -= 2 ;
   }

   sub A_new_method
   {
	my $self = shift ;
        my $key = shift ;
        my $value = $self->FETCH($key) ;
	return "[[$value]]" ;
   }

   1 ;
EOM

    close FILE ;

    BEGIN { push @INC, '.'; }
    eval 'use SubDB ; ';
    main::ok(93, $@ eq "") ;
    my %h ;
    my $X ;
    eval '
	$X = tie(%h, "SubDB","dbbtree.tmp", O_RDWR|O_CREAT, 0640, $DB_BTREE );
	' ;

    main::ok(94, $@ eq "") ;

    my $ret = eval '$h{"fred"} = 3 ; return $h{"fred"} ' ;
    main::ok(95, $@ eq "") ;
    main::ok(96, $ret == 5) ;

    my $value = 0;
    $ret = eval '$X->put("joe", 4) ; $X->get("joe", $value) ; return $value' ;
    main::ok(97, $@ eq "") ;
    main::ok(98, $ret == 10) ;

    $ret = eval ' R_NEXT eq main::R_NEXT ' ;
    main::ok(99, $@ eq "" ) ;
    main::ok(100, $ret == 1) ;

    $ret = eval '$X->A_new_method("joe") ' ;
    main::ok(101, $@ eq "") ;
    main::ok(102, $ret eq "[[11]]") ;

    undef $X;
    untie(%h);
    unlink "SubDB.pm", "dbbtree.tmp" ;

}

exit ;
Commit	Line	Data
f6b705ef	1	#!./perl -w
a0d0e21e	2
a0d0e21e	3	BEGIN {
55497cff	4	@INC = '../lib' if -d '../lib' ;
a0d0e21e	5	require Config; import Config;
	6	if ($Config{'extensions'} !~ /\bDB_File\b/) {
	7	print "1..0\n";
	8	exit 0;
	9	}
	10	}
	11
	12	use DB_File;
	13	use Fcntl;
	14
a6ed719b	15	print "1..102\n";
f6b705ef	16
	17	sub ok
	18	{
	19	my $no = shift ;
	20	my $result = shift ;
	21
	22	print "not " unless $result ;
	23	print "ok $no\n" ;
	24	}
a0d0e21e	25
55497cff	26	sub lexical
	27	{
	28	my(@a) = unpack ("C*", $a) ;
	29	my(@b) = unpack ("C*", $b) ;
	30
	31	my $len = (@a > @b ? @b : @a) ;
	32	my $i = 0 ;
	33
	34	foreach $i ( 0 .. $len -1) {
	35	return $a[$i] - $b[$i] if $a[$i] != $b[$i] ;
	36	}
	37
	38	return @a - @b ;
	39	}
	40
55d68b4a	41	$Dfile = "dbbtree.tmp";
a0d0e21e	42	unlink $Dfile;
	43
	44	umask(0);
	45
	46	# Check the interface to BTREEINFO
	47
f6b705ef	48	my $dbh = new DB_File::BTREEINFO ;
3fe9a6f1	49	ok(1, ! defined $dbh->{flags}) ;
	50	ok(2, ! defined $dbh->{cachesize}) ;
	51	ok(3, ! defined $dbh->{psize}) ;
	52	ok(4, ! defined $dbh->{lorder}) ;
	53	ok(5, ! defined $dbh->{minkeypage}) ;
	54	ok(6, ! defined $dbh->{maxkeypage}) ;
	55	ok(7, ! defined $dbh->{compare}) ;
	56	ok(8, ! defined $dbh->{prefix}) ;
a0d0e21e	57
a0d0e21e	58	$dbh->{flags} = 3000 ;
f6b705ef	59	ok(9, $dbh->{flags} == 3000) ;
a0d0e21e	60
a0d0e21e	61	$dbh->{cachesize} = 9000 ;
f6b705ef	62	ok(10, $dbh->{cachesize} == 9000);
f6b705ef	63
a0d0e21e	64	$dbh->{psize} = 400 ;
f6b705ef	65	ok(11, $dbh->{psize} == 400) ;
a0d0e21e	66
a0d0e21e	67	$dbh->{lorder} = 65 ;
f6b705ef	68	ok(12, $dbh->{lorder} == 65) ;
a0d0e21e	69
a0d0e21e	70	$dbh->{minkeypage} = 123 ;
f6b705ef	71	ok(13, $dbh->{minkeypage} == 123) ;
a0d0e21e	72
a0d0e21e	73	$dbh->{maxkeypage} = 1234 ;
f6b705ef	74	ok(14, $dbh->{maxkeypage} == 1234 );
a0d0e21e	75
a0d0e21e	76	$dbh->{compare} = 1234 ;
f6b705ef	77	ok(15, $dbh->{compare} == 1234) ;
a0d0e21e	78
a0d0e21e	79	$dbh->{prefix} = 1234 ;
f6b705ef	80	ok(16, $dbh->{prefix} == 1234 );
a0d0e21e	81
	82	# Check that an invalid entry is caught both for store & fetch
	83	eval '$dbh->{fred} = 1234' ;
f6b705ef	84	ok(17, $@ =~ /^DB_File::BTREEINFO::STORE - Unknown element 'fred' at/ ) ;
a0d0e21e	85	eval '$q = $dbh->{fred}' ;
f6b705ef	86	ok(18, $@ =~ /^DB_File::BTREEINFO::FETCH - Unknown element 'fred' at/ ) ;
a0d0e21e	87
	88	# Now check the interface to BTREE
	89
f6b705ef	90	ok(19, $X = tie(%h, 'DB_File',$Dfile, O_RDWR\|O_CREAT, 0640, $DB_BTREE )) ;
a0d0e21e	91
	92	($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,$atime,$mtime,$ctime,
	93	$blksize,$blocks) = stat($Dfile);
a6ed719b	94	ok(20, ($mode & 0777) == (($^O eq 'os2' \|\| $^O eq 'MSWin32') ? 0666 : 0640) \|\| $^O eq 'amigaos');
a0d0e21e	95
	96	while (($key,$value) = each(%h)) {
	97	$i++;
	98	}
f6b705ef	99	ok(21, !$i ) ;
a0d0e21e	100
	101	$h{'goner1'} = 'snork';
	102
	103	$h{'abc'} = 'ABC';
f6b705ef	104	ok(22, $h{'abc'} eq 'ABC' );
	105	ok(23, ! defined $h{'jimmy'} ) ;
	106	ok(24, ! exists $h{'jimmy'} ) ;
	107	ok(25, defined $h{'abc'} ) ;
a0d0e21e	108
	109	$h{'def'} = 'DEF';
	110	$h{'jkl','mno'} = "JKL\034MNO";
	111	$h{'a',2,3,4,5} = join("\034",'A',2,3,4,5);
	112	$h{'a'} = 'A';
	113
	114	#$h{'b'} = 'B';
	115	$X->STORE('b', 'B') ;
	116
	117	$h{'c'} = 'C';
	118
	119	#$h{'d'} = 'D';
	120	$X->put('d', 'D') ;
	121
	122	$h{'e'} = 'E';
	123	$h{'f'} = 'F';
	124	$h{'g'} = 'X';
	125	$h{'h'} = 'H';
	126	$h{'i'} = 'I';
	127
	128	$h{'goner2'} = 'snork';
	129	delete $h{'goner2'};
	130
	131
	132	# IMPORTANT - $X must be undefined before the untie otherwise the
	133	# underlying DB close routine will not get called.
	134	undef $X ;
	135	untie(%h);
	136
	137
	138	# tie to the same file again
f6b705ef	139	ok(26, $X = tie(%h,'DB_File',$Dfile, O_RDWR, 0640, $DB_BTREE)) ;
a0d0e21e	140
	141	# Modify an entry from the previous tie
	142	$h{'g'} = 'G';
	143
	144	$h{'j'} = 'J';
	145	$h{'k'} = 'K';
	146	$h{'l'} = 'L';
	147	$h{'m'} = 'M';
	148	$h{'n'} = 'N';
	149	$h{'o'} = 'O';
	150	$h{'p'} = 'P';
	151	$h{'q'} = 'Q';
	152	$h{'r'} = 'R';
	153	$h{'s'} = 'S';
	154	$h{'t'} = 'T';
	155	$h{'u'} = 'U';
	156	$h{'v'} = 'V';
	157	$h{'w'} = 'W';
	158	$h{'x'} = 'X';
	159	$h{'y'} = 'Y';
	160	$h{'z'} = 'Z';
	161
	162	$h{'goner3'} = 'snork';
	163
	164	delete $h{'goner1'};
	165	$X->DELETE('goner3');
	166
	167	@keys = keys(%h);
	168	@values = values(%h);
	169
f6b705ef	170	ok(27, $#keys == 29 && $#values == 29) ;
a0d0e21e	171
f6b705ef	172	$i = 0 ;
a0d0e21e	173	while (($key,$value) = each(%h)) {
2f52a358	174	if ($key eq $keys[$i] && $value eq $values[$i] && $key eq lc($value)) {
a0d0e21e	175	$key =~ y/a-z/A-Z/;
	176	$i++ if $key eq $value;
	177	}
	178	}
	179
f6b705ef	180	ok(28, $i == 30) ;
a0d0e21e	181
55d68b4a	182	@keys = ('blurfl', keys(%h), 'dyick');
f6b705ef	183	ok(29, $#keys == 31) ;
a0d0e21e	184
a0d0e21e	185	#Check that the keys can be retrieved in order
55497cff	186	my @b = keys %h ;
	187	my @c = sort lexical @b ;
	188	ok(30, ArrayCompare(\@b, \@c)) ;
a0d0e21e	189
a0d0e21e	190	$h{'foo'} = '';
f6b705ef	191	ok(31, $h{'foo'} eq '' ) ;
a0d0e21e	192
a0d0e21e	193	$h{''} = 'bar';
f6b705ef	194	ok(32, $h{''} eq 'bar' );
a0d0e21e	195
	196	# check cache overflow and numeric keys and contents
	197	$ok = 1;
	198	for ($i = 1; $i < 200; $i++) { $h{$i + 0} = $i + 0; }
	199	for ($i = 1; $i < 200; $i++) { $ok = 0 unless $h{$i} == $i; }
f6b705ef	200	ok(33, $ok);
a0d0e21e	201
	202	($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,$atime,$mtime,$ctime,
	203	$blksize,$blocks) = stat($Dfile);
f6b705ef	204	ok(34, $size > 0 );
a0d0e21e	205
	206	@h{0..200} = 200..400;
	207	@foo = @h{0..200};
f6b705ef	208	ok(35, join(':',200..400) eq join(':',@foo) );
a0d0e21e	209
	210	# Now check all the non-tie specific stuff
	211
	212
	213	# Check R_NOOVERWRITE flag will make put fail when attempting to overwrite
	214	# an existing record.
	215
	216	$status = $X->put( 'x', 'newvalue', R_NOOVERWRITE) ;
f6b705ef	217	ok(36, $status == 1 );
a0d0e21e	218
	219	# check that the value of the key 'x' has not been changed by the
	220	# previous test
f6b705ef	221	ok(37, $h{'x'} eq 'X' );
a0d0e21e	222
	223	# standard put
	224	$status = $X->put('key', 'value') ;
f6b705ef	225	ok(38, $status == 0 );
a0d0e21e	226
a0d0e21e	227	#check that previous put can be retrieved
f6b705ef	228	$value = 0 ;
a0d0e21e	229	$status = $X->get('key', $value) ;
f6b705ef	230	ok(39, $status == 0 );
f6b705ef	231	ok(40, $value eq 'value' );
a0d0e21e	232
	233	# Attempting to delete an existing key should work
	234
	235	$status = $X->del('q') ;
f6b705ef	236	ok(41, $status == 0 );
a0d0e21e	237	$status = $X->del('') ;
f6b705ef	238	ok(42, $status == 0 );
a0d0e21e	239
a0d0e21e	240	# Make sure that the key deleted, cannot be retrieved
ac1ad7f0	241	ok(43, ! defined $h{'q'}) ;
ac1ad7f0	242	ok(44, ! defined $h{''}) ;
a0d0e21e	243
	244	undef $X ;
	245	untie %h ;
	246
f6b705ef	247	ok(45, $X = tie(%h, 'DB_File',$Dfile, O_RDWR, 0640, $DB_BTREE ));
a0d0e21e	248
	249	# Attempting to delete a non-existant key should fail
	250
	251	$status = $X->del('joe') ;
f6b705ef	252	ok(46, $status == 1 );
a0d0e21e	253
	254	# Check the get interface
	255
	256	# First a non-existing key
	257	$status = $X->get('aaaa', $value) ;
f6b705ef	258	ok(47, $status == 1 );
a0d0e21e	259
	260	# Next an existing key
	261	$status = $X->get('a', $value) ;
f6b705ef	262	ok(48, $status == 0 );
f6b705ef	263	ok(49, $value eq 'A' );
a0d0e21e	264
	265	# seq
	266	# ###
	267
	268	# use seq to find an approximate match
	269	$key = 'ke' ;
	270	$value = '' ;
	271	$status = $X->seq($key, $value, R_CURSOR) ;
f6b705ef	272	ok(50, $status == 0 );
	273	ok(51, $key eq 'key' );
	274	ok(52, $value eq 'value' );
a0d0e21e	275
	276	# seq when the key does not match
	277	$key = 'zzz' ;
	278	$value = '' ;
	279	$status = $X->seq($key, $value, R_CURSOR) ;
f6b705ef	280	ok(53, $status == 1 );
a0d0e21e	281
	282
	283	# use seq to set the cursor, then delete the record @ the cursor.
	284
	285	$key = 'x' ;
	286	$value = '' ;
	287	$status = $X->seq($key, $value, R_CURSOR) ;
f6b705ef	288	ok(54, $status == 0 );
	289	ok(55, $key eq 'x' );
	290	ok(56, $value eq 'X' );
a0d0e21e	291	$status = $X->del(0, R_CURSOR) ;
f6b705ef	292	ok(57, $status == 0 );
a0d0e21e	293	$status = $X->get('x', $value) ;
f6b705ef	294	ok(58, $status == 1 );
a0d0e21e	295
	296	# ditto, but use put to replace the key/value pair.
	297	$key = 'y' ;
	298	$value = '' ;
	299	$status = $X->seq($key, $value, R_CURSOR) ;
f6b705ef	300	ok(59, $status == 0 );
	301	ok(60, $key eq 'y' );
	302	ok(61, $value eq 'Y' );
a0d0e21e	303
	304	$key = "replace key" ;
	305	$value = "replace value" ;
	306	$status = $X->put($key, $value, R_CURSOR) ;
f6b705ef	307	ok(62, $status == 0 );
	308	ok(63, $key eq 'replace key' );
	309	ok(64, $value eq 'replace value' );
a0d0e21e	310	$status = $X->get('y', $value) ;
1f70e1ea	311	ok(65, 1) ; # hard-wire to always pass. the previous test ($status == 1)
	312	# only worked because of a bug in 1.85/6
	313
a0d0e21e	314
	315	# use seq to walk forwards through a file
	316
	317	$status = $X->seq($key, $value, R_FIRST) ;
f6b705ef	318	ok(66, $status == 0 );
a0d0e21e	319	$previous = $key ;
	320
	321	$ok = 1 ;
	322	while (($status = $X->seq($key, $value, R_NEXT)) == 0)
	323	{
	324	($ok = 0), last if ($previous cmp $key) == 1 ;
	325	}
	326
f6b705ef	327	ok(67, $status == 1 );
f6b705ef	328	ok(68, $ok == 1 );
a0d0e21e	329
	330	# use seq to walk backwards through a file
	331	$status = $X->seq($key, $value, R_LAST) ;
f6b705ef	332	ok(69, $status == 0 );
a0d0e21e	333	$previous = $key ;
	334
	335	$ok = 1 ;
	336	while (($status = $X->seq($key, $value, R_PREV)) == 0)
	337	{
	338	($ok = 0), last if ($previous cmp $key) == -1 ;
	339	#print "key = [$key] value = [$value]\n" ;
	340	}
	341
f6b705ef	342	ok(70, $status == 1 );
f6b705ef	343	ok(71, $ok == 1 );
a0d0e21e	344
	345
	346	# check seq FIRST/LAST
	347
	348	# sync
	349	# ####
	350
	351	$status = $X->sync ;
f6b705ef	352	ok(72, $status == 0 );
a0d0e21e	353
	354
	355	# fd
	356	# ##
	357
	358	$status = $X->fd ;
f6b705ef	359	ok(73, $status != 0 );
a0d0e21e	360
	361
	362	undef $X ;
	363	untie %h ;
	364
	365	unlink $Dfile;
	366
	367	# Now try an in memory file
f6b705ef	368	ok(74, $Y = tie(%h, 'DB_File',undef, O_RDWR\|O_CREAT, 0640, $DB_BTREE ));
a0d0e21e	369
	370	# fd with an in memory file should return failure
	371	$status = $Y->fd ;
f6b705ef	372	ok(75, $status == -1 );
a0d0e21e	373
55d68b4a	374
a0d0e21e	375	undef $Y ;
	376	untie %h ;
	377
55d68b4a	378	# Duplicate keys
	379	my $bt = new DB_File::BTREEINFO ;
	380	$bt->{flags} = R_DUP ;
f6b705ef	381	ok(76, $YY = tie(%hh, 'DB_File', $Dfile, O_RDWR\|O_CREAT, 0640, $bt )) ;
55d68b4a	382
	383	$hh{'Wall'} = 'Larry' ;
	384	$hh{'Wall'} = 'Stone' ; # Note the duplicate key
	385	$hh{'Wall'} = 'Brick' ; # Note the duplicate key
f6b705ef	386	$hh{'Wall'} = 'Brick' ; # Note the duplicate key and value
55d68b4a	387	$hh{'Smith'} = 'John' ;
	388	$hh{'mouse'} = 'mickey' ;
	389
	390	# first work in scalar context
f6b705ef	391	ok(77, scalar $YY->get_dup('Unknown') == 0 );
	392	ok(78, scalar $YY->get_dup('Smith') == 1 );
	393	ok(79, scalar $YY->get_dup('Wall') == 4 );
55d68b4a	394
	395	# now in list context
	396	my @unknown = $YY->get_dup('Unknown') ;
f6b705ef	397	ok(80, "@unknown" eq "" );
55d68b4a	398
55d68b4a	399	my @smith = $YY->get_dup('Smith') ;
f6b705ef	400	ok(81, "@smith" eq "John" );
55d68b4a	401
760ac839	402	{
f6b705ef	403	my @wall = $YY->get_dup('Wall') ;
	404	my %wall ;
	405	@wall{@wall} = @wall ;
	406	ok(82, (@wall == 4 && $wall{'Larry'} && $wall{'Stone'} && $wall{'Brick'}) );
760ac839	407	}
55d68b4a	408
	409	# hash
	410	my %unknown = $YY->get_dup('Unknown', 1) ;
f6b705ef	411	ok(83, keys %unknown == 0 );
55d68b4a	412
55d68b4a	413	my %smith = $YY->get_dup('Smith', 1) ;
f6b705ef	414	ok(84, keys %smith == 1 && $smith{'John'}) ;
55d68b4a	415
f6b705ef	416	my %wall = $YY->get_dup('Wall', 1) ;
	417	ok(85, keys %wall == 3 && $wall{'Larry'} == 1 && $wall{'Stone'} == 1
	418	&& $wall{'Brick'} == 2);
55d68b4a	419
	420	undef $YY ;
	421	untie %hh ;
	422	unlink $Dfile;
	423
	424
8e07c86e	425	# test multiple callbacks
	426	$Dfile1 = "btree1" ;
	427	$Dfile2 = "btree2" ;
	428	$Dfile3 = "btree3" ;
	429
36477c24	430	$dbh1 = new DB_File::BTREEINFO ;
ac1ad7f0	431	{ local $^W = 0 ;
ac1ad7f0	432	$dbh1->{compare} = sub { $_[0] <=> $_[1] } ; }
8e07c86e	433
36477c24	434	$dbh2 = new DB_File::BTREEINFO ;
8e07c86e	435	$dbh2->{compare} = sub { $_[0] cmp $_[1] } ;
8e07c86e	436
36477c24	437	$dbh3 = new DB_File::BTREEINFO ;
8e07c86e	438	$dbh3->{compare} = sub { length $_[0] <=> length $_[1] } ;
	439
	440
f6b705ef	441	tie(%h, 'DB_File',$Dfile1, O_RDWR\|O_CREAT, 0640, $dbh1 ) ;
	442	tie(%g, 'DB_File',$Dfile2, O_RDWR\|O_CREAT, 0640, $dbh2 ) ;
	443	tie(%k, 'DB_File',$Dfile3, O_RDWR\|O_CREAT, 0640, $dbh3 ) ;
8e07c86e	444
8e07c86e	445	@Keys = qw( 0123 12 -1234 9 987654321 def ) ;
ac1ad7f0	446	{ local $^W = 0 ;
ac1ad7f0	447	@srt_1 = sort { $a <=> $b } @Keys ; }
8e07c86e	448	@srt_2 = sort { $a cmp $b } @Keys ;
	449	@srt_3 = sort { length $a <=> length $b } @Keys ;
	450
	451	foreach (@Keys) {
ac1ad7f0	452	{ local $^W = 0 ;
ac1ad7f0	453	$h{$_} = 1 ; }
8e07c86e	454	$g{$_} = 1 ;
	455	$k{$_} = 1 ;
	456	}
	457
	458	sub ArrayCompare
	459	{
	460	my($a, $b) = @_ ;
	461
	462	return 0 if @$a != @$b ;
	463
	464	foreach (1 .. length @$a)
	465	{
	466	return 0 unless $$a[$_] eq $$b[$_] ;
	467	}
	468
	469	1 ;
	470	}
	471
f6b705ef	472	ok(86, ArrayCompare (\@srt_1, [keys %h]) );
	473	ok(87, ArrayCompare (\@srt_2, [keys %g]) );
	474	ok(88, ArrayCompare (\@srt_3, [keys %k]) );
8e07c86e	475
	476	untie %h ;
	477	untie %g ;
	478	untie %k ;
	479	unlink $Dfile1, $Dfile2, $Dfile3 ;
	480
f6b705ef	481	# clear
	482	# #####
	483
	484	ok(89, tie(%h, 'DB_File', $Dfile1, O_RDWR\|O_CREAT, 0640, $DB_BTREE ) );
	485	foreach (1 .. 10)
	486	{ $h{$_} = $_ * 100 }
	487
	488	# check that there are 10 elements in the hash
	489	$i = 0 ;
	490	while (($key,$value) = each(%h)) {
	491	$i++;
	492	}
	493	ok(90, $i == 10);
	494
	495	# now clear the hash
	496	%h = () ;
	497
	498	# check it is empty
	499	$i = 0 ;
	500	while (($key,$value) = each(%h)) {
	501	$i++;
	502	}
	503	ok(91, $i == 0);
	504
	505	untie %h ;
	506	unlink $Dfile1 ;
	507
05475680	508	{
	509	# check that attempting to tie an array to a DB_BTREE will fail
	510
	511	my $filename = "xyz" ;
	512	my @x ;
	513	eval { tie @x, 'DB_File', $filename, O_RDWR\|O_CREAT, 0640, $DB_BTREE ; } ;
	514	ok(92, $@ =~ /^DB_File can only tie an associative array to a DB_BTREE database/) ;
	515	unlink $filename ;
	516	}
	517
a6ed719b	518
	519	{
	520	# sub-class test
	521
	522	package Another ;
	523
	524	use strict ;
	525
	526	open(FILE, ">SubDB.pm") or die "Cannot open SubDB.pm: $!\n" ;
	527	print FILE <<'EOM' ;
	528
	529	package SubDB ;
	530
	531	use strict ;
	532	use vars qw( @ISA @EXPORT) ;
	533
	534	require Exporter ;
	535	use DB_File;
	536	@ISA=qw(DB_File);
	537	@EXPORT = @DB_File::EXPORT ;
	538
	539	sub STORE {
	540	my $self = shift ;
	541	my $key = shift ;
	542	my $value = shift ;
	543	$self->SUPER::STORE($key, $value * 2) ;
	544	}
	545
	546	sub FETCH {
	547	my $self = shift ;
	548	my $key = shift ;
	549	$self->SUPER::FETCH($key) - 1 ;
	550	}
	551
	552	sub put {
	553	my $self = shift ;
	554	my $key = shift ;
	555	my $value = shift ;
	556	$self->SUPER::put($key, $value * 3) ;
	557	}
	558
	559	sub get {
	560	my $self = shift ;
	561	$self->SUPER::get($_[0], $_[1]) ;
	562	$_[1] -= 2 ;
	563	}
	564
	565	sub A_new_method
	566	{
	567	my $self = shift ;
	568	my $key = shift ;
	569	my $value = $self->FETCH($key) ;
	570	return "[[$value]]" ;
	571	}
	572
	573	1 ;
	574	EOM
	575
	576	close FILE ;
	577
	578	BEGIN { push @INC, '.'; }
	579	eval 'use SubDB ; ';
	580	main::ok(93, $@ eq "") ;
	581	my %h ;
582	my $X ;
583	eval '
584	$X = tie(%h, "SubDB","dbbtree.tmp", O_RDWR\|O_CREAT, 0640, $DB_BTREE );
585	' ;
586
587	main::ok(94, $@ eq "") ;
588
589	my $ret = eval '$h{"fred"} = 3 ; return $h{"fred"} ' ;
590	main::ok(95, $@ eq "") ;
591	main::ok(96, $ret == 5) ;
592
593	my $value = 0;
594	$ret = eval '$X->put("joe", 4) ; $X->get("joe", $value) ; return $value' ;
595	main::ok(97, $@ eq "") ;
596	main::ok(98, $ret == 10) ;
597
598	$ret = eval ' R_NEXT eq main::R_NEXT ' ;
599	main::ok(99, $@ eq "" ) ;
600	main::ok(100, $ret == 1) ;
601
602	$ret = eval '$X->A_new_method("joe") ' ;
603	main::ok(101, $@ eq "") ;
604	main::ok(102, $ret eq "[[11]]") ;
605
fac76ed7	606	undef $X;
fac76ed7	607	untie(%h);
a6ed719b	608	unlink "SubDB.pm", "dbbtree.tmp" ;
	609
	610	}
	611
a0d0e21e	612	exit ;