f79590550291c9a9e9b4f4e358424ccaa50abf9a
[dbsrgits/DBIx-Class.git] / t / 99dbic_sqlt_parser.t
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 {
9   require DBIx::Class::Storage::DBI;
10   plan skip_all =>
11       'Test needs SQL::Translator ' . DBIx::Class::Storage::DBI->_sqlt_minimum_version
12     if not DBIx::Class::Storage::DBI->_sqlt_version_ok;
13 }
14
15 my $schema = DBICTest->init_schema();
16 # Dummy was yanked out by the sqlt hook test
17 # CustomSql tests the horrific/deprecated ->name(\$sql) hack
18 # YearXXXXCDs and NoViewDefinition are views
19 #
20 my @sources = grep
21   { $_ !~ /^ (?: Dummy | CustomSql | Year\d{4}CDs | NoViewDefinition ) $/x }
22   $schema->sources
23 ;
24
25
26         my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { } } });
27
28         foreach my $source (@sources) {
29                 my $table = get_table($sqlt_schema, $schema, $source);
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);
34     $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def
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
42         foreach my $source (@sources) {
43                 my $table = get_table($sqlt_schema, $schema, $source);
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);
48     $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def
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
56         foreach my $source (@sources) {
57                 my $table = get_table($sqlt_schema, $schema, $source);
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
65
66     {
67         package # hide from PAUSE
68             DBICTest::Schema::NoViewDefinition;
69
70         use base qw/DBICTest::BaseResult/;
71
72         __PACKAGE__->table_class('DBIx::Class::ResultSource::View');
73         __PACKAGE__->table('noviewdefinition');
74
75         1;
76     }
77
78     my $schema_invalid_view = $schema->clone;
79     $schema_invalid_view->register_class('NoViewDefinition', 'DBICTest::Schema::NoViewDefinition');
80
81     eval {
82             my $sqlt_schema = create_schema({ schema => $schema_invalid_view });
83     };
84     like($@, qr/view noviewdefinition is missing a view_definition/, "parser detects views with a view_definition");
85     
86 #    my @views = $sqlt_schema->get_views;
87 #
88 #    # the following views are skipped:
89 #    # Year1999CDs is virtual
90 #    # NoViewDefinition has no view_definition
91 #    is(scalar @views, 1, "number of views ok");
92 #
93 #    foreach my $view (@views) {
94 #        ok($view->is_valid, "view " . $view->name . " is valid");
95 #    }
96 #
97 #    my @expected_view_names = (qw/ year2000cds /);
98 #    my @view_names = sort map { $_->name } @views;
99 #
100 #    is_deeply( @view_names, @expected_view_names, "all expected views included int SQL::Translator schema" );
101 }
102
103 done_testing;
104
105 sub create_schema {
106         my $args = shift;
107
108         my $schema = $args->{schema};
109         my $additional_sqltargs = $args->{args} || {};
110
111         my $sqltargs = {
112                 add_drop_table => 1, 
113                 ignore_constraint_names => 1,
114                 ignore_index_names => 1,
115                 %{$additional_sqltargs}
116                 };
117
118         my $sqlt = SQL::Translator->new( $sqltargs );
119
120         $sqlt->parser('SQL::Translator::Parser::DBIx::Class');
121         return $sqlt->translate({ data => $schema }) || die $sqlt->error;
122 }
123
124 sub get_table {
125     my ($sqlt_schema, $schema, $source) = @_;
126
127     my $table_name = $schema->source($source)->from;
128     $table_name    = $$table_name if ref $table_name;
129
130     return $sqlt_schema->get_table($table_name);
131 }