--- t/00-model-UnitMaterial.t	2007/11/30 16:43:53	14
+++ t/00-model-UnitMaterial.t	2007/11/30 20:13:06	22
@@ -8,7 +8,7 @@
 
 =cut
 
-use Jifty::Test tests => 11;
+use Jifty::Test tests => 20;
 
 # Make sure we can load the model
 use_ok('Arh::Model::UnitMaterial');
@@ -17,15 +17,48 @@
 my $system_user = Arh::CurrentUser->superuser;
 ok($system_user, "Found a system user");
 
+ok(my $place = Arh::Model::Place->new, 'Place' );
+ok($place->create(
+	name => 'mjesto iskapanja',
+), 'create' );
+
+ok(my $campaign = Arh::Model::Campaign->new, 'Campaign' );
+ok($campaign->create(
+	place => $place,
+	name => 'prvo iskapanje',
+	date_from => '2007-11-28',
+	date_to => '2008-03-15',
+), 'create' );
+
+ok( my $unit = Arh::Model::Unit->new, 'Unit' );
+ok( $unit->create(
+	name => 'test unit',
+	number => 42,
+	campaign => $campaign,
+), 'create' );
+
+ok( my $material = Arh::Model::Material->new, 'Material' );
+ok( $material->create(
+	name => 'test material',
+), 'create' );
+
 # Try testing a create
 my $o = Arh::Model::UnitMaterial->new(current_user => $system_user);
-my ($id) = $o->create();
+my ($id) = $o->create(
+	unit => $unit,
+	material => $material,
+);
 ok($id, "UnitMaterial create returned success");
 ok($o->id, "New UnitMaterial has valid id set");
 is($o->id, $id, "Create returned the right id");
 
+ok( $material->create( name => 'another' ), 'another material' );
+
 # And another
-$o->create();
+$o->create(
+	unit => $unit,
+	material => $material,
+);
 ok($o->id, "UnitMaterial create returned another value");
 isnt($o->id, $id, "And it is different from the previous one");