pkg/host: refactor machine info tests

Refactor tests so that they can be used with canned info.
The test fails for lots of different archs and we don't have any tests,
so these will keep breaking.
This change prepared for tests with canned info.

1 files changed, 21 insertions, 34 deletions

diff --git a/pkg/host/machine_info_linux_test.go b/pkg/host/machine_info_linux_test.go
index d318c08b3..682d8b397 100644
--- a/pkg/host/machine_info_linux_test.go
+++ b/pkg/host/machine_info_linux_test.go
@@ -11,42 +11,25 @@ import (
 	"testing"
 )
 
-func TestMachineInfoLinux(t *testing.T) {
-	result, err := CollectMachineInfo()
-	if err != nil {
+func TestReadCPUInfoLinux(t *testing.T) {
+	buf := new(bytes.Buffer)
+	if err := readCPUInfo(buf); err != nil {
 		t.Fatal(err)
 	}
-
-	scanner := bufio.NewScanner(bytes.NewReader(result))
-
-	for scanner.Scan() {
-		line := scanner.Text()
-
-		if line == "[CPU Info]" {
-			checkCPUInfo(t, scanner)
-		}
-		if line == "[KVM]" {
-			checkKVMInfo(t, scanner)
-		}
-	}
+	checkCPUInfo(t, buf.Bytes(), runtime.GOARCH)
 }
 
-func checkCPUInfo(t *testing.T, scanner *bufio.Scanner) {
+func checkCPUInfo(t *testing.T, data []byte, arch string) {
+	t.Logf("input data:\n%s", data)
 	keys := make(map[string]bool)
-	for scanner.Scan() {
-		line := scanner.Text()
-		// End of CPU Info section.
-		if strings.HasPrefix(line, "-----") {
-			break
-		}
-		splitted := strings.Split(line, ":")
+	for s := bufio.NewScanner(bytes.NewReader(data)); s.Scan(); {
+		splitted := strings.Split(s.Text(), ":")
 		if len(splitted) != 2 {
-			t.Fatalf("the format of line \"%s\" is not correct", line)
+			t.Fatalf("the format of line %q is not correct", s.Text())
 		}
 		key := strings.TrimSpace(splitted[0])
 		keys[key] = true
 	}
-
 	importantKeys := map[string][]string{
 		"ppc64le":  {"cpu", "revision", "platform", "model", "machine"},
 		"amd64":    {"vendor_id", "model", "flags"},
@@ -57,23 +40,27 @@ func checkCPUInfo(t *testing.T, scanner *bufio.Scanner) {
 		"mips64le": {"system type", "cpu model", "ASEs implemented"},
 		"riscv64":  {"processor", "isa", "mmu"},
 	}
-	archKeys := importantKeys[runtime.GOARCH]
+	archKeys := importantKeys[arch]
+	if len(archKeys) == 0 {
+		t.Fatalf("unknown arch %v", arch)
+	}
 	for _, name := range archKeys {
 		if !keys[name] {
-			t.Fatalf("key '%s' not found", name)
+			t.Fatalf("key %q not found", name)
 		}
 	}
 }
 
-func checkKVMInfo(t *testing.T, scanner *bufio.Scanner) {
-	for scanner.Scan() {
-		line := scanner.Text()
+func TestReadKVMInfoLinux(t *testing.T) {
+	buf := new(bytes.Buffer)
+	if err := readKVMInfo(buf); err != nil {
+		t.Fatal(err)
+	}
+	for s := bufio.NewScanner(buf); s.Scan(); {
+		line := s.Text()
 		if line == "" {
 			continue
 		}
-		if strings.HasPrefix(line, "-----") {
-			break
-		}
 		splitted := strings.Split(line, ":")
 		if len(splitted) != 2 {
 			t.Fatalf("the format of line \"%s\" is not correct", line)