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;
  10   plan skip_all =>
  11       'Test needs SQL::Translator ' . DBIx::Class->_sqlt_minimum_version
  12     if not DBIx::Class->_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 are views
  19 #
  20 my @sources = grep
  21   { $_ !~ /^ (?: Dummy | CustomSql | Year\d{4}CDs ) $/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 done_testing;
  66
  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 }
  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 }