Add test role management