mirror of
https://github.com/XRPLF/clio.git
synced 2025-11-19 19:25:53 +00:00
107
tests/unit/util/RetryTests.cpp
Normal file
107
tests/unit/util/RetryTests.cpp
Normal file
@@ -0,0 +1,107 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
This file is part of clio: https://github.com/XRPLF/clio
|
||||
Copyright (c) 2024, the clio developers.
|
||||
|
||||
Permission to use, copy, modify, and distribute this software for any
|
||||
purpose with or without fee is hereby granted, provided that the above
|
||||
copyright notice and this permission notice appear in all copies.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
//==============================================================================
|
||||
|
||||
#include "util/Fixtures.hpp"
|
||||
#include "util/Retry.hpp"
|
||||
|
||||
#include <boost/asio/strand.hpp>
|
||||
#include <gmock/gmock.h>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <chrono>
|
||||
#include <cstddef>
|
||||
|
||||
using namespace util;
|
||||
|
||||
struct RetryTests : virtual ::testing::Test {
|
||||
std::chrono::milliseconds const delay_{1};
|
||||
std::chrono::milliseconds const maxDelay_{5};
|
||||
};
|
||||
|
||||
TEST_F(RetryTests, ExponentialBackoffStrategy)
|
||||
{
|
||||
ExponentialBackoffStrategy strategy{delay_, maxDelay_};
|
||||
|
||||
EXPECT_EQ(strategy.getDelay(), delay_);
|
||||
|
||||
strategy.increaseDelay();
|
||||
EXPECT_EQ(strategy.getDelay(), delay_ * 2);
|
||||
|
||||
strategy.increaseDelay();
|
||||
EXPECT_LT(strategy.getDelay(), maxDelay_);
|
||||
|
||||
for (size_t i = 0; i < 10; ++i) {
|
||||
strategy.increaseDelay();
|
||||
EXPECT_EQ(strategy.getDelay(), maxDelay_);
|
||||
EXPECT_EQ(strategy.getDelay(), maxDelay_);
|
||||
}
|
||||
|
||||
strategy.reset();
|
||||
EXPECT_EQ(strategy.getDelay(), delay_);
|
||||
}
|
||||
|
||||
struct RetryWithExponentialBackoffStrategyTests : SyncAsioContextTest, RetryTests {
|
||||
RetryWithExponentialBackoffStrategyTests()
|
||||
{
|
||||
EXPECT_EQ(retry_.attemptNumber(), 0);
|
||||
EXPECT_EQ(retry_.delayValue(), delay_);
|
||||
}
|
||||
|
||||
Retry retry_ = makeRetryExponentialBackoff(delay_, maxDelay_, boost::asio::make_strand(ctx));
|
||||
testing::MockFunction<void()> mockCallback_;
|
||||
};
|
||||
|
||||
TEST_F(RetryWithExponentialBackoffStrategyTests, Retry)
|
||||
{
|
||||
retry_.retry(mockCallback_.AsStdFunction());
|
||||
|
||||
EXPECT_EQ(retry_.attemptNumber(), 0);
|
||||
|
||||
EXPECT_CALL(mockCallback_, Call());
|
||||
runContext();
|
||||
|
||||
EXPECT_EQ(retry_.attemptNumber(), 1);
|
||||
EXPECT_EQ(retry_.delayValue(), delay_ * 2);
|
||||
}
|
||||
|
||||
TEST_F(RetryWithExponentialBackoffStrategyTests, Cancel)
|
||||
{
|
||||
retry_.retry(mockCallback_.AsStdFunction());
|
||||
retry_.cancel();
|
||||
runContext();
|
||||
EXPECT_EQ(retry_.attemptNumber(), 0);
|
||||
|
||||
retry_.cancel();
|
||||
EXPECT_EQ(retry_.attemptNumber(), 0);
|
||||
}
|
||||
|
||||
TEST_F(RetryWithExponentialBackoffStrategyTests, Reset)
|
||||
{
|
||||
retry_.retry(mockCallback_.AsStdFunction());
|
||||
|
||||
EXPECT_CALL(mockCallback_, Call());
|
||||
runContext();
|
||||
|
||||
EXPECT_EQ(retry_.attemptNumber(), 1);
|
||||
EXPECT_EQ(retry_.delayValue(), delay_ * 2);
|
||||
|
||||
retry_.reset();
|
||||
EXPECT_EQ(retry_.attemptNumber(), 0);
|
||||
EXPECT_EQ(retry_.delayValue(), delay_);
|
||||
}
|
||||
Reference in New Issue
Block a user