ceph/src/test/common/test_numa.cc

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3
   4 #include "gtest/gtest.h"
   5 #include "common/numa.h"
   6
   7 TEST(cpu_set, parse_list) {
   8   cpu_set_t cpu_set;
   9   size_t size;
  10
  11   ASSERT_EQ(0, parse_cpu_set_list("0-3", &size, &cpu_set));
  12   ASSERT_EQ(size, 4u);
  13   for (unsigned i = 0; i < size; ++i) {
  14     ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
  15   }
  16
  17   ASSERT_EQ(0, parse_cpu_set_list("0-3,6-7", &size, &cpu_set));
  18   ASSERT_EQ(size, 8u);
  19   for (unsigned i = 0; i < 4; ++i) {
  20     ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
  21   }
  22   for (unsigned i = 4; i < 6; ++i) {
  23     ASSERT_FALSE(CPU_ISSET(i, &cpu_set));
  24   }
  25   for (unsigned i = 6; i < 8; ++i) {
  26     ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
  27   }
  28
  29   ASSERT_EQ(0, parse_cpu_set_list("0-31", &size, &cpu_set));
  30   ASSERT_EQ(size, 32u);
  31   for (unsigned i = 0; i < size; ++i) {
  32     ASSERT_TRUE(CPU_ISSET(i, &cpu_set));
  33   }
  34 }
  35
  36 TEST(cpu_set, to_str_list) {
  37   cpu_set_t cpu_set;
  38   CPU_ZERO(&cpu_set);
  39   CPU_SET(0, &cpu_set);
  40   ASSERT_EQ(std::string("0"), cpu_set_to_str_list(8, &cpu_set));
  41   CPU_SET(1, &cpu_set);
  42   CPU_SET(2, &cpu_set);
  43   CPU_SET(3, &cpu_set);
  44   ASSERT_EQ(std::string("0-3"), cpu_set_to_str_list(8, &cpu_set));
  45   CPU_SET(5, &cpu_set);
  46   ASSERT_EQ(std::string("0-3,5"), cpu_set_to_str_list(8, &cpu_set));
  47   CPU_SET(6, &cpu_set);
  48   CPU_SET(7, &cpu_set);
  49   ASSERT_EQ(std::string("0-3,5-7"), cpu_set_to_str_list(8, &cpu_set));
  50 }
  51
  52 TEST(cpu_set, round_trip_list)
  53 {
  54   for (unsigned i = 0; i < 100; ++i) {
  55     cpu_set_t cpu_set;
  56     size_t size = 32;
  57     CPU_ZERO(&cpu_set);
  58     for (unsigned i = 0; i < 32; ++i) {
  59       if (rand() % 1) {
  60         CPU_SET(i, &cpu_set);
  61       }
  62     }
  63     std::string v = cpu_set_to_str_list(size, &cpu_set);
  64     cpu_set_t cpu_set_2;
  65     size_t size2;
  66     ASSERT_EQ(0, parse_cpu_set_list(v.c_str(), &size2, &cpu_set_2));
  67     for (unsigned i = 0; i < 32; ++i) {
  68       ASSERT_TRUE(CPU_ISSET(i, &cpu_set) == CPU_ISSET(i, &cpu_set_2));
  69     }
  70   }
  71 }
  72