Add tests for roles