cleanup
[gitmo/Moose.git] / t / 002_basic.t
1 #!/usr/bin/perl
2
3 use strict;
4 use warnings;
5
6 use Test::More tests => 24;
7 use Test::Exception;
8
9 BEGIN {
10     use_ok('Moose');           
11 }
12
13 {
14     package BankAccount;
15         use strict;
16         use warnings;
17     use Moose;
18     
19     has 'balance' => (isa => 'Int', is => 'rw', default => 0);
20
21     sub deposit {
22         my ($self, $amount) = @_;
23         $self->balance($self->balance + $amount);
24     }
25     
26     sub withdraw {
27         my ($self, $amount) = @_;
28         my $current_balance = $self->balance();
29         ($current_balance >= $amount)
30             || confess "Account overdrawn";
31         $self->balance($current_balance - $amount);
32     }
33
34         package CheckingAccount;
35         use strict;
36         use warnings;   
37         use Moose;
38
39         extends 'BankAccount';
40         
41     has 'overdraft_account' => (isa => 'BankAccount', is => 'rw');      
42
43         before 'withdraw' => sub {
44                 my ($self, $amount) = @_;
45                 my $overdraft_amount = $amount - $self->balance();
46                 if ($self->overdraft_account && $overdraft_amount > 0) {
47                         $self->overdraft_account->withdraw($overdraft_amount);
48                         $self->deposit($overdraft_amount);
49                 }
50         };
51 }
52
53 my $savings_account = BankAccount->new(balance => 250);
54 isa_ok($savings_account, 'BankAccount');
55
56 is($savings_account->balance, 250, '... got the right savings balance');
57 lives_ok {
58         $savings_account->withdraw(50);
59 } '... withdrew from savings successfully';
60 is($savings_account->balance, 200, '... got the right savings balance after withdrawl');
61
62 $savings_account->deposit(150);
63 is($savings_account->balance, 350, '... got the right savings balance after deposit');
64
65 {
66     my $checking_account = CheckingAccount->new(
67                                                         balance => 100,
68                                                         overdraft_account => $savings_account
69                                                 );
70     isa_ok($checking_account, 'CheckingAccount');
71     isa_ok($checking_account, 'BankAccount');
72
73     is($checking_account->overdraft_account, $savings_account, '... got the right overdraft account');
74
75     is($checking_account->balance, 100, '... got the right checkings balance');
76
77     lives_ok {
78         $checking_account->withdraw(50);
79     } '... withdrew from checking successfully';
80     is($checking_account->balance, 50, '... got the right checkings balance after withdrawl');
81     is($savings_account->balance, 350, '... got the right savings balance after checking withdrawl (no overdraft)');
82
83     lives_ok {
84         $checking_account->withdraw(200);
85     } '... withdrew from checking successfully';
86     is($checking_account->balance, 0, '... got the right checkings balance after withdrawl');
87     is($savings_account->balance, 200, '... got the right savings balance after overdraft withdrawl');
88 }
89
90 {
91     my $checking_account = CheckingAccount->new(
92                                                         balance => 100
93                                                         # no overdraft account
94                                                 );
95     isa_ok($checking_account, 'CheckingAccount');
96     isa_ok($checking_account, 'BankAccount');
97
98     is($checking_account->overdraft_account, undef, '... no overdraft account');
99
100     is($checking_account->balance, 100, '... got the right checkings balance');
101
102     lives_ok {
103         $checking_account->withdraw(50);
104     } '... withdrew from checking successfully';
105     is($checking_account->balance, 50, '... got the right checkings balance after withdrawl');
106
107     dies_ok {
108         $checking_account->withdraw(200);
109     } '... withdrawl failed due to attempted overdraft';
110     is($checking_account->balance, 50, '... got the right checkings balance after withdrawl failure');
111 }
112
113