Add tests for staff retrieval

please split into different methods