Add tests for the Server