sandboxed-api/oss-internship-2020/curl/tests/test_utils.h

// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef TESTS_H_
#define TESTS_H_

#include "../sandbox.h"      // NOLINT(build/include)
#include "curl_sapi.sapi.h"  // NOLINT(build/include)
#include "gtest/gtest.h"
#include "sandboxed_api/util/flag.h"
#include "absl/status/statusor.h"
#include "sandboxed_api/util/status_matchers.h"

namespace curl::tests {

// Helper class that can be used to test Curl sandbox.
class CurlTestUtils {
 protected:
  static constexpr char kUrl[] = "http://127.0.0.1/";

  // Starts a mock server (only once) that will manage connections for the
  // tests. The server listens on a port asynchronously by creating a thread.
  // The port number is stored in port_. Responds with "OK" to a GET request,
  // responds with the POST request fields to a POST request.
  static void StartMockServer();

  // Initializes and sets up the curl handle.
  absl::Status CurlTestSetUp();
  // Cleans up the curl handle.
  absl::Status CurlTestTearDown();

  // Performs a request to the mock server, returning the response.
  absl::StatusOr<std::string> PerformRequest();

  static std::thread server_thread_;
  static int port_;

  std::unique_ptr<curl::CurlSapiSandbox> sandbox_;
  std::unique_ptr<curl::CurlApi> api_;
  std::unique_ptr<sapi::v::RemotePtr> curl_;

 private:
  std::unique_ptr<sapi::v::LenVal> chunk_;
};

}  // namespace curl::tests

#endif  // TESTS_H_