1 files changed, 72 insertions, 0 deletions
diff --git a/src/test/common/test_numa.cc b/src/test/common/test_numa.cc
new file mode 100644
index 000000000..a3bbdf223
--- /dev/null
+++ b/src/test/common/test_numa.cc
@@ -0,0 +1,72 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#include "gtest/gtest.h"
+#include "common/numa.h"
+
+TEST(cpu_set, parse_list) {
+  cpu_set_t cpu_set;
+  size_t size;
+
+  ASSERT_EQ(0, parse_cpu_set_list("0-3", &size, &cpu_set));
+  ASSERT_EQ(size, 4u);
+  for (unsigned i = 0; i < size; ++i) {
+    ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
+  }
+
+  ASSERT_EQ(0, parse_cpu_set_list("0-3,6-7", &size, &cpu_set));
+  ASSERT_EQ(size, 8u);
+  for (unsigned i = 0; i < 4; ++i) {
+    ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
+  }
+  for (unsigned i = 4; i < 6; ++i) {
+    ASSERT_FALSE(CPU_ISSET(i, &cpu_set));
+  }
+  for (unsigned i = 6; i < 8; ++i) {
+    ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
+  }
+
+  ASSERT_EQ(0, parse_cpu_set_list("0-31", &size, &cpu_set));
+  ASSERT_EQ(size, 32u);
+  for (unsigned i = 0; i < size; ++i) {
+    ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
+  }
+}
+
+TEST(cpu_set, to_str_list) {
+  cpu_set_t cpu_set;
+  CPU_ZERO(&cpu_set);
+  CPU_SET(0, &cpu_set);
+  ASSERT_EQ(std::string("0"), cpu_set_to_str_list(8, &cpu_set));
+  CPU_SET(1, &cpu_set);
+  CPU_SET(2, &cpu_set);
+  CPU_SET(3, &cpu_set);
+  ASSERT_EQ(std::string("0-3"), cpu_set_to_str_list(8, &cpu_set));
+  CPU_SET(5, &cpu_set);
+  ASSERT_EQ(std::string("0-3,5"), cpu_set_to_str_list(8, &cpu_set));
+  CPU_SET(6, &cpu_set);
+  CPU_SET(7, &cpu_set);
+  ASSERT_EQ(std::string("0-3,5-7"), cpu_set_to_str_list(8, &cpu_set));
+}
+
+TEST(cpu_set, round_trip_list)
+{
+  for (unsigned i = 0; i < 100; ++i) {
+    cpu_set_t cpu_set;
+    size_t size = 32;
+    CPU_ZERO(&cpu_set);
+    for (unsigned i = 0; i < 32; ++i) {
+      if (rand() % 1) {
+	CPU_SET(i, &cpu_set);
+      }
+    }
+    std::string v = cpu_set_to_str_list(size, &cpu_set);
+    cpu_set_t cpu_set_2;
+    size_t size2;
+    ASSERT_EQ(0, parse_cpu_set_list(v.c_str(), &size2, &cpu_set_2));
+    for (unsigned i = 0; i < 32; ++i) {
+      ASSERT_TRUE(CPU_ISSET(i, &cpu_set) == CPU_ISSET(i, &cpu_set_2));
+    }
+  }
+}
+