tests: Introduce virnetdevbandwidthtest

The only API tested so far would be virNetDevBandwidthMinimal.
But there's more to come!

Signed-off-by: Michal Privoznik <mprivozn@redhat.com>

1 files changed, 155 insertions, 0 deletions

diff --git a/tests/virnetdevbandwidthtest.c b/tests/virnetdevbandwidthtest.c
new file mode 100644
index 0000000000..989018e6f1
--- /dev/null
+++ b/tests/virnetdevbandwidthtest.c
@@ -0,0 +1,155 @@
+/*
+ * Copyright (C) 2014 Red Hat, Inc.
+ *
+ * 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/>.
+ *
+ * Author: Michal Privoznik <mprivozn@redhat.com>
+ */
+
+#include <config.h>
+
+#include "testutils.h"
+#include "virnetdevbandwidth.h"
+#include "netdev_bandwidth_conf.c"
+
+#define VIR_FROM_THIS VIR_FROM_NONE
+
+struct testMinimalStruct {
+    const char *expected_result;
+    const char *band1;
+    const char *band2;
+};
+
+#define PARSE(xml, var)                                                 \
+    do {                                                                \
+        xmlDocPtr doc;                                                  \
+        xmlXPathContextPtr ctxt = NULL;                                 \
+                                                                        \
+        if (!xml)                                                       \
+            break;                                                      \
+                                                                        \
+        if (!(doc = virXMLParseStringCtxt((xml),                        \
+                                          "bandwidth definition",       \
+                                          &ctxt)))                      \
+            goto cleanup;                                               \
+                                                                        \
+        (var) = virNetDevBandwidthParse(ctxt->node,                     \
+                                        VIR_DOMAIN_NET_TYPE_NETWORK);   \
+        xmlFreeDoc(doc);                                                \
+        xmlXPathFreeContext(ctxt);                                      \
+        if (!(var))                                                     \
+            goto cleanup;                                               \
+    } while (0)
+
+static int
+testVirNetDevBandwidthMinimal(const void *data)
+{
+    int ret = -1;
+    const struct testMinimalStruct *info = data;
+    virNetDevBandwidthPtr expected_result = NULL, result = NULL,
+                          band1 = NULL, band2 = NULL;
+
+
+    /* Parse given XMLs */
+    PARSE(info->expected_result, expected_result);
+    PARSE(info->band1, band1);
+    PARSE(info->band2, band2);
+
+    if (virNetDevBandwidthMinimal(&result, band1, band2) < 0)
+        goto cleanup;
+
+    if (!virNetDevBandwidthEqual(expected_result, result)) {
+        virBuffer exp_buf = VIR_BUFFER_INITIALIZER,
+                  res_buf = VIR_BUFFER_INITIALIZER;
+        char *exp = NULL, *res = NULL;
+
+        fprintf(stderr, "expected_result != result");
+
+        if (virNetDevBandwidthFormat(expected_result, &exp_buf) < 0 ||
+            virNetDevBandwidthFormat(result, &res_buf) < 0 ||
+            !(exp = virBufferContentAndReset(&exp_buf)) ||
+            !(res = virBufferContentAndReset(&res_buf))) {
+            fprintf(stderr, "Failed to fail");
+            virBufferFreeAndReset(&exp_buf);
+            virBufferFreeAndReset(&res_buf);
+            VIR_FREE(exp);
+            VIR_FREE(res);
+            goto cleanup;
+        }
+
+        virtTestDifference(stderr, exp, res);
+        VIR_FREE(exp);
+        VIR_FREE(res);
+    }
+
+    ret = 0;
+cleanup:
+    virNetDevBandwidthFree(expected_result);
+    virNetDevBandwidthFree(result);
+    virNetDevBandwidthFree(band1);
+    virNetDevBandwidthFree(band2);
+    return ret;
+}
+
+static int
+mymain(void)
+{
+    int ret = 0;
+
+#define DO_TEST_MINIMAL(r, ...)                             \
+    do {                                                    \
+        struct testMinimalStruct data = {r, __VA_ARGS__};   \
+        if (virtTestRun("virNetDevBandwidthMinimal",        \
+                        testVirNetDevBandwidthMinimal,      \
+                        &data) < 0)                         \
+            ret = -1;                                       \
+    } while (0)
+
+
+    DO_TEST_MINIMAL(NULL, NULL, NULL);
+
+    DO_TEST_MINIMAL("<bandwidth>"
+                    "  <inbound average='1000' peak='5000' burst='5120'/>"
+                    "  <outbound average='128' peak='256' burst='256'/>"
+                    "</bandwidth>",
+                    .band1 = "<bandwidth>"
+                    "  <inbound average='1000' peak='5000' burst='5120'/>"
+                    "  <outbound average='128' peak='256' burst='256'/>"
+                    "</bandwidth>");
+
+    DO_TEST_MINIMAL("<bandwidth>"
+                    "  <inbound average='1000' peak='5000' burst='5120'/>"
+                    "  <outbound average='128' peak='256' burst='256'/>"
+                    "</bandwidth>",
+                    .band2 = "<bandwidth>"
+                    "  <inbound average='1000' peak='5000' burst='5120'/>"
+                    "  <outbound average='128' peak='256' burst='256'/>"
+                    "</bandwidth>");
+    DO_TEST_MINIMAL("<bandwidth>"
+                    "  <inbound average='1' peak='2' floor='3' burst='4'/>"
+                    "  <outbound average='5' peak='6' burst='7'/>"
+                    "</bandwidth>",
+                    "<bandwidth>"
+                    "  <inbound average='1' peak='2' burst='4'/>"
+                    "  <outbound average='0' burst='7'/>"
+                    "</bandwidth>",
+                    "<bandwidth>"
+                    "  <inbound average='1' peak='2' floor='3'/>"
+                    "  <outbound average='5' peak='6'/>"
+                    "</bandwidth>");
+    return ret;
+}
+
+VIRT_TEST_MAIN(mymain);