github/azure-sdk-for-cpp
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Adding tests to verify that request can be canceled while waiting for socket (#782)

Browse Source 
* adding tests for cancel
This commit is contained in:
Victor Vazquez 2020-10-15 10:13:29 -07:00 committed by GitHub
parent f9e6acb358
commit 854bd834cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 48 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
sdk/core/azure-core/test/ut/transport_adapter.cpp
View File

@ -439,4 +439,52 @@ namespace Azure { namespace Core { namespace Test {
CheckBodyFromBuffer(*response, expectedResponseBodySize);
}
TEST_F(TransportAdapter, cancelTransferUpload)
{
Azure::Core::Http::Url host("http://httpbin.org/put");
Azure::Core::Context cancelThis;
auto threadRoutine = [host, cancelThis]() {
// Start a big upload and expect it to throw cancelation
std::vector<uint8_t> bigBuffer(1024 * 1024 * 200, 'x'); // upload 200 Mb
auto stream = Azure::Core::Http::MemoryBodyStream(bigBuffer);
auto request = Azure::Core::Http::Request(Azure::Core::Http::HttpMethod::Put, host, &stream);
// Request will be canceled from main thread throwing the exception
EXPECT_THROW(pipeline.Send(cancelThis, request), Azure::Core::RequestCanceledException);
};
// Start request
std::thread t1(threadRoutine);
// Wait 100 ms so we know upload has started
std::this_thread::sleep_for(std::chrono::milliseconds(100));
cancelThis.Cancel();
t1.join();
}
TEST_F(TransportAdapter, cancelTransferDownload)
{
// public big blob (321MB)
Azure::Core::Http::Url host("https://bigtestfiles.blob.core.windows.net/cpptestfiles/321MB");
Azure::Core::Context cancelThis;
auto threadRoutine = [host, cancelThis]() {
auto request = Azure::Core::Http::Request(Azure::Core::Http::HttpMethod::Get, host);
// Request will be canceled from main thread throwing the exception
EXPECT_THROW(pipeline.Send(cancelThis, request), Azure::Core::RequestCanceledException);
};
// Start request
std::thread t1(threadRoutine);
// Wait 100 ms so we know download has started
std::this_thread::sleep_for(std::chrono::milliseconds(100));
cancelThis.Cancel();
t1.join();
}
}}} // namespace Azure::Core::Test