summaryrefslogtreecommitdiff
path: root/src/mongo/db/s/resharding/resharding_future_util_test.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/mongo/db/s/resharding/resharding_future_util_test.cpp')
-rw-r--r--src/mongo/db/s/resharding/resharding_future_util_test.cpp100
1 files changed, 0 insertions, 100 deletions
diff --git a/src/mongo/db/s/resharding/resharding_future_util_test.cpp b/src/mongo/db/s/resharding/resharding_future_util_test.cpp
deleted file mode 100644
index e37a13a314b..00000000000
--- a/src/mongo/db/s/resharding/resharding_future_util_test.cpp
+++ /dev/null
@@ -1,100 +0,0 @@
-/**
- * Copyright (C) 2022-present MongoDB, Inc.
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the Server Side Public License, version 1,
- * as published by MongoDB, Inc.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * Server Side Public License for more details.
- *
- * You should have received a copy of the Server Side Public License
- * along with this program. If not, see
- * <http://www.mongodb.com/licensing/server-side-public-license>.
- *
- * As a special exception, the copyright holders give permission to link the
- * code of portions of this program with the OpenSSL library under certain
- * conditions as described in each individual source file and distribute
- * linked combinations including the program with the OpenSSL library. You
- * must comply with the Server Side Public License in all respects for
- * all of the code used other than as permitted herein. If you modify file(s)
- * with this exception, you may extend this exception to your version of the
- * file(s), but you are not obligated to do so. If you do not wish to do so,
- * delete this exception statement from your version. If you delete this
- * exception statement from all source files in the program, then also delete
- * it in the license file.
- */
-
-#include "mongo/db/s/resharding/resharding_future_util.h"
-#include "mongo/unittest/unittest.h"
-#include "mongo/util/concurrency/thread_pool.h"
-
-namespace mongo {
-namespace {
-class ReshardingFutureUtilTest : public unittest::Test {
-protected:
- void setUp() override {
- _executor = std::make_shared<ThreadPool>([]() {
- ThreadPool::Options options;
- options.maxThreads = 2;
- return options;
- }());
- _executor->startup();
- }
-
- void tearDown() override {
- _executor->shutdown();
- _executor->join();
- }
-
- std::shared_ptr<ThreadPool> getExecutor() const {
- return _executor;
- }
-
-private:
- std::shared_ptr<ThreadPool> _executor;
-};
-
-TEST_F(ReshardingFutureUtilTest, CancelWhenAnyErrorThenQuiesceDuringExecutorShutdown) {
- CancellationSource cancelSource;
- auto token = cancelSource.token();
- PromiseAndFuture<void> taskThreadsReady;
- AtomicWord<int> tasksRunningCount{0};
- AtomicWord<bool> taskWasCancelled{false};
- auto checkSignalReady = [&]() {
- auto running = tasksRunningCount.addAndFetch(1);
- if (running == 2) {
- taskThreadsReady.promise.emplaceValue();
- }
- };
- PromiseAndFuture<void> executorShutDownTriggered;
- auto quiesced = ExecutorFuture(getExecutor()).then([&]() {
- return resharding::cancelWhenAnyErrorThenQuiesce(
- {ExecutorFuture(getExecutor())
- .then([&]() {
- checkSignalReady();
- executorShutDownTriggered.future.wait();
- uasserted(6791600, "Executor shut down");
- })
- .share(),
- ExecutorFuture(getExecutor())
- .then([&]() {
- checkSignalReady();
- token.onCancel().wait();
- taskWasCancelled.store(true);
- })
- .share()},
- getExecutor(),
- cancelSource);
- });
- taskThreadsReady.future.wait();
- getExecutor()->shutdown();
- executorShutDownTriggered.promise.emplaceValue();
- auto status = quiesced.getNoThrow();
- ASSERT_EQ(status.code(), 6791600);
- ASSERT_TRUE(taskWasCancelled.load());
-}
-} // namespace
-} // namespace mongo