t/017_add_method_modifier.t

   1 use strict;
   2 use warnings;
   3
   4 use Test::More tests => 19;
   5 use Test::Exception;
   6
   7 use Class::MOP;
   8
   9 {
  10
  11     package BankAccount;
  12
  13     use strict;
  14     use warnings;
  15     use metaclass;
  16
  17     use Carp 'confess';
  18
  19     BankAccount->meta->add_attribute(
  20         'balance' => (
  21             accessor => 'balance',
  22             init_arg => 'balance',
  23             default  => 0
  24         )
  25     );
  26
  27     sub new { (shift)->meta->new_object(@_) }
  28
  29     sub deposit {
  30         my ( $self, $amount ) = @_;
  31         $self->balance( $self->balance + $amount );
  32     }
  33
  34     sub withdraw {
  35         my ( $self, $amount ) = @_;
  36         my $current_balance = $self->balance();
  37         ( $current_balance >= $amount )
  38             || confess "Account overdrawn";
  39         $self->balance( $current_balance - $amount );
  40     }
  41
  42     package CheckingAccount;
  43
  44     use strict;
  45     use warnings;
  46     use metaclass;
  47
  48     use base 'BankAccount';
  49
  50     CheckingAccount->meta->add_attribute(
  51         'overdraft_account' => (
  52             accessor => 'overdraft_account',
  53             init_arg => 'overdraft',
  54         )
  55     );
  56
  57     CheckingAccount->meta->add_before_method_modifier(
  58         'withdraw' => sub {
  59             my ( $self, $amount ) = @_;
  60             my $overdraft_amount = $amount - $self->balance();
  61             if ( $overdraft_amount > 0 ) {
  62                 $self->overdraft_account->withdraw($overdraft_amount);
  63                 $self->deposit($overdraft_amount);
  64             }
  65         }
  66     );
  67
  68     ::ok( CheckingAccount->meta->has_method('withdraw'),
  69         '... checking account now has a withdraw method' );
  70     ::isa_ok( CheckingAccount->meta->get_method('withdraw'),
  71         'Class::MOP::Method::Wrapped' );
  72     ::isa_ok( BankAccount->meta->get_method('withdraw'),
  73         'Class::MOP::Method' );
  74 }
  75
  76 my $savings_account = BankAccount->new( balance => 250 );
  77 isa_ok( $savings_account, 'BankAccount' );
  78
  79 is( $savings_account->balance, 250, '... got the right savings balance' );
  80 lives_ok {
  81     $savings_account->withdraw(50);
  82 }
  83 '... withdrew from savings successfully';
  84 is( $savings_account->balance, 200,
  85     '... got the right savings balance after withdrawl' );
  86 dies_ok {
  87     $savings_account->withdraw(250);
  88 }
  89 '... could not withdraw from savings successfully';
  90
  91 $savings_account->deposit(150);
  92 is( $savings_account->balance, 350,
  93     '... got the right savings balance after deposit' );
  94
  95 my $checking_account = CheckingAccount->new(
  96     balance   => 100,
  97     overdraft => $savings_account
  98 );
  99 isa_ok( $checking_account, 'CheckingAccount' );
 100 isa_ok( $checking_account, 'BankAccount' );
 101
 102 is( $checking_account->overdraft_account, $savings_account,
 103     '... got the right overdraft account' );
 104
 105 is( $checking_account->balance, 100, '... got the right checkings balance' );
 106
 107 lives_ok {
 108     $checking_account->withdraw(50);
 109 }
 110 '... withdrew from checking successfully';
 111 is( $checking_account->balance, 50,
 112     '... got the right checkings balance after withdrawl' );
 113 is( $savings_account->balance, 350,
 114     '... got the right savings balance after checking withdrawl (no overdraft)'
 115 );
 116
 117 lives_ok {
 118     $checking_account->withdraw(200);
 119 }
 120 '... withdrew from checking successfully';
 121 is( $checking_account->balance, 0,
 122     '... got the right checkings balance after withdrawl' );
 123 is( $savings_account->balance, 200,
 124     '... got the right savings balance after overdraft withdrawl' );
 125