updated changes

[catagits/Catalyst-Runtime.git] / t / class_traits.t

diff --git a/t/class_traits.t b/t/class_traits.t
index f796323..8c65b38 100644 (file)
--- a/t/class_traits.t
+++ b/t/class_traits.t
@@ -9,6 +9,26 @@ BEGIN {
 
   sub a { 'a' }
   sub b { 'b' }
+
+  package Catalyst::TraitFor::Request::Foo;
+  use Moose::Role;
+
+  sub c { 'c' }
+
+  package TestApp::TraitFor::Request::Bar;
+  use Moose::Role;
+
+  sub d { 'd' }
+
+  package Catalyst::TraitFor::Response::Foo;
+  use Moose::Role;
+
+  sub c { 'c' }
+
+  package TestApp::TraitFor::Response::Bar;
+  use Moose::Role;
+
+  sub d { 'd' }
 }
  
 {
@@ -16,8 +36,8 @@ BEGIN {
  
   use Catalyst;
 
-  __PACKAGE__->request_class_traits([qw/TestRole/]);
-  __PACKAGE__->response_class_traits([qw/TestRole/]);
+  __PACKAGE__->request_class_traits([qw/TestRole Foo Bar/]);
+  __PACKAGE__->response_class_traits([qw/TestRole Foo Bar/]);
   __PACKAGE__->stats_class_traits([qw/TestRole/]);
 
   __PACKAGE__->setup;
@@ -38,7 +58,11 @@ my ($res, $c) = ctx_request '/';
 
 is $c->req->a, 'a';
 is $c->req->b, 'b';
+is $c->req->c, 'c';
+is $c->req->d, 'd';
 is $c->res->a, 'a';
 is $c->res->b, 'b';
+is $c->res->c, 'c';
+is $c->res->d, 'd';
 
 done_testing;