[gitmo/Moose.git] / t / 000_recipes / 004_company.t

#!/usr/bin/perl

use strict;
use warnings;

use Test::More; 

BEGIN {
    eval "use Regexp::Common; use Locale::US;";
    plan skip_all => "Regexp::Common & Locale::US required for this test" if $@;        
    plan tests => 66;    
}

use Test::Exception;
use Scalar::Util 'isweak';

BEGIN {
    use_ok('Moose');           
}

{
    package Address;
    use Moose;
    use Moose::Util::TypeConstraints;
    
    use Locale::US;
    use Regexp::Common 'zip';
    
    my $STATES = Locale::US->new;
    
    subtype USState 
        => as Str
        => where {
            (exists $STATES->{code2state}{uc($_)} || exists $STATES->{state2code}{uc($_)})
        };
        
    subtype USZipCode 
        => as Value
        => where {
            /^$RE{zip}{US}{-extended => 'allow'}$/            
        };
    
    has 'street'   => (is => 'rw', isa => 'Str');
    has 'city'     => (is => 'rw', isa => 'Str');
    has 'state'    => (is => 'rw', isa => 'USState');
    has 'zip_code' => (is => 'rw', isa => 'USZipCode');   
    
    __PACKAGE__->meta->make_immutable(debug => 0);
}{
    
    package Company;
    use Moose;
    use Moose::Util::TypeConstraints;    
    
    has 'name'      => (is => 'rw', isa => 'Str', required => 1);
    has 'address'   => (is => 'rw', isa => 'Address'); 
    has 'employees' => (is => 'rw', isa => 'ArrayRef[Employee]');    
    
    sub BUILD {
        my ($self, $params) = @_;
        if ($params->{employees}) {
            foreach my $employee (@{$params->{employees}}) {
                $employee->company($self);
            }
        }
    }
    
    sub get_employee_count { scalar @{(shift)->employees} }
    
    after 'employees' => sub {
        my ($self, $employees) = @_;
        # if employees is defined, it 
        # has already been type checked
        if (defined $employees) {
            # make sure each gets the 
            # weak ref to the company
            foreach my $employee (@{$employees}) {
                $employee->company($self);
            }            
        }
    };
    
    __PACKAGE__->meta->make_immutable(debug => 0);
}{    
    
    package Person;
    use Moose;
    
    has 'first_name'     => (is => 'rw', isa => 'Str', required => 1);
    has 'last_name'      => (is => 'rw', isa => 'Str', required => 1);       
    has 'middle_initial' => (is => 'rw', isa => 'Str', predicate => 'has_middle_initial');  
    has 'address'        => (is => 'rw', isa => 'Address');
    
    sub full_name {
        my $self = shift;
        return $self->first_name . 
              ($self->has_middle_initial ? ' ' . $self->middle_initial . '. ' : ' ') .
               $self->last_name;
    }

    __PACKAGE__->meta->make_immutable(debug => 0);
}{
      
    package Employee;
    use Moose;  
    
    extends 'Person';
    
    has 'title'   => (is => 'rw', isa => 'Str', required => 1);
    has 'company' => (is => 'rw', isa => 'Company', weak_ref => 1);  
    
    override 'full_name' => sub {
        my $self = shift;
        super() . ', ' . $self->title
    };
    
    __PACKAGE__->meta->make_immutable(debug => 0);
}

my $ii;
lives_ok {
    $ii = Company->new({
        name    => 'Infinity Interactive',
        address => Address->new(
            street   => '565 Plandome Rd., Suite 307',
            city     => 'Manhasset',
            state    => 'NY',
            zip_code => '11030'
        ),
        employees => [
            Employee->new(
                first_name     => 'Jeremy',
                last_name      => 'Shao',
                title          => 'President / Senior Consultant',
                address        => Address->new(city => 'Manhasset', state => 'NY')
            ),
            Employee->new(
                first_name     => 'Tommy',
                last_name      => 'Lee',
                title          => 'Vice President / Senior Developer',
                address        => Address->new(city => 'New York', state => 'NY')
            ),        
            Employee->new(
                first_name     => 'Stevan',
                middle_initial => 'C',
                last_name      => 'Little',
                title          => 'Senior Developer',            
                address        => Address->new(city => 'Madison', state => 'CT')
            ),          
        ]
    });
} '... created the entire company successfully';
isa_ok($ii, 'Company');

is($ii->name, 'Infinity Interactive', '... got the right name for the company');

isa_ok($ii->address, 'Address');
is($ii->address->street, '565 Plandome Rd., Suite 307', '... got the right street address');
is($ii->address->city, 'Manhasset', '... got the right city');
is($ii->address->state, 'NY', '... got the right state');
is($ii->address->zip_code, 11030, '... got the zip code');

is($ii->get_employee_count, 3, '... got the right employee count');

# employee #1

isa_ok($ii->employees->[0], 'Employee');
isa_ok($ii->employees->[0], 'Person');

is($ii->employees->[0]->first_name, 'Jeremy', '... got the right first name');
is($ii->employees->[0]->last_name, 'Shao', '... got the right last name');
ok(!$ii->employees->[0]->has_middle_initial, '... no middle initial');
is($ii->employees->[0]->middle_initial, undef, '... got the right middle initial value');
is($ii->employees->[0]->full_name, 'Jeremy Shao, President / Senior Consultant', '... got the right full name');
is($ii->employees->[0]->title, 'President / Senior Consultant', '... got the right title');
is($ii->employees->[0]->company, $ii, '... got the right company');
ok(isweak($ii->employees->[0]->{company}), '... the company is a weak-ref');

isa_ok($ii->employees->[0]->address, 'Address');
is($ii->employees->[0]->address->city, 'Manhasset', '... got the right city');
is($ii->employees->[0]->address->state, 'NY', '... got the right state');

# employee #2

isa_ok($ii->employees->[1], 'Employee');
isa_ok($ii->employees->[1], 'Person');

is($ii->employees->[1]->first_name, 'Tommy', '... got the right first name');
is($ii->employees->[1]->last_name, 'Lee', '... got the right last name');
ok(!$ii->employees->[1]->has_middle_initial, '... no middle initial');
is($ii->employees->[1]->middle_initial, undef, '... got the right middle initial value');
is($ii->employees->[1]->full_name, 'Tommy Lee, Vice President / Senior Developer', '... got the right full name');
is($ii->employees->[1]->title, 'Vice President / Senior Developer', '... got the right title');
is($ii->employees->[1]->company, $ii, '... got the right company');
ok(isweak($ii->employees->[1]->{company}), '... the company is a weak-ref');

isa_ok($ii->employees->[1]->address, 'Address');
is($ii->employees->[1]->address->city, 'New York', '... got the right city');
is($ii->employees->[1]->address->state, 'NY', '... got the right state');

# employee #3

isa_ok($ii->employees->[2], 'Employee');
isa_ok($ii->employees->[2], 'Person');

is($ii->employees->[2]->first_name, 'Stevan', '... got the right first name');
is($ii->employees->[2]->last_name, 'Little', '... got the right last name');
ok($ii->employees->[2]->has_middle_initial, '... got middle initial');
is($ii->employees->[2]->middle_initial, 'C', '... got the right middle initial value');
is($ii->employees->[2]->full_name, 'Stevan C. Little, Senior Developer', '... got the right full name');
is($ii->employees->[2]->title, 'Senior Developer', '... got the right title');
is($ii->employees->[2]->company, $ii, '... got the right company');
ok(isweak($ii->employees->[2]->{company}), '... the company is a weak-ref');

isa_ok($ii->employees->[2]->address, 'Address');
is($ii->employees->[2]->address->city, 'Madison', '... got the right city');
is($ii->employees->[2]->address->state, 'CT', '... got the right state');

# create new company

my $new_company = Company->new(name => 'Infinity Interactive International');
isa_ok($new_company, 'Company');

my $ii_employees = $ii->employees;
foreach my $employee (@$ii_employees) {
    is($employee->company, $ii, '... has the ii company');
}

$new_company->employees($ii_employees);

foreach my $employee (@{$new_company->employees}) {
    is($employee->company, $new_company, '... has the different company now');
}

## check some error conditions for the subtypes

dies_ok {
    Address->new(street => {}),    
} '... we die correctly with bad args';

dies_ok {
    Address->new(city => {}),    
} '... we die correctly with bad args';

dies_ok {
    Address->new(state => 'British Columbia'),    
} '... we die correctly with bad args';

lives_ok {
    Address->new(state => 'Connecticut'),    
} '... we live correctly with good args';

dies_ok {
    Address->new(zip_code => 'AF5J6$'),    
} '... we die correctly with bad args';

lives_ok {
    Address->new(zip_code => '06443'),    
} '... we live correctly with good args';

dies_ok {
    Company->new(),    
} '... we die correctly without good args';

lives_ok {
    Company->new(name => 'Foo'),    
} '... we live correctly without good args';

dies_ok {
    Company->new(name => 'Foo', employees => [ Person->new ]),    
} '... we die correctly with good args';

lives_ok {
    Company->new(name => 'Foo', employees => []),    
} '... we live correctly with good args';
Commit	Line	Data
5569c072	1	#!/usr/bin/perl
	2
	3	use strict;
	4	use warnings;
	5
51c5e0fb	6	use Test::More;
	7
	8	BEGIN {
	9	eval "use Regexp::Common; use Locale::US;";
	10	plan skip_all => "Regexp::Common & Locale::US required for this test" if $@;
638585e1	11	plan tests => 66;
51c5e0fb	12	}
51c5e0fb	13
5569c072	14	use Test::Exception;
51c5e0fb	15	use Scalar::Util 'isweak';
51c5e0fb	16
5569c072	17	BEGIN {
	18	use_ok('Moose');
	19	}
	20
	21	{
	22	package Address;
5569c072	23	use Moose;
05d9eaf6	24	use Moose::Util::TypeConstraints;
5569c072	25
	26	use Locale::US;
	27	use Regexp::Common 'zip';
	28
	29	my $STATES = Locale::US->new;
	30
	31	subtype USState
	32	=> as Str
	33	=> where {
	34	(exists $STATES->{code2state}{uc($_)} \|\| exists $STATES->{state2code}{uc($_)})
	35	};
	36
	37	subtype USZipCode
	38	=> as Value
	39	=> where {
	40	/^$RE{zip}{US}{-extended => 'allow'}$/
	41	};
	42
182134e8	43	has 'street' => (is => 'rw', isa => 'Str');
	44	has 'city' => (is => 'rw', isa => 'Str');
	45	has 'state' => (is => 'rw', isa => 'USState');
	46	has 'zip_code' => (is => 'rw', isa => 'USZipCode');
5569c072	47
5a3217de	48	__PACKAGE__->meta->make_immutable(debug => 0);
f1917f58	49	}{
	50
	51	package Company;
	52	use Moose;
	53	use Moose::Util::TypeConstraints;
	54
	55	has 'name' => (is => 'rw', isa => 'Str', required => 1);
	56	has 'address' => (is => 'rw', isa => 'Address');
	57	has 'employees' => (is => 'rw', isa => 'ArrayRef[Employee]');
	58
	59	sub BUILD {
	60	my ($self, $params) = @_;
	61	if ($params->{employees}) {
	62	foreach my $employee (@{$params->{employees}}) {
	63	$employee->company($self);
	64	}
	65	}
	66	}
	67
	68	sub get_employee_count { scalar @{(shift)->employees} }
	69
	70	after 'employees' => sub {
	71	my ($self, $employees) = @_;
	72	# if employees is defined, it
	73	# has already been type checked
	74	if (defined $employees) {
	75	# make sure each gets the
	76	# weak ref to the company
	77	foreach my $employee (@{$employees}) {
	78	$employee->company($self);
	79	}
	80	}
	81	};
	82
5a3217de	83	__PACKAGE__->meta->make_immutable(debug => 0);
5cf3dbcf	84	}{
5cf3dbcf	85
5569c072	86	package Person;
5569c072	87	use Moose;
5569c072	88
7c6cacb4	89	has 'first_name' => (is => 'rw', isa => 'Str', required => 1);
7c6cacb4	90	has 'last_name' => (is => 'rw', isa => 'Str', required => 1);
182134e8	91	has 'middle_initial' => (is => 'rw', isa => 'Str', predicate => 'has_middle_initial');
5569c072	92	has 'address' => (is => 'rw', isa => 'Address');
	93
	94	sub full_name {
	95	my $self = shift;
	96	return $self->first_name .
	97	($self->has_middle_initial ? ' ' . $self->middle_initial . '. ' : ' ') .
	98	$self->last_name;
	99	}
5cf3dbcf	100
5a3217de	101	__PACKAGE__->meta->make_immutable(debug => 0);
5cf3dbcf	102	}{
5569c072	103
5569c072	104	package Employee;
5569c072	105	use Moose;
	106
	107	extends 'Person';
	108
7c6cacb4	109	has 'title' => (is => 'rw', isa => 'Str', required => 1);
5569c072	110	has 'company' => (is => 'rw', isa => 'Company', weak_ref => 1);
b6fe348f	111
	112	override 'full_name' => sub {
	113	my $self = shift;
	114	super() . ', ' . $self->title
	115	};
5cf3dbcf	116
5a3217de	117	__PACKAGE__->meta->make_immutable(debug => 0);
5569c072	118	}
	119
	120	my $ii;
	121	lives_ok {
05d9eaf6	122	$ii = Company->new({
5569c072	123	name => 'Infinity Interactive',
	124	address => Address->new(
	125	street => '565 Plandome Rd., Suite 307',
	126	city => 'Manhasset',
	127	state => 'NY',
	128	zip_code => '11030'
	129	),
	130	employees => [
	131	Employee->new(
	132	first_name => 'Jeremy',
	133	last_name => 'Shao',
	134	title => 'President / Senior Consultant',
	135	address => Address->new(city => 'Manhasset', state => 'NY')
	136	),
	137	Employee->new(
	138	first_name => 'Tommy',
	139	last_name => 'Lee',
	140	title => 'Vice President / Senior Developer',
	141	address => Address->new(city => 'New York', state => 'NY')
	142	),
	143	Employee->new(
	144	first_name => 'Stevan',
	145	middle_initial => 'C',
	146	last_name => 'Little',
	147	title => 'Senior Developer',
	148	address => Address->new(city => 'Madison', state => 'CT')
638585e1	149	),
5569c072	150	]
05d9eaf6	151	});
5569c072	152	} '... created the entire company successfully';
	153	isa_ok($ii, 'Company');
	154
	155	is($ii->name, 'Infinity Interactive', '... got the right name for the company');
	156
	157	isa_ok($ii->address, 'Address');
	158	is($ii->address->street, '565 Plandome Rd., Suite 307', '... got the right street address');
	159	is($ii->address->city, 'Manhasset', '... got the right city');
	160	is($ii->address->state, 'NY', '... got the right state');
	161	is($ii->address->zip_code, 11030, '... got the zip code');
	162
638585e1	163	is($ii->get_employee_count, 3, '... got the right employee count');
5569c072	164
	165	# employee #1
	166
	167	isa_ok($ii->employees->[0], 'Employee');
	168	isa_ok($ii->employees->[0], 'Person');
	169
	170	is($ii->employees->[0]->first_name, 'Jeremy', '... got the right first name');
	171	is($ii->employees->[0]->last_name, 'Shao', '... got the right last name');
	172	ok(!$ii->employees->[0]->has_middle_initial, '... no middle initial');
	173	is($ii->employees->[0]->middle_initial, undef, '... got the right middle initial value');
b6fe348f	174	is($ii->employees->[0]->full_name, 'Jeremy Shao, President / Senior Consultant', '... got the right full name');
5569c072	175	is($ii->employees->[0]->title, 'President / Senior Consultant', '... got the right title');
5569c072	176	is($ii->employees->[0]->company, $ii, '... got the right company');
51c5e0fb	177	ok(isweak($ii->employees->[0]->{company}), '... the company is a weak-ref');
5569c072	178
	179	isa_ok($ii->employees->[0]->address, 'Address');
	180	is($ii->employees->[0]->address->city, 'Manhasset', '... got the right city');
	181	is($ii->employees->[0]->address->state, 'NY', '... got the right state');
	182
	183	# employee #2
	184
	185	isa_ok($ii->employees->[1], 'Employee');
	186	isa_ok($ii->employees->[1], 'Person');
	187
	188	is($ii->employees->[1]->first_name, 'Tommy', '... got the right first name');
	189	is($ii->employees->[1]->last_name, 'Lee', '... got the right last name');
	190	ok(!$ii->employees->[1]->has_middle_initial, '... no middle initial');
	191	is($ii->employees->[1]->middle_initial, undef, '... got the right middle initial value');
b6fe348f	192	is($ii->employees->[1]->full_name, 'Tommy Lee, Vice President / Senior Developer', '... got the right full name');
5569c072	193	is($ii->employees->[1]->title, 'Vice President / Senior Developer', '... got the right title');
5569c072	194	is($ii->employees->[1]->company, $ii, '... got the right company');
51c5e0fb	195	ok(isweak($ii->employees->[1]->{company}), '... the company is a weak-ref');
5569c072	196
	197	isa_ok($ii->employees->[1]->address, 'Address');
	198	is($ii->employees->[1]->address->city, 'New York', '... got the right city');
	199	is($ii->employees->[1]->address->state, 'NY', '... got the right state');
	200
	201	# employee #3
	202
	203	isa_ok($ii->employees->[2], 'Employee');
	204	isa_ok($ii->employees->[2], 'Person');
	205
	206	is($ii->employees->[2]->first_name, 'Stevan', '... got the right first name');
	207	is($ii->employees->[2]->last_name, 'Little', '... got the right last name');
	208	ok($ii->employees->[2]->has_middle_initial, '... got middle initial');
	209	is($ii->employees->[2]->middle_initial, 'C', '... got the right middle initial value');
b6fe348f	210	is($ii->employees->[2]->full_name, 'Stevan C. Little, Senior Developer', '... got the right full name');
5569c072	211	is($ii->employees->[2]->title, 'Senior Developer', '... got the right title');
5569c072	212	is($ii->employees->[2]->company, $ii, '... got the right company');
51c5e0fb	213	ok(isweak($ii->employees->[2]->{company}), '... the company is a weak-ref');
5569c072	214
	215	isa_ok($ii->employees->[2]->address, 'Address');
	216	is($ii->employees->[2]->address->city, 'Madison', '... got the right city');
	217	is($ii->employees->[2]->address->state, 'CT', '... got the right state');
	218
ad5ed80c	219	# create new company
	220
	221	my $new_company = Company->new(name => 'Infinity Interactive International');
	222	isa_ok($new_company, 'Company');
	223
	224	my $ii_employees = $ii->employees;
	225	foreach my $employee (@$ii_employees) {
	226	is($employee->company, $ii, '... has the ii company');
	227	}
	228
	229	$new_company->employees($ii_employees);
	230
	231	foreach my $employee (@{$new_company->employees}) {
	232	is($employee->company, $new_company, '... has the different company now');
	233	}
	234
5569c072	235	## check some error conditions for the subtypes
	236
	237	dies_ok {
51c5e0fb	238	Address->new(street => {}),
	239	} '... we die correctly with bad args';
	240
	241	dies_ok {
	242	Address->new(city => {}),
	243	} '... we die correctly with bad args';
	244
	245	dies_ok {
5569c072	246	Address->new(state => 'British Columbia'),
	247	} '... we die correctly with bad args';
	248
	249	lives_ok {
	250	Address->new(state => 'Connecticut'),
	251	} '... we live correctly with good args';
	252
	253	dies_ok {
	254	Address->new(zip_code => 'AF5J6$'),
	255	} '... we die correctly with bad args';
	256
	257	lives_ok {
	258	Address->new(zip_code => '06443'),
	259	} '... we live correctly with good args';
	260
	261	dies_ok {
7c6cacb4	262	Company->new(),
	263	} '... we die correctly without good args';
	264
	265	lives_ok {
	266	Company->new(name => 'Foo'),
	267	} '... we live correctly without good args';
	268
	269	dies_ok {
	270	Company->new(name => 'Foo', employees => [ Person->new ]),
5569c072	271	} '... we die correctly with good args';
	272
	273	lives_ok {
7c6cacb4	274	Company->new(name => 'Foo', employees => []),
5569c072	275	} '... we live correctly with good args';
5569c072	276