X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2F68inflate.t;h=2e7c374ca27a9e8d11d0598b3a07c5543129f1da;hb=ad5d0ee95a0b857e14c822818c3fe6059d022c40;hp=cc444a094e167dfabf46baf01876c0c67d600b2a;hpb=e81a62410d9b030620169dc317c82f87c1e823bf;p=dbsrgits%2FDBIx-Class.git

diff --git a/t/68inflate.t b/t/68inflate.t
index cc444a0..2e7c374 100644
--- a/t/68inflate.t
+++ b/t/68inflate.t
@@ -11,7 +11,7 @@ my $schema = DBICTest->init_schema();
 eval { require DateTime };
 plan skip_all => "Need DateTime for inflation tests" if $@;
 
-plan tests => 15;
+plan tests => 20;
 
 DBICTest::Schema::CD->inflate_column( 'year',
     { inflate => sub { DateTime->new( year => shift ) },
@@ -48,7 +48,7 @@ my $now = DateTime->now;
 $cd->year( $now );
 $cd->update;
 
-($cd) = $schema->resultset("CD")->search( year => $now->year );
+$cd = $schema->resultset("CD")->find(3);
 is( $cd->year->year, $now->year, 'deflate ok' );
 
 # set_inflated_column test
@@ -56,25 +56,49 @@ eval { $cd->set_inflated_column('year', $now) };
 ok(!$@, 'set_inflated_column with DateTime object');
 $cd->update;
 
-($cd) = $schema->resultset("CD")->search( year => $now->year );                 
+$cd = $schema->resultset("CD")->find(3);                 
 is( $cd->year->year, $now->year, 'deflate ok' );
 
+$cd = $schema->resultset("CD")->find(3);                 
+my $before_year = $cd->year->year;
 eval { $cd->set_inflated_column('year', \'year + 1') };
-print STDERR "ERROR: $@" if($@);
 ok(!$@, 'set_inflated_column to "year + 1"');
 $cd->update;
-$cd->discard_changes();
 
-($cd) = $schema->resultset("CD")->search( year => $now->year + 1 );                 
-is( $cd->year->year, $now->year+1, 'deflate ok' );
+$cd = $schema->resultset("CD")->find(3);                 
+is( $cd->year->year, $before_year+1, 'deflate ok' );
 
 # store_inflated_column test
+$cd = $schema->resultset("CD")->find(3);                 
 eval { $cd->store_inflated_column('year', $now) };
 ok(!$@, 'store_inflated_column with DateTime object');
 $cd->update;
 
 is( $cd->year->year, $now->year, 'deflate ok' );
 
+# update tests
+$cd = $schema->resultset("CD")->find(3);                 
+eval { $cd->update({'year' => $now}) };
+ok(!$@, 'update using DateTime object ok');
+is($cd->year->year, $now->year, 'deflate ok');
+
+$cd = $schema->resultset("CD")->find(3);                 
+$before_year = $cd->year->year;
+eval { $cd->update({'year' => \'year + 1'}) };
+ok(!$@, 'update using scalarref ok');
+
+$cd = $schema->resultset("CD")->find(3);                 
+is($cd->year->year, $before_year + 1, 'deflate ok');
+
+# discard_changes test
+$cd = $schema->resultset("CD")->find(3);                 
+# inflate the year
+$before_year = $cd->year->year;
+$cd->update({ year => \'year + 1'});
+$cd->discard_changes;
+
+is($cd->year->year, $before_year + 1, 'discard_changes clears the inflated value');
+ 
 # eval { $cd->store_inflated_column('year', \'year + 1') };
 # print STDERR "ERROR: $@" if($@);
 # ok(!$@, 'store_inflated_column to "year + 1"');