Add tests for staff retrieval