[libvirt] [PATCH 20/20] qemuagenttest: Add tests for CPU plug functions and helpers

[Peter Krempa]

---
 tests/qemuagenttest.c | 119 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 119 insertions(+)

diff --git a/tests/qemuagenttest.c b/tests/qemuagenttest.c
index 81c24f2..33a6d73 100644
--- a/tests/qemuagenttest.c
+++ b/tests/qemuagenttest.c
@@ -423,6 +423,124 @@ cleanup:
     return ret;
 }

+static const char testQemuAgentCPUResponse[] =
+    "{\"return\": "
+    "   ["
+    "       {\"online\": true,"
+    "        \"can-offline\": false,"
+    "        \"logical-id\": 0"
+    "       },"
+    "       {\"online\": true,"
+    "        \"can-offline\": true,"
+    "        \"logical-id\": 1"
+    "       },"
+    "       {\"online\": true,"
+    "        \"can-offline\": true,"
+    "        \"logical-id\": 2"
+    "        },"
+    "       {\"online\": false,"
+    "        \"can-offline\": true,"
+    "        \"logical-id\": 3"
+    "       }"
+    "   ]"
+    "}";
+
+static const char testQemuAgentCPUArguments1[] =
+    "[{\"logical-id\":0,\"online\":true},"
+     "{\"logical-id\":1,\"online\":false},"
+     "{\"logical-id\":2,\"online\":true},"
+     "{\"logical-id\":3,\"online\":false}]";
+
+static const char testQemuAgentCPUArguments2[] =
+    "[{\"logical-id\":0,\"online\":true},"
+     "{\"logical-id\":1,\"online\":true},"
+     "{\"logical-id\":2,\"online\":true},"
+     "{\"logical-id\":3,\"online\":true}]";
+
+static int
+testQemuAgentCPU(const void *data)
+{
+    virDomainXMLOptionPtr xmlopt = (virDomainXMLOptionPtr)data;
+    qemuMonitorTestPtr test = qemuMonitorTestNewAgent(xmlopt);
+    qemuAgentCPUInfoPtr cpuinfo = NULL;
+    int nvcpus;
+    int ret = -1;
+
+    if (!test)
+        return -1;
+
+    if (qemuMonitorTestAddAgentSyncResponse(test) < 0)
+        goto cleanup;
+
+    if (qemuMonitorTestAddItem(test, "guest-get-vcpus",
+                               testQemuAgentCPUResponse) < 0)
+        goto cleanup;
+
+    /* get cpus */
+    if ((nvcpus = qemuAgentGetVCPUs(qemuMonitorTestGetAgent(test),
+                                    &cpuinfo)) < 0)
+        goto cleanup;
+
+    if (nvcpus != 4) {
+        virReportError(VIR_ERR_INTERNAL_ERROR,
+                       "Expected '4' cpus, got '%d'", nvcpus);
+        goto cleanup;
+    }
+
+    /* try to unplug one */
+    if (qemuAgentUpdateCPUInfo(2, cpuinfo, nvcpus) < 0)
+        goto cleanup;
+
+    if (qemuMonitorTestAddAgentSyncResponse(test) < 0)
+        goto cleanup;
+
+    if (qemuMonitorTestAddItemParams(test, "guest-set-vcpus",
+                                     "{ \"return\" : 4 }",
+                                     "vcpus", testQemuAgentCPUArguments1,
+                                     NULL) < 0)
+        goto cleanup;
+
+    if ((nvcpus = qemuAgentSetVCPUs(qemuMonitorTestGetAgent(test),
+                                    cpuinfo, nvcpus)) < 0)
+        goto cleanup;
+
+    if (nvcpus != 4) {
+        virReportError(VIR_ERR_INTERNAL_ERROR,
+                       "Expected '4' cpus updated , got '%d'", nvcpus);
+        goto cleanup;
+    }
+
+    /* try to hotplug two */
+    if (qemuMonitorTestAddAgentSyncResponse(test) < 0)
+        goto cleanup;
+
+    if (qemuMonitorTestAddItemParams(test, "guest-set-vcpus",
+                                     "{ \"return\" : 4 }",
+                                     "vcpus", testQemuAgentCPUArguments2,
+                                     NULL) < 0)
+        goto cleanup;
+
+    if (qemuAgentUpdateCPUInfo(4, cpuinfo, nvcpus) < 0)
+        goto cleanup;
+
+    if ((nvcpus = qemuAgentSetVCPUs(qemuMonitorTestGetAgent(test),
+                                    cpuinfo, nvcpus)) < 0)
+        goto cleanup;
+
+    if (nvcpus != 4) {
+        virReportError(VIR_ERR_INTERNAL_ERROR,
+                       "Expected '4' cpus updated , got '%d'", nvcpus);
+        goto cleanup;
+    }
+
+    ret = 0;
+
+cleanup:
+    VIR_FREE(cpuinfo);
+    qemuMonitorTestFree(test);
+    return ret;
+}
+

 static int
 mymain(void)
@@ -452,6 +570,7 @@ mymain(void)
     DO_TEST(Shutdown);
     DO_TEST(ArbitraryCommand);
     DO_TEST(Timeout);
+    DO_TEST(CPU);

     virObjectUnref(xmlopt);

-- 
1.8.3.2