60 lines
1.9 KiB
C++
60 lines
1.9 KiB
C++
// 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 "absl/flags/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_
|