9cbaab9fe03d299a214512bb33c3d567d886ccde
[dbsrgits/DBIx-Class.git] / t / 68inflate.t
1 use strict;
2 use warnings;  
3
4 use Test::More;
5 use lib qw(t/lib);
6 use DBICTest;
7
8 DBICTest::Schema::CD->add_column('year');
9 my $schema = DBICTest->init_schema();
10
11 eval { require DateTime };
12 plan skip_all => "Need DateTime for inflation tests" if $@;
13
14 plan tests => 22;
15
16 DBICTest::Schema::CD->inflate_column( 'year',
17     { inflate => sub { DateTime->new( year => shift ) },
18       deflate => sub { shift->year } }
19 );
20 Class::C3->reinitialize;
21
22 # inflation test
23 my $cd = $schema->resultset("CD")->find(3);
24
25 is( ref($cd->year), 'DateTime', 'year is a DateTime, ok' );
26
27 is( $cd->year->year, 1997, 'inflated year ok' );
28
29 is( $cd->year->month, 1, 'inflated month ok' );
30
31 eval { $cd->year(\'year +1'); };
32 ok(!$@, 'updated year using a scalarref');
33 $cd->update();
34 $cd->discard_changes();
35
36 is( ref($cd->year), 'DateTime', 'year is still a DateTime, ok' );
37
38 is( $cd->year->year, 1998, 'updated year, bypassing inflation' );
39
40 is( $cd->year->month, 1, 'month is still 1' );  
41
42 # get_inflated_column test
43
44 is( ref($cd->get_inflated_column('year')), 'DateTime', 'get_inflated_column produces a DateTime');
45
46 # deflate test
47 my $now = DateTime->now;
48 $cd->year( $now );
49 $cd->update;
50
51 $cd = $schema->resultset("CD")->find(3);
52 is( $cd->year->year, $now->year, 'deflate ok' );
53
54 # set_inflated_column test
55 eval { $cd->set_inflated_column('year', $now) };
56 ok(!$@, 'set_inflated_column with DateTime object');
57 $cd->update;
58
59 $cd = $schema->resultset("CD")->find(3);                 
60 is( $cd->year->year, $now->year, 'deflate ok' );
61
62 $cd = $schema->resultset("CD")->find(3);                 
63 my $before_year = $cd->year->year;
64 eval { $cd->set_inflated_column('year', \'year + 1') };
65 ok(!$@, 'set_inflated_column to "year + 1"');
66 $cd->update;
67
68 $cd = $schema->resultset("CD")->find(3);                 
69 is( $cd->year->year, $before_year+1, 'deflate ok' );
70
71 # store_inflated_column test
72 $cd = $schema->resultset("CD")->find(3);                 
73 eval { $cd->store_inflated_column('year', $now) };
74 ok(!$@, 'store_inflated_column with DateTime object');
75 $cd->update;
76
77 is( $cd->year->year, $now->year, 'deflate ok' );
78
79 # update tests
80 $cd = $schema->resultset("CD")->find(3);                 
81 eval { $cd->update({'year' => $now}) };
82 ok(!$@, 'update using DateTime object ok');
83 is($cd->year->year, $now->year, 'deflate ok');
84
85 $cd = $schema->resultset("CD")->find(3);                 
86 $before_year = $cd->year->year;
87 eval { $cd->update({'year' => \'year + 1'}) };
88 ok(!$@, 'update using scalarref ok');
89
90 $cd = $schema->resultset("CD")->find(3);                 
91 is($cd->year->year, $before_year + 1, 'deflate ok');
92
93 my $upd = { 'year' => $now->truncate(to => 'month'), title => 'An Updated Title' };
94 my $upd_copy = { %$upd };
95 eval { $cd->update($upd) };
96 ok(!$@, 'update worked okay');
97 ok(exists $upd->{year} && $upd->{year} == $upd_copy->{year}, "update doesn't change passed hashref");
98
99 # discard_changes test
100 $cd = $schema->resultset("CD")->find(3);                 
101 # inflate the year
102 $before_year = $cd->year->year;
103 $cd->update({ year => \'year + 1'});
104 $cd->discard_changes;
105
106 is($cd->year->year, $before_year + 1, 'discard_changes clears the inflated value');
107  
108 # eval { $cd->store_inflated_column('year', \'year + 1') };
109 # print STDERR "ERROR: $@" if($@);
110 # ok(!$@, 'store_inflated_column to "year + 1"');
111
112 # is_deeply( $cd->year, \'year + 1', 'deflate ok' );
113