--- lib/Arh/Test.pm	2007/12/06 16:57:13	27
+++ lib/Arh/Test.pm	2007/12/10 14:56:12	28
@@ -3,31 +3,74 @@
 use strict;
 use warnings;
 
+my $uid = 0;
 
-sub create_unit {
+sub unique {
+	my ( $self, $value ) = @_;
+	return $value . ' ' . $uid++;
+}
+
+sub Place {
+	my $self = shift;
 
 	my $place = Arh::Model::Place->new;
 	$place->create(
-		name => 'mjesto iskapanja',
-	) || die "can't create place";
+		name => $self->unique('mjesto iskapanja'),
+	) || die "can't create";
+	return $place;
+}
+
+sub Campaign {
+	my $self = shift;
 
 	my $campaign = Arh::Model::Campaign->new;
 	$campaign->create(
-		place => $place,
-		name => 'prvo iskapanje',
+		place => $self->Place,
+		name => $self->unique('iskapanje'),
 		date_from => '2007-11-28',
 		date_to => '2008-03-15',
-	) || die "can't create capaign";
+	) || die "can't create";
+
+	return $campaign;
+}
+
+sub Unit {
+	my $self = shift;
 
 	my $unit = Arh::Model::Unit->new;
 	$unit->create(
-		name => 'test unit',
+		name => $self->unique('test unit'),
 		inv_id => 42,
-		campaign => $campaign,
-	) || die "can't create unit";
+		campaign => $self->Campaign,
+	) || die "can't create";
 
 	return $unit;
 
 }
 
+sub PictureType {
+	my $self = shift;
+
+	my $type = Arh::Model::PictureType->new;
+	$type->create(
+		name => $self->unique('slika'),
+	) || die "can't create";
+
+	return $type;
+
+}
+
+sub Picture {
+	my $self = shift;
+
+	my $picture = Arh::Model::Picture->new;
+	$picture->create(
+		filename => $self->unique('foo') . '.jpg',
+		campaign => $self->Campaign,
+		type => $self->PictureType,
+	) || die "can't create";
+
+	return $picture;
+}
+
 1;