Add more test coverage for the server.