use strict;
use warnings;
use Test::More;
use lib qw(t/lib);
use DBICTest;
eval "use SQL::Translator";
plan skip_all => 'SQL::Translator required' if $@;
my $schema = DBICTest->init_schema;
plan tests => 53;
my $translator = SQL::Translator->new(
parser_args => {
'DBIx::Schema' => $schema,
},
producer_args => {},
);
$translator->parser('SQL::Translator::Parser::DBIx::Class');
$translator->producer('SQLite');
my $output = $translator->translate();
# Note that the constraints listed here are the only ones that are tested -- if
# more exist in the Schema than are listed here and all listed constraints are
# correct, the test will still pass. If you add a class with UNIQUE or FOREIGN
# KEY constraints to DBICTest::Schema, add tests here if you think the existing
# test coverage is not sufficient
my %fk_constraints = (
# TwoKeys
twokeys => [
{
'display' => 'twokeys->cd',
'selftable' => 'twokeys', 'foreigntable' => 'cd',
'selfcols' => ['cd'], 'foreigncols' => ['cdid'],
on_delete => '', on_update => '',
},
{
'display' => 'twokeys->artist',
'selftable' => 'twokeys', 'foreigntable' => 'artist',
'selfcols' => ['artist'], 'foreigncols' => ['artistid'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
],
# FourKeys_to_TwoKeys
fourkeys_to_twokeys => [
{
'display' => 'fourkeys_to_twokeys->twokeys',
'selftable' => 'fourkeys_to_twokeys', 'foreigntable' => 'twokeys',
'selfcols' => ['t_artist', 't_cd'], 'foreigncols' => ['artist', 'cd'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
{
'display' => 'fourkeys_to_twokeys->fourkeys',
'selftable' => 'fourkeys_to_twokeys', 'foreigntable' => 'fourkeys',
'selfcols' => [qw(f_foo f_bar f_hello f_goodbye)],
'foreigncols' => [qw(foo bar hello goodbye)],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
],
# CD_to_Producer
cd_to_producer => [
{
'display' => 'cd_to_producer->cd',
'selftable' => 'cd_to_producer', 'foreigntable' => 'cd',
'selfcols' => ['cd'], 'foreigncols' => ['cdid'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
{
'display' => 'cd_to_producer->producer',
'selftable' => 'cd_to_producer', 'foreigntable' => 'producer',
'selfcols' => ['producer'], 'foreigncols' => ['producerid'],
on_delete => '', on_update => '',
},
],
# Self_ref_alias
self_ref_alias => [
{
'display' => 'self_ref_alias->self_ref for self_ref',
'selftable' => 'self_ref_alias', 'foreigntable' => 'self_ref',
'selfcols' => ['self_ref'], 'foreigncols' => ['id'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
{
'display' => 'self_ref_alias->self_ref for alias',
'selftable' => 'self_ref_alias', 'foreigntable' => 'self_ref',
'selfcols' => ['alias'], 'foreigncols' => ['id'],
on_delete => '', on_update => '',
},
],
# CD
cd => [
{
'display' => 'cd->artist',
'selftable' => 'cd', 'foreigntable' => 'artist',
'selfcols' => ['artist'], 'foreigncols' => ['artistid'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
],
# Artist_undirected_map
artist_undirected_map => [
{
'display' => 'artist_undirected_map->artist for id1',
'selftable' => 'artist_undirected_map', 'foreigntable' => 'artist',
'selfcols' => ['id1'], 'foreigncols' => ['artistid'],
on_delete => 'CASCADE', on_update => '',
},
{
'display' => 'artist_undirected_map->artist for id2',
'selftable' => 'artist_undirected_map', 'foreigntable' => 'artist',
'selfcols' => ['id2'], 'foreigncols' => ['artistid'],
on_delete => 'CASCADE', on_update => '',
},
],
# Track
track => [
{
'display' => 'track->cd',
'selftable' => 'track', 'foreigntable' => 'cd',
'selfcols' => ['cd'], 'foreigncols' => ['cdid'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
],
# TreeLike
treelike => [
{
'display' => 'treelike->treelike for parent',
'selftable' => 'treelike', 'foreigntable' => 'treelike',
'selfcols' => ['parent'], 'foreigncols' => ['id'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
],
# TwoKeyTreeLike
twokeytreelike => [
{
'display' => 'twokeytreelike->twokeytreelike for parent1,parent2',
'selftable' => 'twokeytreelike', 'foreigntable' => 'twokeytreelike',
'selfcols' => ['parent1', 'parent2'], 'foreigncols' => ['id1','id2'],
on_delete => '', on_update => '',
},
],
# Tags
tags => [
{
'display' => 'tags->cd',
'selftable' => 'tags', 'foreigntable' => 'cd',
'selfcols' => ['cd'], 'foreigncols' => ['cdid'],
on_delete => 'CASCADE', on_update => 'CASCADE',
},
],
# Bookmark
bookmark => [
{
'display' => 'bookmark->link',
'selftable' => 'bookmark', 'foreigntable' => 'link',
'selfcols' => ['link'], 'foreigncols' => ['id'],
on_delete => '', on_update => '',
},
],
);
my %unique_constraints = (
# CD
cd => [
{
'display' => 'cd artist and title unique',
'table' => 'cd', 'cols' => ['artist', 'title'],
},
],
# Producer
producer => [
{
'display' => 'producer name unique',
'table' => 'producer', 'cols' => ['name'],
},
],
# TwoKeyTreeLike
twokeytreelike => [
{
'display' => 'twokeytreelike name unique',
'table' => 'twokeytreelike', 'cols' => ['name'],
},
],
# Employee
# Constraint is commented out in DBICTest/Schema/Employee.pm
# employee => [
# {
# 'display' => 'employee position and group_id unique',
# 'table' => 'employee', cols => ['position', 'group_id'],
# },
# ],
);
my $tschema = $translator->schema();
# Test that nonexistent constraints are not found
my $constraint = get_constraint('FOREIGN KEY', 'cd', ['title'], 'cd', ['year']);
ok( !defined($constraint), 'nonexistent FOREIGN KEY constraint not found' );
$constraint = get_constraint('UNIQUE', 'cd', ['artist']);
ok( !defined($constraint), 'nonexistent UNIQUE constraint not found' );
for my $expected_constraints (keys %fk_constraints) {
for my $expected_constraint (@{ $fk_constraints{$expected_constraints} }) {
my $desc = $expected_constraint->{display};
my $constraint = get_constraint(
'FOREIGN KEY',
$expected_constraint->{selftable}, $expected_constraint->{selfcols},
$expected_constraint->{foreigntable}, $expected_constraint->{foreigncols},
);
ok( defined($constraint), "FOREIGN KEY constraint matching `$desc' found" );
test_fk($expected_constraint, $constraint);
}
}
for my $expected_constraints (keys %unique_constraints) {
for my $expected_constraint (@{ $unique_constraints{$expected_constraints} }) {
my $desc = $expected_constraint->{display};
my $constraint = get_constraint(
'UNIQUE', $expected_constraint->{table}, $expected_constraint->{cols},
);
ok( defined($constraint), "UNIQUE constraint matching `$desc' found" );
}
}
# Returns the Constraint object for the specified constraint type, table and
# columns from the SQL::Translator schema, or undef if no matching constraint
# is found.
#
# NB: $type is either 'FOREIGN KEY' or 'UNIQUE'. In UNIQUE constraints the last
# two parameters are not used.
sub get_constraint {
my ($type, $table_name, $cols, $f_table, $f_cols) = @_;
$f_table ||= ''; # For UNIQUE constraints, reference_table is ''
$f_cols ||= [];
my $table = $tschema->get_table($table_name);
my %fields = map { $_ => 1 } @$cols;
my %f_fields = map { $_ => 1 } @$f_cols;
CONSTRAINT:
for my $constraint ( $table->get_constraints ) {
next unless $constraint->type eq $type;
next unless $constraint->reference_table eq $f_table;
my %rev_fields = map { $_ => 1 } $constraint->fields;
my %rev_f_fields = map { $_ => 1 } $constraint->reference_fields;
# Check that the given fields are a subset of the constraint's fields
for my $field ($constraint->fields) {
next CONSTRAINT unless $fields{$field};
}
if ($type eq 'FOREIGN KEY') {
for my $f_field ($constraint->reference_fields) {
next CONSTRAINT unless $f_fields{$f_field};
}
}
# Check that the constraint's fields are a subset of the given fields
for my $field (@$cols) {
next CONSTRAINT unless $rev_fields{$field};
}
if ($type eq 'FOREIGN KEY') {
for my $f_field (@$f_cols) {
next CONSTRAINT unless $rev_f_fields{$f_field};
}
}
return $constraint; # everything passes, found the constraint
}
return undef; # didn't find a matching constraint
}
# Test parameters in a FOREIGN KEY constraint other than columns
sub test_fk {
my ($expected, $got) = @_;
my $desc = $expected->{display};
is( $got->on_delete, $expected->{on_delete},
"on_delete parameter correct for `$desc'" );
is( $got->on_update, $expected->{on_update},
"on_update parameter correct for `$desc'" );
}