X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2F29html.t;h=3796b4b8000c7d83bc3a6d1e43ada007e2d9449b;hb=6a12468d4fb21ebc69c272a8d722636b1d216e2b;hp=b4e3ebd0e04117a457bbb74785e0d659188dd122;hpb=1f6b1a7b53638e95584012b446de4a741c037966;p=dbsrgits%2FSQL-Translator.git

diff --git a/t/29html.t b/t/29html.t
index b4e3ebd..3796b4b 100644
--- a/t/29html.t
+++ b/t/29html.t
@@ -9,18 +9,20 @@
 use strict;
 use vars qw(%HANDLERS);
 use Test::More;
+use Test::SQL::Translator qw(maybe_plan);
 use SQL::Translator;
 
-my ($p, $tables, $classes);
-eval {
-    require HTML::Parser;
-    $p = HTML::Parser->new(api_version => 3);
-    $p->strict_names(1); 
-};
-if ($@) {
-    plan skip_all => "Missing HTML::Parser";
+BEGIN {
+    maybe_plan(5,
+        'HTML::Parser',
+        'SQL::Translator::Parser::MySQL',
+        'SQL::Translator::Producer::HTML');
 }
 
+my ($p, $tables, $classes);
+$p = HTML::Parser->new(api_version => 3);
+$p->strict_names(1);
+
 my $create = q|
 CREATE TABLE foo (
     int id PRIMARY KEY AUTO_INCREMENT NOT NULL,
@@ -29,20 +31,17 @@ CREATE TABLE foo (
 |;
 
 my $tr = SQL::Translator->new(parser => 'MySQL', producer => 'HTML');
-my $parsed = $tr->translate(data => $create);
+my $parsed = $tr->translate(data => $create) or die $tr->error;
 my $status;
 
 eval {
-    $status = $p->parse($parsed);    
+    $status = $p->parse($parsed);
 };
 if ($@) {
-    plan tests => 1;
+    daig $@;
     fail("Unable to parse the output!");
-    exit 1;
 }
 
-plan tests => 5;
-
 # General
 ok($parsed, "Parsed table OK");
 ok($status, "Parsed HTML OK");
@@ -50,7 +49,7 @@ ok($status, "Parsed HTML OK");
 $p->handler(start => @{$HANDLERS{count_tables}});
 $p->parse($parsed);
 
-is($tables, 2, "One table in the SQL produces 2 <table> tags");
+is($tables, 3, "One table in the SQL produces 3 <table> tags");
 $tables = $classes = 0;
 
 $p->handler(start => @{$HANDLERS{count_classes}});
@@ -95,6 +94,6 @@ BEGIN {
                     $classes++;
                 }
             }, 'tagname,attr',
-        ], 
+        ],
     );
 }