From a7fa4c6a9206d6a38128d1890e6bf08f72fd67d8 Mon Sep 17 00:00:00 2001
From: Colin Kuskie <colink@perldreamer.com>
Date: Mon, 7 Aug 2006 05:00:28 +0000
Subject: [PATCH] Make testing conditional on a successful module load. Add a
 test to check the reference of the created object. Clean up at the end of the
 test so you can run it more than once at a time.

---
 t/AdSpace.t | 34 +++++++++++++++++++++++++++-------
 1 file changed, 27 insertions(+), 7 deletions(-)

diff --git a/t/AdSpace.t b/t/AdSpace.t
index 37cfa72f2..2f526ae80 100644
--- a/t/AdSpace.t
+++ b/t/AdSpace.t
@@ -13,18 +13,38 @@ use strict;
 use lib "$FindBin::Bin/lib";
 use WebGUI::Test;
 use WebGUI::Session;
-use WebGUI::AdSpace;
-use Test::More tests => 3; # increment this value for each test you create
+
+#use Test::More tests => 3;
+use Test::More;
+
+my $numTests = 4; # increment this value for each test you create
+++$numTests; ##For conditional testing on module load
+
+plan tests => $numTests;
+
+my $loaded = use_ok('WebGUI::AdSpace');
 
 my $session = WebGUI::Test->session;
+my $adSpace;
 
-my $adSpace = WebGUI::AdSpace->create($session, {name=>"Alfred"});
+SKIP: {
 
-my $data = $session->db->quickHashRef("select adSpaceId, name from adSpace where adSpaceId=?",[$adSpace->getId]);
+	skip "Unable to load WebGUI::AdSpace", $numTests-1 unless $loaded;
 
-ok(exists $data->{adSpaceId}, "create()");
-is($data->{name}, $adSpace->get("name"), "get()");
-is($data->{adSpaceId}, $adSpace->getId, "getId()");
+	my $adSpace = WebGUI::AdSpace->create($session, {name=>"Alfred"});
 
+	isa_ok($adSpace, 'WebGUI::AdSpace');
 
+	my $data = $session->db->quickHashRef("select adSpaceId, name from adSpace where adSpaceId=?",[$adSpace->getId]);
 
+	ok(exists $data->{adSpaceId}, "create()");
+	is($data->{name}, $adSpace->get("name"), "get()");
+	is($data->{adSpaceId}, $adSpace->getId, "getId()");
+
+}
+
+END {
+	if (defined $adSpace and ref $adSpace eq 'WebGUI::AdSpace') {
+		$adSpace->delete;
+	}
+}