#!/usr/bin/perl -w
#
# t/schema.t -- Tests for the wallet schema class.
#
# Written by Russ Allbery <rra@stanford.edu>
# Copyright 2007, 2008 Board of Trustees, Leland Stanford Jr. University
#
# See LICENSE for licensing terms.

use Test::More tests => 8;

use DBI;
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), 28, ' 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");

# 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';