summaryrefslogtreecommitdiffstats
path: root/src/seastar/tests/unit/queue_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/seastar/tests/unit/queue_test.cc')
-rw-r--r--src/seastar/tests/unit/queue_test.cc70
1 files changed, 70 insertions, 0 deletions
diff --git a/src/seastar/tests/unit/queue_test.cc b/src/seastar/tests/unit/queue_test.cc
new file mode 100644
index 00000000..a8405e0a
--- /dev/null
+++ b/src/seastar/tests/unit/queue_test.cc
@@ -0,0 +1,70 @@
+/*
+ * This file is open source software, licensed to you under the terms
+ * of the Apache License, Version 2.0 (the "License"). See the NOTICE file
+ * distributed with this work for additional information regarding copyright
+ * ownership. You may not use this file except in compliance with the License.
+ *
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+/*
+ * Copyright 2018 ScyllaDB
+ */
+
+#include <seastar/testing/test_case.hh>
+#include <seastar/core/queue.hh>
+#include <seastar/core/thread.hh>
+#include <seastar/core/sleep.hh>
+
+using namespace seastar;
+using namespace std::chrono_literals;
+
+SEASTAR_TEST_CASE(test_queue_pop_after_abort) {
+ return async([] {
+ queue<int> q(1);
+ bool exception = false;
+ bool timer = false;
+ future<> done = make_ready_future();
+ q.abort(std::make_exception_ptr(std::runtime_error("boom")));
+ done = sleep(1ms).then([&] {
+ timer = true;
+ q.abort(std::make_exception_ptr(std::runtime_error("boom")));
+ });
+ try {
+ q.pop_eventually().get();
+ } catch(...) {
+ exception = !timer;
+ }
+ BOOST_REQUIRE(exception);
+ done.get();
+ });
+}
+
+SEASTAR_TEST_CASE(test_queue_push_abort) {
+ return async([] {
+ queue<int> q(1);
+ bool exception = false;
+ bool timer = false;
+ future<> done = make_ready_future();
+ q.abort(std::make_exception_ptr(std::runtime_error("boom")));
+ done = sleep(1ms).then([&] {
+ timer = true;
+ q.abort(std::make_exception_ptr(std::runtime_error("boom")));
+ });
+ try {
+ q.push_eventually(1).get();
+ } catch(...) {
+ exception = !timer;
+ }
+ BOOST_REQUIRE(exception);
+ done.get();
+ });
+}