Add ward assignment tests for roles