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 qw/user_id email password
24 firstname lastname active
28 __PACKAGE__->set_primary_key('user_id');
32 my $ret = $self->next::method(@_);
33 if ( $ret->admin ) { ### If this is an admin, rebless for extra functions
34 $self->ensure_class_loaded($admin_class);
35 bless $ret, $admin_class;
41 return "I am a regular user.";
44 package My::Schema::Result::User::Admin;
48 use base qw/My::Schema::Result::User/;
50 # This line is important
51 __PACKAGE__->table('users');
54 return "I am an admin.";
58 return "I am doing admin stuff";
63 use base qw/DBICTest::BaseSchema/;
65 My::Schema->register_class( Admin => 'My::Schema::Result::User::Admin' );
66 My::Schema->register_class( User => 'My::Schema::Result::User' );
72 email => 'someguy@place.com',
78 email => 'someadmin@adminplace.com',
83 ok( my $schema = My::Schema->connect(DBICTest->_database) );
86 $schema->storage->dbh->do(
87 "create table users (user_id, email, password, firstname, lastname, active, admin)"
92 my $user = $schema->resultset('User')->create($user_data);
93 my $admin = $schema->resultset('User')->create($admin_data);
95 is( ref $user, 'My::Schema::Result::User' );
97 local $TODO = 'New objects should also be inflated';
98 is( ref $admin, 'My::Schema::Result::User::Admin' );
101 my $user = $schema->resultset('User')->single($user_data);
102 my $admin = $schema->resultset('User')->single($admin_data);
104 is( ref $user, 'My::Schema::Result::User' );
105 is( ref $admin, 'My::Schema::Result::User::Admin' );
107 is( $user->password, 'pass1' );
108 is( $admin->password, 'pass2' );
109 is( $user->hello, 'I am a regular user.' );
110 is( $admin->hello, 'I am an admin.' );
112 ok( !$user->can('do_admin_stuff') );
113 ok( $admin->can('do_admin_stuff') );