summaryrefslogtreecommitdiffstats
path: root/src/test/gather.cc
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/test/gather.cc103
1 files changed, 103 insertions, 0 deletions
diff --git a/src/test/gather.cc b/src/test/gather.cc
new file mode 100644
index 00000000..49640362
--- /dev/null
+++ b/src/test/gather.cc
@@ -0,0 +1,103 @@
+// -*- 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) 2011 Greg Farnum <gregory.farnum@dreamhost.com>
+ *
+ * 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 "include/Context.h"
+#include "gtest/gtest.h"
+
+class C_Checker : public Context {
+public:
+ bool *finish_called;
+ int *result;
+ C_Checker(bool* _finish_called, int *r) :
+ finish_called(_finish_called), result(r) {}
+ void finish(int r) override { *finish_called = true; *result = r; }
+};
+
+TEST(ContextGather, Constructor) {
+ C_GatherBuilder gather(g_ceph_context);
+ EXPECT_FALSE(gather.has_subs());
+ EXPECT_TRUE(gather.get() == NULL);
+}
+
+TEST(ContextGather, OneSub) {
+ C_GatherBuilder gather(g_ceph_context);
+ Context *sub = gather.new_sub();
+ EXPECT_EQ(1, gather.num_subs_created());
+ EXPECT_EQ(1, gather.num_subs_remaining());
+
+ bool finish_called = false;
+ int result = 0;
+ C_Checker *checker = new C_Checker(&finish_called, &result);
+ gather.set_finisher(checker);
+ gather.activate();
+ sub->complete(0);
+ EXPECT_TRUE(finish_called);
+ EXPECT_EQ(0, result);
+}
+
+TEST(ContextGather, ManySubs) {
+ bool finish_called = false;
+ int result = 0;
+ C_GatherBuilder gather(g_ceph_context, new C_Checker(&finish_called, &result));
+ int sub_count = 8;
+ Context* subs[sub_count];
+ //create subs and test
+ for (int i = 0; i < sub_count; ++i) {
+ subs[i] = gather.new_sub();
+ EXPECT_EQ(i+1, gather.num_subs_created());
+ EXPECT_EQ(i+1, gather.num_subs_remaining());
+ }
+ EXPECT_TRUE(gather.has_subs());
+ gather.activate();
+
+ //finish all except one sub
+ for (int j = 0; j < sub_count - 1; ++j) {
+ subs[j]->complete(0);
+ EXPECT_FALSE(finish_called);
+ }
+
+ //finish last one and check asserts
+ subs[sub_count-1]->complete(0);
+ EXPECT_TRUE(finish_called);
+}
+
+TEST(ContextGather, AlternatingSubCreateFinish) {
+ C_GatherBuilder gather(g_ceph_context);
+ int sub_count = 8;
+ bool finish_called = false;
+ int result = 0;
+ C_Checker *checker = new C_Checker(&finish_called, &result);
+ gather.set_finisher(checker);
+ Context* subs[sub_count];
+
+ //create half the subs
+ for (int i = 0; i < sub_count / 2; ++i) {
+ subs[i] = gather.new_sub();
+ EXPECT_EQ(i + 1, gather.num_subs_created());
+ EXPECT_EQ(i + 1, gather.num_subs_remaining());
+ }
+
+ //alternate finishing first half of subs and creating last half of subs
+ for (int j = 0; j < sub_count / 2; ++j) {
+ subs[j]->complete(0);
+ subs[sub_count / 2 + j] = gather.new_sub();
+ }
+ gather.activate();
+
+ //finish last half of subs
+ for (int k = sub_count / 2; k < sub_count; ++k) {
+ subs[k]->complete(0);
+ }
+
+ EXPECT_TRUE(finish_called);
+}