t/99dbic_sqlt_parser.t

   1 use strict;
   2 use warnings;
   3
   4 use Test::More;
   5 use Test::Exception;
   6 use lib qw(t/lib);
   7 use DBICTest;
   8
   9 BEGIN {
  10   require DBIx::Class::Storage::DBI;
  11   plan skip_all =>
  12       'Test needs SQL::Translator ' . DBIx::Class::Storage::DBI->_sqlt_minimum_version
  13     if not DBIx::Class::Storage::DBI->_sqlt_version_ok;
  14 }
  15
  16 my $schema = DBICTest->init_schema();
  17 # Dummy was yanked out by the sqlt hook test
  18 # CustomSql tests the horrific/deprecated ->name(\$sql) hack
  19 # YearXXXXCDs are views
  20 #
  21 my @sources = grep
  22   { $_ !~ /^ (?: Dummy | CustomSql | Year\d{4}CDs ) $/x }
  23   $schema->sources
  24 ;
  25
  26 {
  27   my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { } } });
  28
  29   foreach my $source (@sources) {
  30     my $table = get_table($sqlt_schema, $schema, $source);
  31
  32     my $fk_count = scalar(grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints);
  33     my @indices = $table->get_indices;
  34     my $index_count = scalar(@indices);
  35     $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def
  36     is($index_count, $fk_count, "correct number of indices for $source with no args");
  37   }
  38 }
  39
  40 {
  41   my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { add_fk_index => 1 } } });
  42
  43   foreach my $source (@sources) {
  44     my $table = get_table($sqlt_schema, $schema, $source);
  45
  46     my $fk_count = scalar(grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints);
  47     my @indices = $table->get_indices;
  48     my $index_count = scalar(@indices);
  49     $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def
  50     is($index_count, $fk_count, "correct number of indices for $source with add_fk_index => 1");
  51   }
  52 }
  53
  54 {
  55   my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { add_fk_index => 0 } } });
  56
  57   foreach my $source (@sources) {
  58     my $table = get_table($sqlt_schema, $schema, $source);
  59
  60     my @indices = $table->get_indices;
  61     my $index_count = scalar(@indices);
  62     is($index_count, 0, "correct number of indices for $source with add_fk_index => 0");
  63   }
  64 }
  65
  66 {
  67     {
  68         package # hide from PAUSE
  69             DBICTest::Schema::NoViewDefinition;
  70
  71         use base qw/DBICTest::BaseResult/;
  72
  73         __PACKAGE__->table_class('DBIx::Class::ResultSource::View');
  74         __PACKAGE__->table('noviewdefinition');
  75
  76         1;
  77     }
  78
  79     my $schema_invalid_view = $schema->clone;
  80     $schema_invalid_view->register_class('NoViewDefinition', 'DBICTest::Schema::NoViewDefinition');
  81
  82     throws_ok { create_schema({ schema => $schema_invalid_view }) }
  83         qr/view noviewdefinition is missing a view_definition/,
  84         'parser detects views with a view_definition';
  85 }
  86
  87 lives_ok (sub {
  88   my $sqlt_schema = create_schema ({
  89     schema => $schema,
  90     args => {
  91       parser_args => {
  92         sources => ['CD']
  93       },
  94     },
  95   });
  96
  97   is_deeply (
  98     [$sqlt_schema->get_tables ],
  99     ['cd'],
 100     'sources limitng with relationships works',
 101   );
 102
 103 });
 104
 105 done_testing;
 106
 107 sub create_schema {
 108   my $args = shift;
 109
 110   my $schema = $args->{schema};
 111   my $additional_sqltargs = $args->{args} || {};
 112
 113   my $sqltargs = {
 114     add_drop_table => 1,
 115     ignore_constraint_names => 1,
 116     ignore_index_names => 1,
 117     %{$additional_sqltargs}
 118   };
 119
 120   my $sqlt = SQL::Translator->new( $sqltargs );
 121
 122   $sqlt->parser('SQL::Translator::Parser::DBIx::Class');
 123   return $sqlt->translate({ data => $schema }) || die $sqlt->error;
 124 }
 125
 126 sub get_table {
 127     my ($sqlt_schema, $schema, $source) = @_;
 128
 129     my $table_name = $schema->source($source)->from;
 130     $table_name    = $$table_name if ref $table_name;
 131
 132     return $sqlt_schema->get_table($table_name);
 133 }