From eb64e8752b551678d9ec88706039c94ab5ce2fea Mon Sep 17 00:00:00 2001
From: "Pradipta Kr. Banerjee" <pradipta.banerjee@gmail.com>
Date: Tue, 11 Feb 2014 14:36:21 +0000
Subject: [PATCH] vircapstest: Introduce virCapabilitiesGetCpusForNodemask test

This test creates a Fake NUMA topology with non-sequential cell ids
to check if libvirt properly handles the same

Signed-off-by: Shivaprasad G Bhat <sbhat@linux.vnet.ibm.com>
Signed-off-by: Pradipta Kr. Banerjee <bpradip@in.ibm.com>
---
 .gitignore          |   1 +
 tests/Makefile.am   |   5 ++
 tests/vircapstest.c | 129 ++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 135 insertions(+)
 create mode 100644 tests/vircapstest.c

diff --git a/.gitignore b/.gitignore
index 757dfedd71..5fe60c68d2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -193,6 +193,7 @@
 /tests/virauthconfigtest
 /tests/virbitmaptest
 /tests/virbuftest
+/tests/vircapstest
 /tests/vircgrouptest
 /tests/virdbustest
 /tests/virdrivermoduletest
diff --git a/tests/Makefile.am b/tests/Makefile.am
index eb96f38994..1698455c6a 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -144,6 +144,7 @@ test_programs = virshtest sockettest \
 	virstoragetest \
 	virnetdevbandwidthtest \
 	virkmodtest \
+	vircapstest \
 	$(NULL)
 
 if WITH_REMOTE
@@ -672,6 +673,10 @@ virkmodtest_SOURCES = \
 	virkmodtest.c testutils.h testutils.c
 virkmodtest_LDADD = $(LDADDS)
 
+vircapstest_SOURCES = \
+	vircapstest.c testutils.h testutils.c
+vircapstest_LDADD = $(LDADDS)
+
 if WITH_LIBVIRTD
 libvirtdconftest_SOURCES = \
 	libvirtdconftest.c testutils.h testutils.c \
diff --git a/tests/vircapstest.c b/tests/vircapstest.c
new file mode 100644
index 0000000000..a40771d7cf
--- /dev/null
+++ b/tests/vircapstest.c
@@ -0,0 +1,129 @@
+/*
+ * Copyright (C) IBM Corp 2014
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library.  If not, see
+ * <http://www.gnu.org/licenses/>.
+ *
+ */
+
+#include <config.h>
+#include <stdlib.h>
+
+#include "testutils.h"
+#include "capabilities.h"
+#include "virbitmap.h"
+
+
+#define VIR_FROM_THIS VIR_FROM_NONE
+
+#define MAX_CELLS 4
+#define MAX_CPUS_IN_CELL 2
+#define MAX_MEM_IN_CELL 2097152
+
+
+/*
+ * Build  NUMA Toplogy with cell id starting from (0 + seq)
+ * for testing
+*/
+static virCapsPtr
+buildNUMATopology(int seq)
+{
+    virCapsPtr caps;
+    virCapsHostNUMACellCPUPtr cell_cpus;
+    int core_id, cell_id;
+    int id;
+
+    if ((caps = virCapabilitiesNew(VIR_ARCH_X86_64, 0, 0)) == NULL)
+        goto error;
+
+    id = 0;
+    for (cell_id = 0; cell_id < MAX_CELLS; cell_id++) {
+        if (VIR_ALLOC_N(cell_cpus, MAX_CPUS_IN_CELL) < 0)
+            goto error;
+
+        for (core_id = 0; core_id < MAX_CPUS_IN_CELL; core_id++) {
+            cell_cpus[core_id].id = id + core_id;
+            cell_cpus[core_id].socket_id = cell_id + seq;
+            cell_cpus[core_id].core_id = id + core_id;
+            if (!(cell_cpus[core_id].siblings =
+                  virBitmapNew(MAX_CPUS_IN_CELL)))
+                goto error;
+            ignore_value(virBitmapSetBit(cell_cpus[core_id].siblings, id));
+        }
+        id++;
+
+        if (virCapabilitiesAddHostNUMACell(caps, cell_id + seq,
+                                           MAX_CPUS_IN_CELL,
+                                           MAX_MEM_IN_CELL,
+                                           cell_cpus) < 0)
+           goto error;
+
+        cell_cpus = NULL;
+    }
+
+    return caps;
+
+error:
+    virObjectUnref(caps);
+    return NULL;
+
+}
+
+
+static int
+test_virCapabilitiesGetCpusForNodemask(const void *data ATTRIBUTE_UNUSED)
+{
+    const char *nodestr = "3,4,5,6";
+    virBitmapPtr nodemask = NULL;
+    virBitmapPtr cpumap = NULL;
+    virCapsPtr caps;
+    int mask_size = 8;
+    int ret = -1;
+
+    /*
+     * Build a NUMA topology with cell_id (NUMA node id
+     * being 3(0 + 3),4(1 + 3), 5 and 6
+     */
+    if (!(caps = buildNUMATopology(3)))
+        goto error;
+
+    if (virBitmapParse(nodestr, 0, &nodemask, mask_size) < 0)
+        goto error;
+
+    if (!(cpumap = virCapabilitiesGetCpusForNodemask(caps, nodemask)))
+        goto error;
+
+    ret = 0;
+
+error:
+    virBitmapFree(nodemask);
+    virBitmapFree(cpumap);
+    return ret;
+
+}
+
+
+static int
+mymain(void)
+{
+    int ret = 0;
+
+    if (virtTestRun("virCapabilitiesGetCpusForNodemask",
+                    test_virCapabilitiesGetCpusForNodemask, NULL) < 0)
+        ret = -1;
+
+    return ret;
+}
+
+VIRT_TEST_MAIN(mymain)
-- 
GitLab