public PingResponse ping() {
                PingResponse response = new PingResponse(createMockHealthResponse("test-cluster", ClusterHealthStatus.RED)) {
                    @Override
                    public int getStatus() {
                        // RED status should return non-zero status
                        return 2;
                    }
                };
                return response;
            }
        };

    }

    private static class TestHttpServletResponse implements HttpServletResponse {
        private int status = 0;

        @Override
        public void setStatus(int sc) {
            this.status = sc;
        }

        public int getStatus() {
            return status;
        }

        @Override
        public java.util.Collection<String> getHeaderNames() {

                    continue;
                }

                // check status
                for (int i = 0; i < startedCrawlerNum; i++) {
                    if (crawlerList.get(i).getCrawlerContext().getStatus() == CrawlerStatus.DONE
                            && Constants.RUNNING.equals(crawlerStatusList.get(i))) {
                        crawlerList.get(i).awaitTermination();

}

TEST(CAPI, TestRemoteExecuteUpdateServerDefResourceAccessAsync) {
  TestRemoteExecuteUpdateServerDefResourceAccess(true);
}

void TestRemoteExecuteUpdateServerDefWithFailures(bool async) {
  // Fail fast on GetStatus requests so we can get errors instead of timeout
  // when updating cluster with non-exsitent worker
  tensorflow::setenv("GRPC_FAIL_FAST", "TRUE", /*overwrite=*/1);

  tensorflow::ServerDef server_def = GetServerDef(2);

        manager.process(request, response, chain);
        assertEquals(404, response.getStatus());

        // Test with different status code
        manager = new StatusCodeWebApiManager(201);
        response = new TestHttpServletResponse();
        manager.process(request, response, chain);
        assertEquals(201, response.getStatus());
    }

    @Test

        if (logger.isDebugEnabled()) {
            logger.debug("[{}] refresh status: {} ({}/{}/{})", index, response.getStatus(), response.getTotalShards(),
                    response.getSuccessfulShards(), response.getFailedShards());
        }
        return response.getStatus().getStatus();
    }

    /**
     * Calculates the memory size of a document data map.

            @Override
            public void addIntHeader(String name, int value) {
            }

            @Override
            public void setStatus(int sc) {
            }

            @Override
            public int getStatus() {
                return 0;
            }

            @Override
            public String getHeader(String name) {
                return null;

            return "http://test.example.com";
        }

        @Override
        public String getParentUrl() {
            return null;
        }

        @Override
        public Integer getStatus() {
            return 200;
        }

        @Override
        public Integer getHttpStatusCode() {
            return 200;
        }

        @Override
        public String getMethod() {

        final SearchEngineClient searchEngineClient = ComponentUtil.getSearchEngineClient();
        try {
            final PingResponse pingResponse = searchEngineClient.ping();
            writeJsonResponse(pingResponse.getStatus() == 0 ? HttpServletResponse.SC_OK : HttpServletResponse.SC_SERVICE_UNAVAILABLE,
                    "\"data\":" + pingResponse.getMessage());
        } catch (final Exception e) {
            if (logger.isDebugEnabled()) {

     * @param response the HTTP response
     * @throws IOException if an I/O error occurs
     */
    protected void sendBlockedResponse(final HttpServletResponse response) throws IOException {
        response.setStatus(HttpServletResponse.SC_FORBIDDEN);
        response.setContentType("application/json;charset=UTF-8");
        response.getWriter().write("{\"error\":\"ip_blocked\",\"message\":\"Access denied.\"}");
    }

    /**