Commit | Line | Data |
0e2c6809 |
1 | #!/usr/bin/perl |
2 | use strict; |
3 | use warnings; |
4 | use Test::More; |
5 | use lib qw(t/lib); |
6 | use DBICTest; |
7 | |
8 | BEGIN { |
b2b2e7fd |
9 | require DBIx::Class::Storage::DBI; |
7f6f5b69 |
10 | plan skip_all => |
b2b2e7fd |
11 | 'Test needs SQL::Translator ' . DBIx::Class::Storage::DBI->_sqlt_minimum_version |
12 | if not DBIx::Class::Storage::DBI->_sqlt_version_ok; |
0e2c6809 |
13 | } |
14 | |
15 | my $schema = DBICTest->init_schema(); |
0fc7cd47 |
16 | # Dummy was yanked out by the sqlt hook test |
8e8a8d91 |
17 | # CustomSql tests the horrific/deprecated ->name(\$sql) hack |
0fc7cd47 |
18 | # YearXXXXCDs are views |
8e8a8d91 |
19 | # |
20 | my @sources = grep |
21 | { $_ !~ /^ (?: Dummy | CustomSql | Year\d{4}CDs ) $/x } |
22 | $schema->sources |
23 | ; |
0fc7cd47 |
24 | |
0e2c6809 |
25 | { |
26 | my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { } } }); |
27 | |
181c0934 |
28 | foreach my $source (@sources) { |
c2b7c5dc |
29 | my $table = get_table($sqlt_schema, $schema, $source); |
0e2c6809 |
30 | |
31 | my $fk_count = scalar(grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints); |
32 | my @indices = $table->get_indices; |
33 | my $index_count = scalar(@indices); |
2581038c |
34 | $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def |
0e2c6809 |
35 | is($index_count, $fk_count, "correct number of indices for $source with no args"); |
36 | } |
37 | } |
38 | |
39 | { |
40 | my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { add_fk_index => 1 } } }); |
41 | |
181c0934 |
42 | foreach my $source (@sources) { |
c2b7c5dc |
43 | my $table = get_table($sqlt_schema, $schema, $source); |
0e2c6809 |
44 | |
45 | my $fk_count = scalar(grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints); |
46 | my @indices = $table->get_indices; |
47 | my $index_count = scalar(@indices); |
2581038c |
48 | $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def |
0e2c6809 |
49 | is($index_count, $fk_count, "correct number of indices for $source with add_fk_index => 1"); |
50 | } |
51 | } |
52 | |
53 | { |
54 | my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { add_fk_index => 0 } } }); |
55 | |
181c0934 |
56 | foreach my $source (@sources) { |
c2b7c5dc |
57 | my $table = get_table($sqlt_schema, $schema, $source); |
0e2c6809 |
58 | |
59 | my @indices = $table->get_indices; |
60 | my $index_count = scalar(@indices); |
61 | is($index_count, 0, "correct number of indices for $source with add_fk_index => 0"); |
62 | } |
63 | } |
64 | |
7f6f5b69 |
65 | done_testing; |
66 | |
0e2c6809 |
67 | sub create_schema { |
68 | my $args = shift; |
69 | |
70 | my $schema = $args->{schema}; |
71 | my $additional_sqltargs = $args->{args} || {}; |
72 | |
73 | my $sqltargs = { |
74 | add_drop_table => 1, |
75 | ignore_constraint_names => 1, |
76 | ignore_index_names => 1, |
77 | %{$additional_sqltargs} |
78 | }; |
79 | |
80 | my $sqlt = SQL::Translator->new( $sqltargs ); |
81 | |
82 | $sqlt->parser('SQL::Translator::Parser::DBIx::Class'); |
83 | return $sqlt->translate({ data => $schema }) or die $sqlt->error; |
84 | } |
c2b7c5dc |
85 | |
86 | sub get_table { |
87 | my ($sqlt_schema, $schema, $source) = @_; |
88 | |
89 | my $table_name = $schema->source($source)->from; |
90 | $table_name = $$table_name if ref $table_name; |
91 | |
92 | return $sqlt_schema->get_table($table_name); |
93 | } |