t/29html.t

   1 #!/usr/local/bin/perl -w
   2 # vim: set ft=perl:
   3
   4 # This test creates an HTML::Parser instance and uses it to selectively
   5 # parse the output of the HTML producer.  Rather than try to ensure
   6 # that the produced HTML turns into a particular parse tree or anything
   7 # like that, it performs some heuristics on the output.
   8
   9 use strict;
  10 use vars qw(%HANDLERS);
  11 use Test::More;
  12 use Test::SQL::Translator qw(maybe_plan);
  13 use SQL::Translator;
  14
  15 BEGIN {
  16     maybe_plan(5,
  17         'HTML::Parser',
  18         'SQL::Translator::Parser::MySQL',
  19         'SQL::Translator::Producer::HTML');
  20 }
  21
  22 my ($p, $tables, $classes);
  23 $p = HTML::Parser->new(api_version => 3);
  24 $p->strict_names(1);
  25
  26 my $create = q|
  27 CREATE TABLE foo (
  28     int id PRIMARY KEY AUTO_INCREMENT NOT NULL,
  29     name VARCHAR(255)
  30 );
  31 |;
  32
  33 my $tr = SQL::Translator->new(parser => 'MySQL', producer => 'HTML');
  34 my $parsed = $tr->translate(data => $create) or die $tr->error;
  35 my $status;
  36
  37 eval {
  38     $status = $p->parse($parsed);
  39 };
  40 if ($@) {
  41     daig $@;
  42     fail("Unable to parse the output!");
  43 }
  44
  45 # General
  46 ok($parsed, "Parsed table OK");
  47 ok($status, "Parsed HTML OK");
  48
  49 $p->handler(start => @{$HANDLERS{count_tables}});
  50 $p->parse($parsed);
  51
  52 is($tables, 3, "One table in the SQL produces 3 <table> tags");
  53 $tables = $classes = 0;
  54
  55 $p->handler(start => @{$HANDLERS{count_classes}});
  56 $p->parse($parsed);
  57
  58 is($classes, 1, "One 'LinkTable' class");
  59 $tables = $classes = 0;
  60
  61 $p->handler(start => @{$HANDLERS{sqlfairy}});
  62 $p->parse($parsed);
  63
  64 is($classes, 1, "SQLfairy plug is alive and well ");
  65 $tables = $classes = 0;
  66
  67 # Handler functions for the parser
  68 BEGIN {
  69     %HANDLERS = (
  70         count_tables => [
  71             sub {
  72                 my $tagname = shift;
  73                 $tables++ if ($tagname eq 'table');
  74             }, 'tagname',
  75         ],
  76
  77         count_classes => [
  78             sub {
  79                 my ($tagname, $attr) = @_;
  80                 if ($tagname eq 'table' &&
  81                     $attr->{'class'} &&
  82                     $attr->{'class'} eq 'LinkTable') {
  83                     $classes++;
  84                 }
  85             }, 'tagname,attr',
  86         ],
  87
  88         sqlfairy => [
  89             sub {
  90                 my ($tagname, $attr) = @_;
  91                 if ($tagname eq 'a' &&
  92                     $attr->{'href'} &&
  93                     $attr->{'href'} =~ /sqlfairy/i) {
  94                     $classes++;
  95                 }
  96             }, 'tagname,attr',
  97         ],
  98     );
  99 }