1 BEGIN { do "./t/lib/ANFANG.pm" or die ( $@ || $! ) }
11 package My::Schema::Result::User;
15 use base qw/DBIx::Class::Core/;
17 ### Define what our admin class is, for ensure_class_loaded()
18 my $admin_class = __PACKAGE__ . '::Admin';
20 __PACKAGE__->table('users');
22 __PACKAGE__->add_columns(
23 user_id => { retrieve_on_insert => 1 },
24 qw( email password firstname lastname active admin ),
27 __PACKAGE__->set_primary_key('user_id');
31 my $ret = $self->next::method(@_);
32 if ( $ret->admin ) { ### If this is an admin, rebless for extra functions
33 $self->ensure_class_loaded($admin_class);
34 bless $ret, $admin_class;
40 return "I am a regular user.";
43 package My::Schema::Result::User::Admin;
47 use base qw/My::Schema::Result::User/;
49 # This line is important
50 __PACKAGE__->table('users');
53 return "I am an admin.";
57 return "I am doing admin stuff";
62 use base qw/DBICTest::BaseSchema/;
64 My::Schema->register_class( Admin => 'My::Schema::Result::User::Admin' );
65 My::Schema->register_class( User => 'My::Schema::Result::User' );
71 email => 'someguy@place.com',
77 email => 'someadmin@adminplace.com',
82 ok( my $schema = My::Schema->connect(DBICTest->_database) );
85 $schema->storage->dbh->do(
86 "create table users (user_id, email, password, firstname, lastname, active, admin)"
91 my $user = $schema->resultset('User')->create($user_data);
92 my $admin = $schema->resultset('User')->create($admin_data);
94 is( ref $user, 'My::Schema::Result::User' );
96 local $TODO = 'New objects should also be inflated';
97 is( ref $admin, 'My::Schema::Result::User::Admin' );
100 my $user = $schema->resultset('User')->single($user_data);
101 my $admin = $schema->resultset('User')->single($admin_data);
103 is( ref $user, 'My::Schema::Result::User' );
104 is( ref $admin, 'My::Schema::Result::User::Admin' );
106 is( $user->password, 'pass1' );
107 is( $admin->password, 'pass2' );
108 is( $user->hello, 'I am a regular user.' );
109 is( $admin->hello, 'I am an admin.' );
111 ok( !$user->can('do_admin_stuff') );
112 ok( $admin->can('do_admin_stuff') );