def testGrantPermssion( self ): permissionManager = MockPermissionManager() user = User( None, permissionManager ) user.grantPermission( 'perm1' ) self.assertTrue( user.hasPermission( 'perm1' )) self.assertFalse( user.hasPermission( 'perm2' )) self.assertEquals([ 'perm1' ], permissionManager.grantedPermissions )
def testRemovePermission( self ): permissionManager = MockPermissionManager() user = User( None, permissionManager ) user.grantPermission( 'perm1' ) user.grantPermission( 'perm2' ) user.removePermission( 'perm1' ) self.assertFalse( user.hasPermission( 'perm1' )) self.assertTrue( user.hasPermission( 'perm2' ))