From 6667f4e90b476fdf1f6044a5388c1b9f18787e46 Mon Sep 17 00:00:00 2001
From: Hui Zhu <teawater@gmail.com>
Date: Tue, 21 Jan 2020 10:26:59 +0800
Subject: [PATCH] qmp_test: Add TestExecMemdevAdd and TestExecQomSet

Add TestExecMemdevAdd and TestExecQomSet to qmp_test.go.
They can test ExecMemdevAdd and ExecQomSet.

Signed-off-by: Hui Zhu <teawater@antfin.com>
---
 qemu/qmp_test.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/qemu/qmp_test.go b/qemu/qmp_test.go
index 71f97bcad..fc1b2a875 100644
--- a/qemu/qmp_test.go
+++ b/qemu/qmp_test.go
@@ -1294,6 +1294,24 @@ func TestExecSetMigrateArguments(t *testing.T) {
 	<-disconnectedCh
 }
 
+// Checks add memory device
+func TestExecMemdevAdd(t *testing.T) {
+	connectedCh := make(chan *QMPVersion)
+	disconnectedCh := make(chan struct{})
+	buf := newQMPTestCommandBuffer(t)
+	buf.AddCommand("object-add", nil, "return", nil)
+	buf.AddCommand("device_add", nil, "return", nil)
+	cfg := QMPConfig{Logger: qmpTestLogger{}}
+	q := startQMPLoop(buf, cfg, connectedCh, disconnectedCh)
+	checkVersion(t, connectedCh)
+	err := q.ExecMemdevAdd(context.Background(), "memory-backend-ram", "mem0", "", 128, true, "virtio-mem-pci", "virtiomem0")
+	if err != nil {
+		t.Fatalf("Unexpected error: %v", err)
+	}
+	q.Shutdown()
+	<-disconnectedCh
+}
+
 // Checks hotplug memory
 func TestExecHotplugMemory(t *testing.T) {
 	connectedCh := make(chan *QMPVersion)
@@ -1692,3 +1710,20 @@ func TestQMPExecQueryQmpStatus(t *testing.T) {
 	q.Shutdown()
 	<-disconnectedCh
 }
+
+// Checks qom-set
+func TestExecQomSet(t *testing.T) {
+	connectedCh := make(chan *QMPVersion)
+	disconnectedCh := make(chan struct{})
+	buf := newQMPTestCommandBuffer(t)
+	buf.AddCommand("qom-set", nil, "return", nil)
+	cfg := QMPConfig{Logger: qmpTestLogger{}}
+	q := startQMPLoop(buf, cfg, connectedCh, disconnectedCh)
+	checkVersion(t, connectedCh)
+	err := q.ExecQomSet(context.Background(), "virtiomem0", "requested-size", 1024)
+	if err != nil {
+		t.Fatalf("Unexpected error: %v", err)
+	}
+	q.Shutdown()
+	<-disconnectedCh
+}