1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#!/usr/bin/perl -w
#
# Tests for the wallet schema class.
#
# Written by Russ Allbery <rra@stanford.edu>
# Copyright 2007, 2008, 2011
# The Board of Trustees of the Leland Stanford Junior University
#
# See LICENSE for licensing terms.
use Test::More tests => 16;
use DBI ();
use POSIX qw(strftime);
use Wallet::Config ();
use Wallet::Schema ();
use lib 't/lib';
use Util;
my $schema = Wallet::Schema->new;
ok (defined $schema, 'Wallet::Schema creation');
ok ($schema->isa ('Wallet::Schema'), ' and class verification');
my @sql = $schema->sql;
ok (@sql > 0, 'sql() returns something');
is (scalar (@sql), 32, ' and returns the right number of statements');
# Connect to a database and test create.
db_setup;
my $connect = "DBI:${Wallet::Config::DB_DRIVER}:${Wallet::Config::DB_INFO}";
my $user = $Wallet::Config::DB_USER;
my $password = $Wallet::Config::DB_PASSWORD;
$dbh = DBI->connect ($connect, $user, $password);
if (not defined $dbh) {
die "cannot connect to database $connect: $DBI::errstr\n";
}
$dbh->{RaiseError} = 1;
$dbh->{PrintError} = 0;
eval { $schema->create ($dbh) };
is ($@, '', "create() doesn't die");
# Check that the version number is correct.
my $sql = "select md_version from metadata";
my $version = $dbh->selectall_arrayref ($sql);
is (@$version, 1, 'metadata has correct number of rows');
is (@{ $version->[0] }, 1, ' and correct number of columns');
is ($version->[0][0], 1, ' and the schema version is correct');
# Test upgrading the database from version 0. SQLite cannot drop table
# columns, so we have to kill the table and then recreate it.
$dbh->do ("drop table metadata");
if (lc ($Wallet::Config::DB_DRIVER) eq 'sqlite') {
($sql) = grep { /create table objects/ } $schema->sql;
$sql =~ s/ob_comment .*,//;
$dbh->do ("drop table objects")
or die "cannot drop objects table: $DBI::errstr\n";
$dbh->do ($sql)
or die "cannot recreate objects table: $DBI::errstr\n";
} else {
$dbh->do ("alter table objects drop column ob_comment")
or die "cannot drop ob_comment column: $DBI::errstr\n";
}
eval { $schema->upgrade ($dbh) };
is ($@, '', "upgrade() doesn't die");
$sql = "select md_version from metadata";
$version = $dbh->selectall_arrayref ($sql);
is (@$version, 1, ' and metadata has correct number of rows');
is (@{ $version->[0] }, 1, ' and correct number of columns');
is ($version->[0][0], 1, ' and the schema version is correct');
$sql = "insert into objects (ob_type, ob_name, ob_created_by, ob_created_from,
ob_created_on, ob_comment) values ('file', 'test', 'test',
'test.example.org', ?, 'a test comment')";
$dbh->do ($sql, undef, strftime ('%Y-%m-%d %T', localtime time));
$sql = "select ob_comment from objects where ob_name = 'test'";
my ($comment) = $dbh->selectrow_array ($sql);
is ($comment, 'a test comment', ' and ob_comment was added to objects');
# Test dropping the database.
eval { $schema->drop ($dbh) };
is ($@, '', "drop() doesn't die");
# Make sure all the tables are gone.
SKIP: {
if (lc ($Wallet::Config::DB_DRIVER) eq 'sqlite') {
my $sql = "select name from sqlite_master where type = 'table'";
my $sth = $dbh->prepare ($sql);
$sth->execute;
my ($table, @tables);
while (defined ($table = $sth->fetchrow_array)) {
push (@tables, $table) unless $table =~ /^sqlite_/;
}
is ("@tables", '', ' and there are no tables in the database');
} elsif (lc ($Wallet::Config::DB_DRIVER) eq 'mysql') {
my $sql = "show tables";
my $sth = $dbh->prepare ($sql);
$sth->execute;
my ($table, @tables);
while (defined ($table = $sth->fetchrow_array)) {
push (@tables, $table);
}
is ("@tables", '', ' and there are no tables in the database');
} else {
skip 1;
}
}
eval { $schema->create ($dbh) };
is ($@, '', ' and we can run create again');
# Clean up.
eval { $schema->drop ($dbh) };
unlink 'wallet-db';
|
