Update tests to use maybe_plan.

[dbsrgits/SQL-Translator.git] / t / 09sqlt-diagram.t

diff --git a/t/09sqlt-diagram.t b/t/09sqlt-diagram.t
index 9e47a36..c18488e 100644 (file)
--- a/t/09sqlt-diagram.t
+++ b/t/09sqlt-diagram.t
@@ -7,6 +7,13 @@ use File::Spec::Functions qw(catfile updir tmpdir);
 use File::Temp qw(mktemp);
 use FindBin qw($Bin);
 use Test;
+use Test::SQL::Translator qw(maybe_plan);
+
+BEGIN {
+    maybe_plan(3,
+        'SQL::Translator::Parser::MySQL',
+        'SQL::Translator::Producer::Diagram');
+}
 
 my @script = qw(blib script sqlt-diagram);
 my @data = qw(data mysql Apache-Session-MySQL.sql);
@@ -21,19 +28,9 @@ my $test_data = (-d "t")
 
 my $tmp = mktemp('sqlXXXXX');
 
-BEGIN {
-    plan tests => 3;
-}
-
 ok(-e $sqlt_diagram); 
-eval { require GD; };
-if ($@ && $@ =~ /locate GD.pm in /) {
-    skip($@, "GD not installed");
-    skip($@, "GD not installed");
-} else {
-    my @cmd = ($sqlt_diagram, "-d", "MySQL", "-o", $tmp, $test_data);
-    eval { system(@cmd); };
-    ok(!$@ && ($? == 0));
-    ok(-e $tmp); 
-    eval { unlink $tmp; };
-}
+my @cmd = ($sqlt_diagram, "-d", "MySQL", "-o", $tmp, $test_data);
+eval { system(@cmd); };
+ok(!$@ && ($? == 0));
+ok(-e $tmp); 
+eval { unlink $tmp; };