added some notes in the tests and fixed get_from_storage to actually use the new...

[dbsrgits/DBIx-Class.git] / t / 68inflate.t

diff --git a/t/68inflate.t b/t/68inflate.t
index cc444a0..3aa428d 100644 (file)
--- a/t/68inflate.t
+++ b/t/68inflate.t
@@ -5,15 +5,16 @@ use Test::More;
 use lib qw(t/lib);
 use DBICTest;
 
-DBICTest::Schema::CD->add_column('year');
 my $schema = DBICTest->init_schema();
 
 eval { require DateTime };
 plan skip_all => "Need DateTime for inflation tests" if $@;
 
-plan tests => 15;
+plan tests => 21;
 
-DBICTest::Schema::CD->inflate_column( 'year',
+$schema->class('CD')
+#DBICTest::Schema::CD
+->inflate_column( 'year',
     { inflate => sub { DateTime->new( year => shift ) },
       deflate => sub { shift->year } }
 );
@@ -48,7 +49,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 +57,53 @@ 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');
+
+my $copy = $cd->copy({ year => $now, title => "zemoose" });
+
+isnt( $copy->year->year, $before_year, "copy" );
+ 
 # eval { $cd->store_inflated_column('year', \'year + 1') };
 # print STDERR "ERROR: $@" if($@);
 # ok(!$@, 'store_inflated_column to "year + 1"');