119 lines
2.8 KiB
Go
119 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
// Mock labels for testing
|
|
var testLabels = map[string]map[string]string{
|
|
"redhat-bootc": {
|
|
"com.redhat.bootc": "true",
|
|
"ostree.bootable": "true",
|
|
},
|
|
"debian-bootc": {
|
|
"com.debian.bootc": "true",
|
|
"ostree.bootable": "true",
|
|
},
|
|
"both-labels": {
|
|
"com.redhat.bootc": "true",
|
|
"com.debian.bootc": "true",
|
|
"ostree.bootable": "true",
|
|
},
|
|
"no-bootc": {
|
|
"some.other.label": "value",
|
|
},
|
|
"no-ostree": {
|
|
"com.debian.bootc": "true",
|
|
},
|
|
}
|
|
|
|
func isBootcImage(labels map[string]string) bool {
|
|
// Check for Red Hat bootc label
|
|
if val, exists := labels["com.redhat.bootc"]; exists && val == "true" {
|
|
return true
|
|
}
|
|
|
|
// Check for Debian bootc label
|
|
if val, exists := labels["com.debian.bootc"]; exists && val == "true" {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func validateBootcImage(labels map[string]string, imageRef string) error {
|
|
if !isBootcImage(labels) {
|
|
return fmt.Errorf("image %s is not a bootc image (missing com.redhat.bootc=true or com.debian.bootc=true label)", imageRef)
|
|
}
|
|
|
|
// Check for required OSTree labels
|
|
if val, exists := labels["ostree.bootable"]; !exists || val != "true" {
|
|
return fmt.Errorf("image %s is not a bootc image (missing ostree.bootable=true label)", imageRef)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getBootcType(labels map[string]string) string {
|
|
if val, exists := labels["com.redhat.bootc"]; exists && val == "true" {
|
|
return "redhat"
|
|
}
|
|
|
|
if val, exists := labels["com.debian.bootc"]; exists && val == "true" {
|
|
return "debian"
|
|
}
|
|
|
|
return "unknown"
|
|
}
|
|
|
|
func main() {
|
|
fmt.Println("Testing Debian bootc validation logic...")
|
|
fmt.Println("")
|
|
|
|
tests := []struct {
|
|
name string
|
|
labels map[string]string
|
|
expect bool
|
|
}{
|
|
{"Red Hat bootc", testLabels["redhat-bootc"], true},
|
|
{"Debian bootc", testLabels["debian-bootc"], true},
|
|
{"Both labels", testLabels["both-labels"], true},
|
|
{"No bootc", testLabels["no-bootc"], false},
|
|
{"No ostree", testLabels["no-ostree"], true}, // Should be true because it has bootc label
|
|
}
|
|
|
|
passed := 0
|
|
total := len(tests)
|
|
|
|
for _, test := range tests {
|
|
fmt.Printf("Test: %s\n", test.name)
|
|
fmt.Printf("Labels: %v\n", test.labels)
|
|
|
|
isBootc := isBootcImage(test.labels)
|
|
bootcType := getBootcType(test.labels)
|
|
err := validateBootcImage(test.labels, "test-image")
|
|
|
|
fmt.Printf("Is bootc: %t (expected: %t)\n", isBootc, test.expect)
|
|
fmt.Printf("Bootc type: %s\n", bootcType)
|
|
fmt.Printf("Validation error: %v\n", err)
|
|
|
|
if isBootc == test.expect {
|
|
fmt.Printf("✅ PASS\n")
|
|
passed++
|
|
} else {
|
|
fmt.Printf("❌ FAIL\n")
|
|
}
|
|
fmt.Println("")
|
|
}
|
|
|
|
fmt.Printf("Test Results: %d/%d passed\n", passed, total)
|
|
|
|
if passed == total {
|
|
fmt.Println("🎉 All validation logic tests passed!")
|
|
os.Exit(0)
|
|
} else {
|
|
fmt.Println("❌ Some validation logic tests failed")
|
|
os.Exit(1)
|
|
}
|
|
}
|