From 68cdf64fe5070c8e25a938d563fbcced23571398 Mon Sep 17 00:00:00 2001
From: Alice Frosi <afrosi@de.ibm.com>
Date: Thu, 25 Jul 2019 15:53:33 +0200
Subject: [PATCH] test: add cpu topology tests

Add cpu driver types in TestQMPCPUDeviceAdd

Signed-off-by: Alice Frosi <afrosi@de.ibm.com>
---
 qemu/qmp_test.go | 31 +++++++++++++++++--------------
 1 file changed, 17 insertions(+), 14 deletions(-)

diff --git a/qemu/qmp_test.go b/qemu/qmp_test.go
index 7598c8cd0..11b33a0e0 100644
--- a/qemu/qmp_test.go
+++ b/qemu/qmp_test.go
@@ -1059,29 +1059,32 @@ func TestQMPPCIVFIOMediatedDeviceAdd(t *testing.T) {
 
 // Checks that CPU are correctly added using device_add
 func TestQMPCPUDeviceAdd(t *testing.T) {
-	connectedCh := make(chan *QMPVersion)
-	disconnectedCh := make(chan struct{})
-	buf := newQMPTestCommandBuffer(t)
-	buf.AddCommand("device_add", nil, "return", nil)
-	cfg := QMPConfig{Logger: qmpTestLogger{}}
-	q := startQMPLoop(buf, cfg, connectedCh, disconnectedCh)
-	checkVersion(t, connectedCh)
-	driver := "qemu64-x86_64-cpu"
+	drivers := []string{"host-x86_64-cpu", "host-s390x-cpu", "host-powerpc64-cpu"}
 	cpuID := "cpu-0"
 	socketID := "0"
 	dieID := "0"
 	coreID := "1"
 	threadID := "0"
-	q.version = &QMPVersion{
+	version := &QMPVersion{
 		Major: 4,
 		Minor: 1,
 	}
-	err := q.ExecuteCPUDeviceAdd(context.Background(), driver, cpuID, socketID, dieID, coreID, threadID, "")
-	if err != nil {
-		t.Fatalf("Unexpected error %v", err)
+	for _, d := range drivers {
+		connectedCh := make(chan *QMPVersion)
+		disconnectedCh := make(chan struct{})
+		buf := newQMPTestCommandBuffer(t)
+		buf.AddCommand("device_add", nil, "return", nil)
+		cfg := QMPConfig{Logger: qmpTestLogger{}}
+		q := startQMPLoop(buf, cfg, connectedCh, disconnectedCh)
+		checkVersion(t, connectedCh)
+		q.version = version
+		err := q.ExecuteCPUDeviceAdd(context.Background(), d, cpuID, socketID, dieID, coreID, threadID, "")
+		if err != nil {
+			t.Fatalf("Unexpected error %v", err)
+		}
+		q.Shutdown()
+		<-disconnectedCh
 	}
-	q.Shutdown()
-	<-disconnectedCh
 }
 
 // Checks that hotpluggable CPUs are listed correctly