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 4e385a1..c18488e 100644 (file)
--- a/t/09sqlt-diagram.t
+++ b/t/09sqlt-diagram.t
@@ -4,11 +4,18 @@
 use strict;
 
 use File::Spec::Functions qw(catfile updir tmpdir);
-use File::Temp qw(tempfile);
+use File::Temp qw(mktemp);
 use FindBin qw($Bin);
 use Test;
+use Test::SQL::Translator qw(maybe_plan);
 
-my @script = qw(blib script sqlt-diagram.pl);
+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);
 
 my $sqlt_diagram = (-d "blib")
@@ -19,24 +26,11 @@ my $test_data = (-d "t")
     ? catfile($Bin, @data)
     : catfile($Bin, "t", @data);
 
-my (undef, $tmp) = tempfile("sqlXXXXX",
-                             OPEN   => 0,
-                             UNLINK => 1,
-                             SUFFIX => '.png',
-                             DIR    => tmpdir);
-
-BEGIN {
-    plan tests => 3;
-}
+my $tmp = mktemp('sqlXXXXX');
 
 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); 
-}
+my @cmd = ($sqlt_diagram, "-d", "MySQL", "-o", $tmp, $test_data);
+eval { system(@cmd); };
+ok(!$@ && ($? == 0));
+ok(-e $tmp); 
+eval { unlink $tmp; };