mirror of https://gitee.com/bigwinds/arangodb
94 lines
3.4 KiB
C++
94 lines
3.4 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test suite for CuckooFilter based index selectivity estimator
|
|
///
|
|
/// @file
|
|
///
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2017 ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// 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
|
|
///
|
|
/// http://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.
|
|
///
|
|
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// @author Andreas Streichardt
|
|
/// @author Copyright 2017, ArangoDB GmbH, Cologne, Germany
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "catch.hpp"
|
|
|
|
#include "Rest/HttpRequest.h"
|
|
#include "SimpleHttpClient/Communicator.h"
|
|
#include "SimpleHttpClient/Callbacks.h"
|
|
#include "SimpleHttpClient/Destination.h"
|
|
|
|
#include <thread>
|
|
#include <chrono>
|
|
|
|
using namespace arangodb;
|
|
using namespace arangodb::communicator;
|
|
|
|
TEST_CASE("requests are properly aborted", "[communicator]" ) {
|
|
Communicator communicator;
|
|
|
|
bool callbacksCalled = false;
|
|
|
|
communicator::Callbacks callbacks([&callbacksCalled](std::unique_ptr<GeneralResponse> response) {
|
|
WARN("RESULT: " << GeneralResponse::responseString(response->responseCode()));
|
|
REQUIRE(false); // it should be aborted?!
|
|
callbacksCalled = true;
|
|
}, [&callbacksCalled](int errorCode, std::unique_ptr<GeneralResponse> response) {
|
|
REQUIRE(!response);
|
|
REQUIRE(errorCode == TRI_COMMUNICATOR_REQUEST_ABORTED);
|
|
callbacksCalled = true;
|
|
});
|
|
auto request = std::unique_ptr<HttpRequest>(HttpRequest::createHttpRequest(rest::ContentType::TEXT, "", 0, {}));
|
|
request->setRequestType(RequestType::GET);
|
|
communicator::Options opt;
|
|
auto destination = Destination("http://www.example.com");
|
|
communicator.addRequest(std::move(destination), std::move(request), callbacks, opt);
|
|
communicator.work_once();
|
|
communicator.abortRequests();
|
|
while (communicator.work_once() > 0) {
|
|
std::this_thread::sleep_for(std::chrono::microseconds(1));
|
|
}
|
|
REQUIRE(callbacksCalled);
|
|
}
|
|
|
|
TEST_CASE("requests will call the progress callback", "[communicator]") {
|
|
Communicator communicator;
|
|
|
|
communicator::Callbacks callbacks([](std::unique_ptr<GeneralResponse> response) {
|
|
}, [](int errorCode, std::unique_ptr<GeneralResponse> response) {
|
|
});
|
|
|
|
auto request = std::unique_ptr<HttpRequest>(HttpRequest::createHttpRequest(rest::ContentType::TEXT, "", 0, {}));
|
|
request->setRequestType(RequestType::GET);
|
|
|
|
// CURL_LAST /* never use! */ HA! FOOL!
|
|
CURLcode curlRc = CURL_LAST;
|
|
communicator::Options opt;
|
|
opt._curlRcFn = std::make_shared<std::function<void(CURLcode)>>([&curlRc](CURLcode rc) {
|
|
curlRc = rc;
|
|
});
|
|
|
|
auto destination = Destination("http://www.example.com");
|
|
communicator.addRequest(std::move(destination), std::move(request), callbacks, opt);
|
|
communicator.work_once();
|
|
communicator.abortRequests();
|
|
while (communicator.work_once() > 0) {
|
|
std::this_thread::sleep_for(std::chrono::microseconds(1));
|
|
}
|
|
REQUIRE(curlRc == CURLE_ABORTED_BY_CALLBACK); // curlRcFn was called
|
|
}
|