Add tests for admin functionality