--- t/00-model-Unit.t	2007/11/29 00:40:32	3
+++ t/00-model-Unit.t	2007/11/29 16:40:10	9
@@ -17,11 +17,25 @@
 my $system_user = Arh::CurrentUser->superuser;
 ok($system_user, "Found a system user");
 
+my $place = Arh::Model::Place->new;
+$place->create(
+	name => 'mjesto iskapanja',
+);
+
+my $campaign = Arh::Model::Campaign->new(current_user => $system_user);
+$campaign->create(
+	place => $place,
+	name => 'prvo iskapanje',
+	date_from => '2007-11-28',
+	date_to => '2008-03-15',
+);
+
 # Try testing a create
 my $o = Arh::Model::Unit->new(current_user => $system_user);
 my ($id) = $o->create(
 	name => 'test unit',
 	number => 'TEST42',
+	campaign => $campaign,
 );
 ok($id, "Unit create returned success");
 ok($o->id, "New Unit has valid id set");
@@ -31,6 +45,7 @@
 $o->create(
 	name => 'unit 2',
 	number => 'TEST2',
+	campaign => $campaign,
 );
 ok($o->id, "Unit create returned another value");
 isnt($o->id, $id, "And it is different from the previous one");
@@ -38,7 +53,7 @@
 # Searches in general
 my $collection =  Arh::Model::UnitCollection->new(current_user => $system_user);
 $collection->unlimit;
-is($collection->count, 2, "Finds two records");
+is($collection->count, 3, "Finds 3 records");
 
 # Searches in specific
 $collection->limit(column => 'id', value => $o->id);
@@ -51,5 +66,5 @@
 
 # And the other one is still there
 $collection->unlimit;
-is($collection->count, 1, "Still one left");
+is($collection->count, 2, "Still 2 left");