Function TEST

be/src/rpc/hs2-http-test.cc:106–137

Test that the HTTP server can successfuly read chunked requests. None of our usual clients are capable of generating chunked requests directly, but we may still have to process chunked requests, eg. if the requests are being proxied through something like Apache Knox, so we use curl to test it. Unfortunately, its difficult to craft a request that Thrift will actually recognize as an rpc, so mostly

// difficult to craft a request that Thrift will actually recognize as an rpc, so mostly
// what we're really testing here is just that the server doesn't hang or crash.
TEST(ThriftHttpTest, TestChunkedRequests) {
  std::shared_ptr<TestHS2Service> service(new TestHS2Service());
  std::shared_ptr<TProcessor> hs2_http_processor(
      new ImpalaHiveServer2ServiceProcessor(service));
  int port = FindUnusedEphemeralPort();
  ThriftServer* http_server;
  ThriftServerBuilder http_builder("test-http-server", hs2_http_processor, port);
  ASSERT_OK(
      http_builder.transport_type(ThriftServer::TransportType::HTTP).Build(&http_server));
  ASSERT_OK(http_server->Start());

  string curl_output;
  // Only run this if curl is available.
  if (RunShellProcess("curl --version", &curl_output)) {
    string host = Substitute("http://127.0.0.1:$0", port);
    // Send a plain, non-chunked request.
    system(Substitute("curl -X POST -v '$0'", host).c_str());
    // Send a chunked request with a small amount of data.
    system(Substitute("curl -d somedata -H 'Transfer-Encoding: chunked' -v '$0'", host)
               .c_str());
    string filename = Substitute("$0/testdata/data/decimal_rtf_tbl.txt", IMPALA_HOME);
    EXPECT_TRUE(filesystem::exists(filename));
    // Send a chunked request with a large amount of data.
    system(
        Substitute("curl -d @$0 -H 'Transfer-Encoding: chunked' -v '$1'", filename, host)
            .c_str());
  } else {
    LOG(INFO) << "Skipping test, curl was not present: " << curl_output;
  }

  http_server->StopForTesting();
}

// Test that the HTTP server can be connected to successfully with Kerberos.
TEST(Hs2HttpTest, TestSpnego) {