diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 18:45:59 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 18:45:59 +0000 |
commit | 19fcec84d8d7d21e796c7624e521b60d28ee21ed (patch) | |
tree | 42d26aa27d1e3f7c0b8bd3fd14e7d7082f5008dc /src/test/rbd_mirror/test_mock_Throttler.cc | |
parent | Initial commit. (diff) | |
download | ceph-19fcec84d8d7d21e796c7624e521b60d28ee21ed.tar.xz ceph-19fcec84d8d7d21e796c7624e521b60d28ee21ed.zip |
Adding upstream version 16.2.11+ds.upstream/16.2.11+dsupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/test/rbd_mirror/test_mock_Throttler.cc')
-rw-r--r-- | src/test/rbd_mirror/test_mock_Throttler.cc | 253 |
1 files changed, 253 insertions, 0 deletions
diff --git a/src/test/rbd_mirror/test_mock_Throttler.cc b/src/test/rbd_mirror/test_mock_Throttler.cc new file mode 100644 index 000000000..ab562c18d --- /dev/null +++ b/src/test/rbd_mirror/test_mock_Throttler.cc @@ -0,0 +1,253 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2016 SUSE LINUX GmbH + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ + +#include "test/rbd_mirror/test_mock_fixture.h" +#include "test/librbd/mock/MockImageCtx.h" + +namespace librbd { + +namespace { + +struct MockTestImageCtx : public librbd::MockImageCtx { + MockTestImageCtx(librbd::ImageCtx &image_ctx) + : librbd::MockImageCtx(image_ctx) { + } +}; + +} // anonymous namespace + +} // namespace librbd + +// template definitions +#include "tools/rbd_mirror/Throttler.cc" + +namespace rbd { +namespace mirror { + +class TestMockThrottler : public TestMockFixture { +public: + typedef Throttler<librbd::MockTestImageCtx> MockThrottler; + +}; + +TEST_F(TestMockThrottler, Single_Sync) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + C_SaferCond on_start; + throttler.start_op("ns", "id", &on_start); + ASSERT_EQ(0, on_start.wait()); + throttler.finish_op("ns", "id"); +} + +TEST_F(TestMockThrottler, Multiple_Syncs) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(2); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + C_SaferCond on_start3; + throttler.start_op("ns", "id3", &on_start3); + C_SaferCond on_start4; + throttler.start_op("ns", "id4", &on_start4); + + ASSERT_EQ(0, on_start2.wait()); + throttler.finish_op("ns", "id2"); + ASSERT_EQ(0, on_start3.wait()); + throttler.finish_op("ns", "id3"); + ASSERT_EQ(0, on_start1.wait()); + throttler.finish_op("ns", "id1"); + ASSERT_EQ(0, on_start4.wait()); + throttler.finish_op("ns", "id4"); +} + +TEST_F(TestMockThrottler, Cancel_Running_Sync) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + C_SaferCond on_start; + throttler.start_op("ns", "id", &on_start); + ASSERT_EQ(0, on_start.wait()); + ASSERT_FALSE(throttler.cancel_op("ns", "id")); + throttler.finish_op("ns", "id"); +} + +TEST_F(TestMockThrottler, Cancel_Waiting_Sync) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(1); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + + ASSERT_EQ(0, on_start1.wait()); + ASSERT_TRUE(throttler.cancel_op("ns", "id2")); + ASSERT_EQ(-ECANCELED, on_start2.wait()); + throttler.finish_op("ns", "id1"); +} + +TEST_F(TestMockThrottler, Cancel_Running_Sync_Start_Waiting) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(1); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + + ASSERT_EQ(0, on_start1.wait()); + ASSERT_FALSE(throttler.cancel_op("ns", "id1")); + throttler.finish_op("ns", "id1"); + ASSERT_EQ(0, on_start2.wait()); + throttler.finish_op("ns", "id2"); +} + +TEST_F(TestMockThrottler, Duplicate) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(1); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + ASSERT_EQ(0, on_start1.wait()); + + C_SaferCond on_start2; + throttler.start_op("ns", "id1", &on_start2); + ASSERT_EQ(0, on_start2.wait()); + + C_SaferCond on_start3; + throttler.start_op("ns", "id2", &on_start3); + C_SaferCond on_start4; + throttler.start_op("ns", "id2", &on_start4); + ASSERT_EQ(-ENOENT, on_start3.wait()); + + throttler.finish_op("ns", "id1"); + ASSERT_EQ(0, on_start4.wait()); + throttler.finish_op("ns", "id2"); +} + +TEST_F(TestMockThrottler, Duplicate2) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(2); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + ASSERT_EQ(0, on_start1.wait()); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + ASSERT_EQ(0, on_start2.wait()); + + C_SaferCond on_start3; + throttler.start_op("ns", "id3", &on_start3); + C_SaferCond on_start4; + throttler.start_op("ns", "id3", &on_start4); // dup + ASSERT_EQ(-ENOENT, on_start3.wait()); + + C_SaferCond on_start5; + throttler.start_op("ns", "id4", &on_start5); + + throttler.finish_op("ns", "id1"); + ASSERT_EQ(0, on_start4.wait()); + + throttler.finish_op("ns", "id2"); + ASSERT_EQ(0, on_start5.wait()); + + C_SaferCond on_start6; + throttler.start_op("ns", "id5", &on_start6); + + throttler.finish_op("ns", "id3"); + ASSERT_EQ(0, on_start6.wait()); + + throttler.finish_op("ns", "id4"); + throttler.finish_op("ns", "id5"); +} + +TEST_F(TestMockThrottler, Increase_Max_Concurrent_Syncs) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(2); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + C_SaferCond on_start3; + throttler.start_op("ns", "id3", &on_start3); + C_SaferCond on_start4; + throttler.start_op("ns", "id4", &on_start4); + C_SaferCond on_start5; + throttler.start_op("ns", "id5", &on_start5); + + ASSERT_EQ(0, on_start1.wait()); + ASSERT_EQ(0, on_start2.wait()); + + throttler.set_max_concurrent_ops(4); + + ASSERT_EQ(0, on_start3.wait()); + ASSERT_EQ(0, on_start4.wait()); + + throttler.finish_op("ns", "id4"); + ASSERT_EQ(0, on_start5.wait()); + + throttler.finish_op("ns", "id1"); + throttler.finish_op("ns", "id2"); + throttler.finish_op("ns", "id3"); + throttler.finish_op("ns", "id5"); +} + +TEST_F(TestMockThrottler, Decrease_Max_Concurrent_Syncs) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(4); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + C_SaferCond on_start3; + throttler.start_op("ns", "id3", &on_start3); + C_SaferCond on_start4; + throttler.start_op("ns", "id4", &on_start4); + C_SaferCond on_start5; + throttler.start_op("ns", "id5", &on_start5); + + ASSERT_EQ(0, on_start1.wait()); + ASSERT_EQ(0, on_start2.wait()); + ASSERT_EQ(0, on_start3.wait()); + ASSERT_EQ(0, on_start4.wait()); + + throttler.set_max_concurrent_ops(2); + + throttler.finish_op("ns", "id1"); + throttler.finish_op("ns", "id2"); + throttler.finish_op("ns", "id3"); + + ASSERT_EQ(0, on_start5.wait()); + + throttler.finish_op("ns", "id4"); + throttler.finish_op("ns", "id5"); +} + +TEST_F(TestMockThrottler, Drain) { + MockThrottler throttler(g_ceph_context, "rbd_mirror_concurrent_image_syncs"); + throttler.set_max_concurrent_ops(1); + + C_SaferCond on_start1; + throttler.start_op("ns", "id1", &on_start1); + C_SaferCond on_start2; + throttler.start_op("ns", "id2", &on_start2); + + ASSERT_EQ(0, on_start1.wait()); + throttler.drain("ns", -ESTALE); + ASSERT_EQ(-ESTALE, on_start2.wait()); +} + +} // namespace mirror +} // namespace rbd |