librbd: add tests for the C api

Signed-off-by: Josh Durgin <josh.durgin@dreamhost.com>

diff --git a/src/Makefile.am b/src/Makefile.am
index 481ed5134c73f047f72ed033f975a676babb798e..aeb39d63109d76b39ff5f8ba52b738a714c511eb 100644 (file)
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -261,6 +261,12 @@ rbd_LDADD = librados.la -lpthread -lm $(CRYPTOPP_LIBS)
 rbd_CXXFLAGS = ${AM_CXXFLAGS}
 bin_PROGRAMS += rbd
 
+if WITH_DEBUG
+testlibrbd_SOURCES = testlibrbd.c
+testlibrbd_LDADD = librados.a librbd.a libcrush.a -lpthread -lm $(CRYPTOPP_LIBS)
+bin_PROGRAMS += testlibrbd
+endif
+
 if WITH_RADOSGW
 libradosgw_a_SOURCES = \
        rgw/rgw_fs.cc \

diff --git a/src/testlibrbd.c b/src/testlibrbd.c
new file mode 100644 (file)
index 0000000..59b35d9
--- /dev/null
+++ b/src/testlibrbd.c
@@ -0,0 +1,171 @@
+// -*- 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 New Dream Network
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public
+ * License version 2, as published by the Free Software
+ * Foundation.  See file COPYING.
+ *
+ */
+
+#define __STDC_FORMAT_MACROS
+#include "include/librbd.h"
+
+#include <assert.h>
+#include <errno.h>
+#include <inttypes.h>
+#include <stdarg.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <time.h>
+
+#define TEST_IMAGE "testimg"
+#define TEST_POOL "librbdtest"
+#define TEST_SNAP "testsnap"
+#define MB_BYTES(mb) (mb << 20)
+
+void test_create_and_stat(const char *name, size_t size)
+{
+  rbd_image_info_t info;
+  assert(rbd_create_image(TEST_POOL, name, size) == 0);
+  assert(rbd_stat_image(TEST_POOL, name, &info) == 0);
+  printf("image has size %llu and order %d\n", (unsigned long long) info.size, info.order);
+  assert(info.size == size);
+}
+
+void test_resize_and_stat(const char *name, size_t size)
+{
+  rbd_image_info_t info;
+  assert(rbd_resize_image(TEST_POOL, name, size) == 0);
+  assert(rbd_stat_image(TEST_POOL, name, &info) == 0);
+  printf("image has size %llu and order %d\n", (unsigned long long) info.size, info.order);
+  assert(info.size == size);
+}
+
+void test_ls(int num_expected, ...)
+{
+  char **names;
+  int num_images, i, j;
+  char *expected;
+  va_list ap;
+  names = (char **) malloc(sizeof(char **) * 10);
+  num_images = rbd_list_images(TEST_POOL, names, 10);
+  printf("num images is: %d\nexpected: %d\n", num_images, num_expected);
+  assert(num_images == num_expected);
+
+  for (i = 0; i < num_images; i++) {
+    printf("image: %s\n", names[i]);
+  }
+
+  va_start(ap, num_expected);
+  for (i = num_expected; i > 0; i--) {
+    expected = va_arg(ap, char *);
+    printf("expected = %s\n", expected);
+    int found = 0;
+    for (j = 0; j < num_images; j++) {
+      if (names[j] == NULL)
+       continue;
+      if (strcmp(names[j], expected) == 0) {
+       printf("found %s\n", names[j]);
+       free(names[j]);
+       names[j] = NULL;
+       found = 1;
+       break;
+      }
+    }
+    assert(found);
+  }
+
+  for (i = 0; i < num_images; i++) {
+    assert(names[i] == NULL);
+  }
+  free(names);
+}
+
+void test_delete(const char *name)
+{
+  assert(rbd_remove_image(TEST_POOL, name) == 0);
+}
+
+void test_create_snap(const char *image, const char *name)
+{
+  assert(rbd_create_snap(TEST_POOL, image, name) == 0);
+}
+
+void test_ls_snaps(char *image, int num_expected, ...)
+{
+  rbd_snap_info_t *snaps;
+  int num_snaps, i, j, expected_size;
+  char *expected;
+  va_list ap;
+  snaps = (rbd_snap_info_t *) malloc(sizeof(rbd_snap_info_t *) * 10);
+  num_snaps = rbd_list_snaps(TEST_POOL, image, snaps, 10);
+  printf("num snaps is: %d\nexpected: %d\n", num_snaps, num_expected);
+  assert(num_snaps == num_expected);
+
+  for (i = 0; i < num_snaps; i++) {
+    printf("snap: %s\n", snaps[i].name);
+  }
+
+  va_start(ap, num_expected);
+  for (i = num_expected; i > 0; i--) {
+    expected = va_arg(ap, char *);
+    expected_size = va_arg(ap, int);
+    int found = 0;
+    for (j = 0; j < num_snaps; j++) {
+      if (snaps[j].name == NULL)
+       continue;
+      if (strcmp(snaps[j].name, expected) == 0) {
+       printf("found %s with size %llu\n", snaps[j].name, (unsigned long long) snaps[j].size);
+       assert(snaps[j].size == expected_size);
+       free((void *) snaps[j].name);
+       snaps[j].name = NULL;
+       found = 1;
+       break;
+      }
+    }
+    assert(found);
+  }
+
+  for (i = 0; i < num_snaps; i++) {
+    assert(snaps[i].name == NULL);
+  }
+  free(snaps);
+}
+
+void test_delete_snap(const char *image, const char *name)
+{
+  assert(rbd_remove_snap(TEST_POOL, image, name) == 0);
+}
+
+int main(int argc, const char **argv) 
+{
+  rbd_initialize(0, NULL);
+  test_ls(0);
+  test_create_and_stat(TEST_IMAGE, MB_BYTES(1));
+  test_ls(1, TEST_IMAGE);
+  test_ls_snaps(TEST_IMAGE, 0);
+  test_create_snap(TEST_IMAGE, TEST_SNAP);
+  test_ls_snaps(TEST_IMAGE, 1, TEST_SNAP, MB_BYTES(1));
+  test_resize_and_stat(TEST_IMAGE, MB_BYTES(2));
+  test_create_snap(TEST_IMAGE, TEST_SNAP "1");
+  test_ls_snaps(TEST_IMAGE, 2, TEST_SNAP, MB_BYTES(1), TEST_SNAP "1", MB_BYTES(2));
+  test_delete_snap(TEST_IMAGE, TEST_SNAP);
+  test_ls_snaps(TEST_IMAGE, 1, TEST_SNAP "1", MB_BYTES(2));
+  test_delete_snap(TEST_IMAGE, TEST_SNAP "1");
+  test_ls_snaps(TEST_IMAGE, 0);
+  test_create_and_stat(TEST_IMAGE "1", MB_BYTES(2));
+  test_ls(2, TEST_IMAGE, TEST_IMAGE "1");
+  test_delete(TEST_IMAGE);
+  test_ls(1, TEST_IMAGE "1");
+  test_delete(TEST_IMAGE "1");
+  test_ls(0);
+  rbd_shutdown();
+  return 0;
+}
+