fix get model (#4)

bios/bios_info.go

@@ -26,20 +26,22 @@ func GetModel() string {
 		if err != nil {
 			return ""
 		}
-		if string(content) == "ZimaCube" {
+
+		if strings.Contains(strings.ToLower(string(content)), "zimacube") {
 			return ZIMACUBE
 		}
-		if string(content) == "ZimaCubePro" {
+		if strings.Contains(strings.ToLower(string(content)), "zimacubepro") {
 			return ZIMACUBEPRO
 		}
-		return ""
 
+		return ""
 	}
 }
+
 func GetSerialNumber() string {
 	src := "/sys/class/dmi/id/board_version"
 	_, err := os.Stat(src)
-	//ccc
+	// ccc
 	if os.IsNotExist(err) {
 		return ""
 	} else {

bios/bios_info_test.go

@@ -0,0 +1,22 @@
+package bios
+
+import "testing"
+
+func TestGetModel(t *testing.T) {
+	t.Run("ZimaCube", func(t *testing.T) {
+		// ccc
+		model := GetModel()
+		t.Log("Model:", model)
+		if model != ZIMACUBE {
+			t.Errorf("Expected %s, got %s", ZIMACUBE, model)
+		}
+	})
+	t.Run("ZimaCubePro", func(t *testing.T) {
+		// ccc
+		model := GetModel()
+		t.Log("Model:", model)
+		if model != ZIMACUBEPRO {
+			t.Errorf("Expected %s, got %s", ZIMACUBEPRO, model)
+		}
+	})
+}