diff options
Diffstat (limited to 'vbench/src/tests/time_queue/time_queue_test.cpp')
-rw-r--r-- | vbench/src/tests/time_queue/time_queue_test.cpp | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/vbench/src/tests/time_queue/time_queue_test.cpp b/vbench/src/tests/time_queue/time_queue_test.cpp new file mode 100644 index 00000000000..d6be9501e6d --- /dev/null +++ b/vbench/src/tests/time_queue/time_queue_test.cpp @@ -0,0 +1,65 @@ +// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +#include <vespa/vespalib/testkit/testapp.h> +#include <vbench/test/all.h> +#include <vespa/vespalib/util/sync.h> + +using namespace vbench; + +TEST_MT_FFF("time queue", 2, TimeQueue<int>(10.0, 5.0), vespalib::Gate(), vespalib::Gate()) { + if (thread_id == 0) { + f1.insert(std::unique_ptr<int>(new int(1)), 1.0); + f1.insert(std::unique_ptr<int>(new int(2)), 3.0); + f1.insert(std::unique_ptr<int>(new int(3)), 2.0); + f2.countDown(); + f1.insert(std::unique_ptr<int>(new int(4)), 100.0); + f1.insert(std::unique_ptr<int>(new int(5)), 101.0); + f3.countDown(); + } else { + double delay; + std::vector<std::unique_ptr<int> > list; + EXPECT_TRUE(f2.await(20000)); + EXPECT_FALSE(f3.await(20)); + { + f1.extract(1.5, list, delay); + ASSERT_EQUAL(1u, list.size()); + EXPECT_EQUAL(1, *list[0]); + EXPECT_EQUAL(0.5, delay); + list.clear(); + } + { + f1.extract(10.0, list, delay); + ASSERT_EQUAL(2u, list.size()); + EXPECT_EQUAL(3, *list[0]); + EXPECT_EQUAL(2, *list[1]); + EXPECT_EQUAL(5.0, delay); + list.clear(); + } + { + f1.extract(99.25, list, delay); + EXPECT_EQUAL(0u, list.size()); + EXPECT_EQUAL(5.0, delay); + } + EXPECT_TRUE(f3.await(20000)); + { + f1.extract(99.25, list, delay); + EXPECT_EQUAL(0u, list.size()); + EXPECT_EQUAL(0.75, delay); + } + f1.discard(); + { + f1.extract(101.5, list, delay); + EXPECT_EQUAL(0u, list.size()); + EXPECT_EQUAL(5.0, delay); + } + f1.close(); + f1.insert(std::unique_ptr<int>(new int(6)), 102.0); + f1.insert(std::unique_ptr<int>(new int(7)), 103.0); + { + f1.extract(103.5, list, delay); + EXPECT_EQUAL(0u, list.size()); + EXPECT_EQUAL(5.0, delay); + } + } +} + +TEST_MAIN() { TEST_RUN_ALL(); } |