1*ec63e07aSXin Li // Copyright 2020 Google LLC 2*ec63e07aSXin Li // 3*ec63e07aSXin Li // Licensed under the Apache License, Version 2.0 (the "License"); 4*ec63e07aSXin Li // you may not use this file except in compliance with the License. 5*ec63e07aSXin Li // You may obtain a copy of the License at 6*ec63e07aSXin Li // 7*ec63e07aSXin Li // https://www.apache.org/licenses/LICENSE-2.0 8*ec63e07aSXin Li // 9*ec63e07aSXin Li // Unless required by applicable law or agreed to in writing, software 10*ec63e07aSXin Li // distributed under the License is distributed on an "AS IS" BASIS, 11*ec63e07aSXin Li // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12*ec63e07aSXin Li // See the License for the specific language governing permissions and 13*ec63e07aSXin Li // limitations under the License. 14*ec63e07aSXin Li 15*ec63e07aSXin Li #ifndef TESTS_H_ 16*ec63e07aSXin Li #define TESTS_H_ 17*ec63e07aSXin Li 18*ec63e07aSXin Li #include "../sandbox.h" // NOLINT(build/include) 19*ec63e07aSXin Li #include "curl_sapi.sapi.h" // NOLINT(build/include) 20*ec63e07aSXin Li #include "gtest/gtest.h" 21*ec63e07aSXin Li #include "absl/flags/flag.h" 22*ec63e07aSXin Li #include "absl/status/statusor.h" 23*ec63e07aSXin Li #include "sandboxed_api/util/status_matchers.h" 24*ec63e07aSXin Li 25*ec63e07aSXin Li namespace curl::tests { 26*ec63e07aSXin Li 27*ec63e07aSXin Li // Helper class that can be used to test Curl sandbox. 28*ec63e07aSXin Li class CurlTestUtils { 29*ec63e07aSXin Li protected: 30*ec63e07aSXin Li static constexpr char kUrl[] = "http://127.0.0.1/"; 31*ec63e07aSXin Li 32*ec63e07aSXin Li // Starts a mock server (only once) that will manage connections for the 33*ec63e07aSXin Li // tests. The server listens on a port asynchronously by creating a thread. 34*ec63e07aSXin Li // The port number is stored in port_. Responds with "OK" to a GET request, 35*ec63e07aSXin Li // responds with the POST request fields to a POST request. 36*ec63e07aSXin Li static void StartMockServer(); 37*ec63e07aSXin Li 38*ec63e07aSXin Li // Initializes and sets up the curl handle. 39*ec63e07aSXin Li absl::Status CurlTestSetUp(); 40*ec63e07aSXin Li // Cleans up the curl handle. 41*ec63e07aSXin Li absl::Status CurlTestTearDown(); 42*ec63e07aSXin Li 43*ec63e07aSXin Li // Performs a request to the mock server, returning the response. 44*ec63e07aSXin Li absl::StatusOr<std::string> PerformRequest(); 45*ec63e07aSXin Li 46*ec63e07aSXin Li static std::thread server_thread_; 47*ec63e07aSXin Li static int port_; 48*ec63e07aSXin Li 49*ec63e07aSXin Li std::unique_ptr<curl::CurlSapiSandbox> sandbox_; 50*ec63e07aSXin Li std::unique_ptr<curl::CurlApi> api_; 51*ec63e07aSXin Li std::unique_ptr<sapi::v::RemotePtr> curl_; 52*ec63e07aSXin Li 53*ec63e07aSXin Li private: 54*ec63e07aSXin Li std::unique_ptr<sapi::v::LenVal> chunk_; 55*ec63e07aSXin Li }; 56*ec63e07aSXin Li 57*ec63e07aSXin Li } // namespace curl::tests 58*ec63e07aSXin Li 59*ec63e07aSXin Li #endif // TESTS_H_ 60