initial code repo

[stor4nfv.git] / src / ceph / src / test / common / test_blkdev.cc

diff --git a/src/ceph/src/test/common/test_blkdev.cc b/src/ceph/src/test/common/test_blkdev.cc
new file mode 100644 (file)
index 0000000..1dcfd91
--- /dev/null
+++ b/src/ceph/src/test/common/test_blkdev.cc
@@ -0,0 +1,130 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#include <string.h>
+#include <errno.h>
+#include <stdlib.h>
+
+#include "include/types.h"
+#include "common/blkdev.h"
+
+#include "gtest/gtest.h"
+#include <iostream>
+
+  using namespace std;
+
+TEST(blkdev, get_block_device_base) {
+  char buf[PATH_MAX*2];
+  char buf3[PATH_MAX*2];
+
+  ASSERT_EQ(-EINVAL, get_block_device_base("/etc/notindev", buf, 100));
+
+  for (int i=0; i<2; ++i) {
+    string root;
+    if (i == 0) {
+      const char* env = getenv("CEPH_ROOT");
+      ASSERT_NE(env, nullptr) << "Environment Variable CEPH_ROOT not found!";
+      root = string(env) + "/src/test/common/test_blkdev_sys_block";
+    }
+    set_block_device_sandbox_dir(root.c_str());
+
+    // work backwards
+    sprintf(buf, "%s/sys/block", root.c_str());
+    DIR *dir = opendir(buf);
+    ASSERT_NE(dir, nullptr);
+    struct dirent *de = nullptr;
+    while ((de = ::readdir(dir))) {
+      if (de->d_name[0] == '.')
+       continue;
+
+      char base[PATH_MAX];
+      sprintf(base, "/dev/%s", de->d_name);
+      printf("base %s (%s)\n", base, de->d_name);
+      for (char *p = base; *p; ++p)
+       if (*p == '!')
+         *p = '/';
+
+      ASSERT_EQ(-ERANGE, get_block_device_base(base, buf3, 1));
+      ASSERT_EQ(0, get_block_device_base(base, buf3, sizeof(buf3)));
+      printf("  got '%s' expected '%s'\n", buf3, de->d_name);
+      ASSERT_EQ(0, strcmp(de->d_name, buf3));
+      printf("  discard granularity = %lld .. supported = %d\n",
+            (long long)get_block_device_int_property(base, "queue/discard_granularity"),
+            (int)block_device_support_discard(base));
+
+      char subdirfn[PATH_MAX];
+      sprintf(subdirfn, "%s/sys/block/%s", root.c_str(), de->d_name);
+      DIR *subdir = opendir(subdirfn);
+      ASSERT_TRUE(subdir);
+      struct dirent *de2 = nullptr;
+      while ((de2 = ::readdir(subdir))) {
+       if (de2->d_name[0] == '.')
+         continue;
+       // partiions will be prefixed with the base name
+       if (strncmp(de2->d_name, de->d_name, strlen(de->d_name))) {
+         //printf("skipping %s\n", de2->d_name);
+         continue;
+       }
+       char part[PATH_MAX];
+       sprintf(part, "/dev/%s", de2->d_name);
+       for (char *p = part; *p; ++p)
+         if (*p == '!')
+           *p = '/';
+       printf(" part %s (%s %s)\n", part, de->d_name, de2->d_name);
+
+       ASSERT_EQ(0, get_block_device_base(part, buf3, sizeof(buf3)));
+       printf("  got '%s' expected '%s'\n", buf3, de->d_name);
+       ASSERT_EQ(0, strcmp(buf3, de->d_name));
+       printf("  discard granularity = %lld .. supported = %d\n",
+              (long long)get_block_device_int_property(part, "queue/discard_granularity"),
+              (int)block_device_support_discard(part));
+      }
+
+      closedir(subdir);
+    }
+    closedir(dir);
+  }
+}
+
+TEST(blkdev, device_model)
+{
+  const char* env = getenv("CEPH_ROOT");
+  ASSERT_NE(env, nullptr) << "Environment Variable CEPH_ROOT not found!";
+  string root = string(env) + "/src/test/common/test_blkdev_sys_block";
+  set_block_device_sandbox_dir(root.c_str());
+
+  char model[1000] = {0};
+  int rc = block_device_model("sda", model, sizeof(model));
+  ASSERT_EQ(0, rc);
+
+  printf("model '%s'\n", model);
+  ASSERT_EQ(strcmp(model, "myfancymodel"), 0);
+}
+
+TEST(blkdev, get_block_device_string_property)
+{
+  const char* env = getenv("CEPH_ROOT");
+  ASSERT_NE(env, nullptr) << "Environment Variable CEPH_ROOT not found!";
+  string root = string(env) + "/src/test/common/test_blkdev_sys_block";
+  set_block_device_sandbox_dir(root.c_str());
+
+  char val[1000] = {0};
+  int rc = get_block_device_string_property("sda", "device/model",
+                                           val, sizeof(val));
+  ASSERT_EQ(0, rc);
+  printf("val '%s'\n", val);
+  ASSERT_EQ(strcmp(val, "myfancymodel"), 0);
+}
+
+TEST(blkdev, is_rotational)
+{
+  const char* env = getenv("CEPH_ROOT");
+  ASSERT_NE(env, nullptr) << "Environment Variable CEPH_ROOT not found!";
+  string root = string(env) + "/src/test/common/test_blkdev_sys_block";
+  set_block_device_sandbox_dir(root.c_str());
+
+  ASSERT_FALSE(block_device_is_rotational("sda"));
+  ASSERT_TRUE(block_device_is_rotational("sdb"));
+}
+
+