From 176d6f2bfe864bc687f55cf25791b49e65565c06 Mon Sep 17 00:00:00 2001 From: Mark Stemm Date: Mon, 4 May 2020 17:40:42 -0700 Subject: [PATCH] Make token bucket unit test pass valgrind Previously, valgrind was complaining about the leaked token bucket. Signed-off-by: Mark Stemm --- tests/engine/test_token_bucket.cpp | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/engine/test_token_bucket.cpp b/tests/engine/test_token_bucket.cpp index c95e39a2..3351c50c 100644 --- a/tests/engine/test_token_bucket.cpp +++ b/tests/engine/test_token_bucket.cpp @@ -14,6 +14,8 @@ See the License for the specific language governing permissions and limitations under the License. */ +#include + #include "token_bucket.h" #include @@ -21,7 +23,7 @@ using namespace Catch::literals; TEST_CASE("token bucket default ctor", "[token_bucket]") { - auto tb = new token_bucket(); + auto tb = std::make_shared(); REQUIRE(tb->get_tokens() == 1); @@ -38,7 +40,7 @@ TEST_CASE("token bucket default ctor", "[token_bucket]") TEST_CASE("token bucket ctor with custom timer", "[token_bucket]") { auto t = []() -> uint64_t { return 22; }; - auto tb = new token_bucket(t); + auto tb = std::make_shared(t); REQUIRE(tb->get_tokens() == 1); REQUIRE(tb->get_last_seen() == 22); @@ -46,7 +48,7 @@ TEST_CASE("token bucket ctor with custom timer", "[token_bucket]") TEST_CASE("token bucket with 2 tokens/sec rate, max 10 tokens", "[token_bucket]") { - auto tb = new token_bucket(); + auto tb = std::make_shared(); tb->init(2.0, 10, 1); SECTION("claiming 5 tokens")