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

Adding retry policy to unit tests (#707)

Browse Source
This commit is contained in:
Victor Vazquez 2020-10-07 14:10:11 -07:00 committed by GitHub
parent 91de5a1bf3
commit 400ca5ae29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,7 +15,11 @@ namespace Azure { namespace Core { namespace Test {
std::vector<std::unique_ptr<Azure::Core::Http::HttpPolicy>> p;
std::shared_ptr<Azure::Core::Http::HttpTransport> transport
= std::make_shared<Azure::Core::Http::CurlTransport>();
Azure::Core::Http::RetryOptions opt;
opt.RetryDelay = std::chrono::milliseconds(10);
// Retry policy will help to prevent server-occasionally-errors
p.push_back(std::make_unique<Azure::Core::Http::RetryPolicy>(opt));
p.push_back(std::make_unique<Azure::Core::Http::TransportPolicy>(std::move(transport)));
return p;
}
@ -143,7 +147,7 @@ namespace Azure { namespace Core { namespace Test {
auto request = Azure::Core::Http::Request(Azure::Core::Http::HttpMethod::Get, host);
// loop sending request
for (auto i = 0; i < 500; i++)
for (auto i = 0; i < 50; i++)
{
auto response = pipeline.Send(context, request);
auto expectedResponseBodySize = std::stoull(response->GetHeaders().at("content-length"));

7
sdk/core/azure-core/test/ut/transport_adapter_file_upload.cpp
View File

@ -26,7 +26,12 @@ namespace Azure { namespace Core { namespace Test {
Azure::Core::Http::HttpStatusCode code,
Azure::Core::Http::HttpStatusCode expectedCode)
{
EXPECT_TRUE(code == expectedCode);
EXPECT_PRED2(
[](Azure::Core::Http::HttpStatusCode a, Azure::Core::Http::HttpStatusCode b) {
return a == b;
},
code,
expectedCode);
}
void TransportAdapter::CheckBodyFromBuffer(