test/has_role_test.rb in aegis-1.1.3 vs test/has_role_test.rb in aegis-1.1.4

@@ -5,25 +5,30 @@
   context "Objects that have an aegis role" do
 
     setup do 
       @guest = User.new(:role_name => "guest")
       @student = User.new(:role_name => "student")
+      @student_subclass = UserSubclass.new(:role_name => "student")
       @admin = User.new(:role_name => "admin")
     end
     
     should "know their role" do
       assert :guest, @guest.role.name
       assert :student, @student.role.name
+      assert :student_subclass, @student.role.name
       assert :admin, @admin.role.name
     end
     
     should "know if they belong to a role" do
       assert @guest.guest?
       assert !@guest.student?
       assert !@guest.admin?
       assert !@student.guest?
+      assert !@student_subclass.guest?
       assert @student.student?
+      assert @student_subclass.student?
       assert !@student.admin?
+      assert !@student_subclass.admin?
       assert !@admin.guest?
       assert !@admin.student?
       assert @admin.admin?
     end