deb-bootc-image-builder/bib/internal/debos/ostree_test.go

package debos

import (
	"os"
	"strings"
	"testing"

	"github.com/osbuild/images/pkg/arch"
)

func TestCreateOSTreeTemplate(t *testing.T) {
	ostreeConfig := OSTreeConfig{
		Repository: "/ostree/repo",
		Branch:     "debian/trixie/amd64",
		Subject:    "Test OSTree commit",
		Body:       "Test body",
		Mode:       "bare-user",
	}

	template := CreateOSTreeTemplate("amd64", "trixie", "debian:trixie", ostreeConfig)

	if template == nil {
		t.Fatal("Template should not be nil")
	}

	if template.Architecture != "amd64" {
		t.Errorf("Expected architecture amd64, got %s", template.Architecture)
	}

	if template.Suite != "trixie" {
		t.Errorf("Expected suite trixie, got %s", template.Suite)
	}

	if template.OSTree.Repository != "/ostree/repo" {
		t.Errorf("Expected OSTree repository /ostree/repo, got %s", template.OSTree.Repository)
	}

	if template.OSTree.Branch != "debian/trixie/amd64" {
		t.Errorf("Expected OSTree branch debian/trixie/amd64, got %s", template.OSTree.Branch)
	}

	// Check that OSTree-specific actions were added
	foundOstreeCommit := false
	for _, action := range template.Actions {
		if action.Action == "ostree-commit" {
			foundOstreeCommit = true
			break
		}
	}

	if !foundOstreeCommit {
		t.Error("Expected to find ostree-commit action")
	}
}

func TestCreateBootcOSTreeTemplate(t *testing.T) {
	template := CreateBootcOSTreeTemplate("amd64", "trixie", "debian:trixie")

	if template == nil {
		t.Fatal("Template should not be nil")
	}

	if template.Architecture != "amd64" {
		t.Errorf("Expected architecture amd64, got %s", template.Architecture)
	}

	if template.Suite != "trixie" {
		t.Errorf("Expected suite trixie, got %s", template.Suite)
	}

	// Check default OSTree configuration
	expectedBranch := "debian/trixie/amd64"
	if template.OSTree.Branch != expectedBranch {
		t.Errorf("Expected OSTree branch %s, got %s", expectedBranch, template.OSTree.Branch)
	}

	if template.OSTree.Mode != "bare-user" {
		t.Errorf("Expected OSTree mode bare-user, got %s", template.OSTree.Mode)
	}
}

func TestGetArchSuffix(t *testing.T) {
	testCases := []struct {
		arch     string
		expected string
	}{
		{"amd64", "amd64"},
		{"arm64", "arm64"},
		{"armhf", "armhf"},
		{"i386", "i386"},
		{"unknown", "amd64"}, // default case
	}

	for _, tc := range testCases {
		result := getArchSuffix(tc.arch)
		if result != tc.expected {
			t.Errorf("For arch %s, expected suffix %s, got %s", tc.arch, tc.expected, result)
		}
	}
}

func TestGenerateOSTreeConfigScript(t *testing.T) {
	config := OSTreeConfig{
		Repository: "/ostree/repo",
		Mode:       "bare-user",
	}

	script := generateOSTreeConfigScript(config)

	// Check that the script contains expected elements
	expectedElements := []string{
		"ostree init",
		"bare-user",
		"/ostree/repo",
		"dracut",
		"systemctl enable",
	}

	for _, element := range expectedElements {
		if !strings.Contains(script, element) {
			t.Errorf("Expected script to contain %s", element)
		}
	}
}

func TestGenerateBootloaderConfigScript(t *testing.T) {
	script := generateBootloaderConfigScript("amd64", "trixie")

	// Check that the script contains expected elements
	expectedElements := []string{
		"GRUB_TIMEOUT=5",
		"ostree=/ostree/boot.1/debian/trixie/amd64",
		"update-grub",
	}

	for _, element := range expectedElements {
		if !strings.Contains(script, element) {
			t.Errorf("Expected script to contain %s", element)
		}
	}
}

func TestNewOSTreeBuilder(t *testing.T) {
	// Create temporary directories
	workDir, err := os.MkdirTemp("", "ostree-builder-work")
	if err != nil {
		t.Fatalf("Failed to create temp work directory: %v", err)
	}
	defer os.RemoveAll(workDir)

	outputDir, err := os.MkdirTemp("", "ostree-builder-output")
	if err != nil {
		t.Fatalf("Failed to create temp output directory: %v", err)
	}
	defer os.RemoveAll(outputDir)

	// Test creating OSTree builder
	builder, err := NewOSTreeBuilder(workDir, outputDir)
	if err != nil {
		t.Fatalf("Failed to create OSTree builder: %v", err)
	}

	if builder == nil {
		t.Fatal("Builder should not be nil")
	}

	if builder.workDir != workDir {
		t.Errorf("Expected workDir %s, got %s", workDir, builder.workDir)
	}

	if builder.outputDir != outputDir {
		t.Errorf("Expected outputDir %s, got %s", outputDir, builder.outputDir)
	}
}

func TestBuildBootcOSTree(t *testing.T) {
	// Create temporary directories
	workDir, err := os.MkdirTemp("", "ostree-builder-work")
	if err != nil {
		t.Fatalf("Failed to create temp work directory: %v", err)
	}
	defer os.RemoveAll(workDir)

	outputDir, err := os.MkdirTemp("", "ostree-builder-output")
	if err != nil {
		t.Fatalf("Failed to create temp output directory: %v", err)
	}
	defer os.RemoveAll(outputDir)

	builder, err := NewOSTreeBuilder(workDir, outputDir)
	if err != nil {
		t.Fatalf("Failed to create OSTree builder: %v", err)
	}

	// Get current architecture
	currentArch := arch.Current()

	// Create build options
	options := &BuildOptions{
		Architecture:   currentArch,
		Suite:          "trixie",
		ContainerImage: "debian:trixie",
		ImageTypes:     []string{"qcow2"},
		OutputDir:      outputDir,
		WorkDir:        workDir,
	}

	// Test building bootc OSTree image
	// Note: This will likely fail in the test environment, but we can test the setup
	result, err := builder.BuildBootcOSTree(options)
	
	// We expect this to fail in the test environment, but we can check the setup
	if result != nil {
		t.Logf("Build result: Success=%t, OutputPath=%s", result.Success, result.OutputPath)
	}
}