Blame internal/distro/fedora33/distro_test.go

Packit Service 509fd4
package fedora33_test
Packit Service 509fd4
Packit Service 509fd4
import (
Packit Service 509fd4
	"testing"
Packit Service 509fd4
Packit Service 509fd4
	"github.com/osbuild/osbuild-composer/internal/blueprint"
Packit Service bcdfb1
	"github.com/osbuild/osbuild-composer/internal/distro"
Packit Service 509fd4
	"github.com/osbuild/osbuild-composer/internal/distro/distro_test_common"
Packit Service 509fd4
	"github.com/osbuild/osbuild-composer/internal/distro/fedora33"
Packit Service 509fd4
	"github.com/stretchr/testify/assert"
Packit Service 509fd4
)
Packit Service 509fd4
Packit Service 509fd4
func TestFilenameFromType(t *testing.T) {
Packit Service 509fd4
	type args struct {
Packit Service 509fd4
		outputFormat string
Packit Service 509fd4
	}
Packit Service 509fd4
	tests := []struct {
Packit Service 509fd4
		name    string
Packit Service 509fd4
		args    args
Packit Service 509fd4
		want    string
Packit Service 509fd4
		want1   string
Packit Service 509fd4
		wantErr bool
Packit Service 509fd4
	}{
Packit Service 509fd4
		{
Packit Service 509fd4
			name:  "ami",
Packit Service 509fd4
			args:  args{"ami"},
Packit Service 509fd4
			want:  "image.raw",
Packit Service 509fd4
			want1: "application/octet-stream",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:  "openstack",
Packit Service 509fd4
			args:  args{"openstack"},
Packit Service 509fd4
			want:  "disk.qcow2",
Packit Service 509fd4
			want1: "application/x-qemu-disk",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:  "qcow2",
Packit Service 509fd4
			args:  args{"qcow2"},
Packit Service 509fd4
			want:  "disk.qcow2",
Packit Service 509fd4
			want1: "application/x-qemu-disk",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:  "vhd",
Packit Service 509fd4
			args:  args{"vhd"},
Packit Service 509fd4
			want:  "disk.vhd",
Packit Service 509fd4
			want1: "application/x-vhd",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:  "vmdk",
Packit Service 509fd4
			args:  args{"vmdk"},
Packit Service 509fd4
			want:  "disk.vmdk",
Packit Service 509fd4
			want1: "application/x-vmdk",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:    "invalid-output-type",
Packit Service 509fd4
			args:    args{"foobar"},
Packit Service 509fd4
			wantErr: true,
Packit Service 509fd4
		},
Packit Service 509fd4
	}
Packit Service 509fd4
	for _, tt := range tests {
Packit Service 509fd4
		t.Run(tt.name, func(t *testing.T) {
Packit Service 509fd4
			dist := fedora33.New()
Packit Service 509fd4
			arch, _ := dist.GetArch("x86_64")
Packit Service 509fd4
			imgType, err := arch.GetImageType(tt.args.outputFormat)
Packit Service 509fd4
			if (err != nil) != tt.wantErr {
Packit Service 509fd4
				t.Errorf("Arch.GetImageType() error = %v, wantErr %v", err, tt.wantErr)
Packit Service 509fd4
				return
Packit Service 509fd4
			}
Packit Service 509fd4
			if !tt.wantErr {
Packit Service 509fd4
				got := imgType.Filename()
Packit Service 509fd4
				got1 := imgType.MIMEType()
Packit Service 509fd4
				if got != tt.want {
Packit Service 509fd4
					t.Errorf("ImageType.Filename()  got = %v, want %v", got, tt.want)
Packit Service 509fd4
				}
Packit Service 509fd4
				if got1 != tt.want1 {
Packit Service 509fd4
					t.Errorf("ImageType.MIMEType() got1 = %v, want %v", got1, tt.want1)
Packit Service 509fd4
				}
Packit Service 509fd4
			}
Packit Service 509fd4
		})
Packit Service 509fd4
	}
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestImageType_BuildPackages(t *testing.T) {
Packit Service 509fd4
	x8664BuildPackages := []string{
Packit Service 509fd4
		"dnf",
Packit Service 509fd4
		"dosfstools",
Packit Service 509fd4
		"e2fsprogs",
Packit Service 509fd4
		"grub2-pc",
Packit Service 509fd4
		"policycoreutils",
Packit Service 509fd4
		"qemu-img",
Packit Service 509fd4
		"selinux-policy-targeted",
Packit Service 509fd4
		"systemd",
Packit Service 509fd4
		"tar",
Packit Service 509fd4
		"xz",
Packit Service 509fd4
	}
Packit Service 509fd4
	aarch64BuildPackages := []string{
Packit Service 509fd4
		"dnf",
Packit Service 509fd4
		"dosfstools",
Packit Service 509fd4
		"e2fsprogs",
Packit Service 509fd4
		"policycoreutils",
Packit Service 509fd4
		"qemu-img",
Packit Service 509fd4
		"selinux-policy-targeted",
Packit Service 509fd4
		"systemd",
Packit Service 509fd4
		"tar",
Packit Service 509fd4
		"xz",
Packit Service 509fd4
	}
Packit Service 509fd4
	buildPackages := map[string][]string{
Packit Service 509fd4
		"x86_64":  x8664BuildPackages,
Packit Service 509fd4
		"aarch64": aarch64BuildPackages,
Packit Service 509fd4
	}
Packit Service 509fd4
	d := fedora33.New()
Packit Service 509fd4
	for _, archLabel := range d.ListArches() {
Packit Service 509fd4
		archStruct, err := d.GetArch(archLabel)
Packit Service 509fd4
		if err != nil {
Packit Service 509fd4
			t.Errorf("d.GetArch(%v) returned err = %v; expected nil", archLabel, err)
Packit Service 509fd4
			continue
Packit Service 509fd4
		}
Packit Service 509fd4
		for _, itLabel := range archStruct.ListImageTypes() {
Packit Service 509fd4
			itStruct, err := archStruct.GetImageType(itLabel)
Packit Service 509fd4
			if err != nil {
Packit Service 509fd4
				t.Errorf("d.GetArch(%v) returned err = %v; expected nil", archLabel, err)
Packit Service 509fd4
				continue
Packit Service 509fd4
			}
Packit Service 15f37d
			buildPkgs := itStruct.PackageSets(blueprint.Blueprint{})["build-packages"]
Packit Service 15f37d
			assert.NotNil(t, buildPkgs)
Packit Service 509fd4
			if itLabel == "fedora-iot-commit" {
Packit Service 509fd4
				// For now we only include rpm-ostree when building fedora-iot-commit image types, this we may want
Packit Service 509fd4
				// to reconsider. The only reason to specia-case it is that it might pull in a lot of dependencies
Packit Service 509fd4
				// for a niche usecase.
Packit Service 15f37d
				assert.ElementsMatch(t, append(buildPackages[archLabel], "rpm-ostree"), buildPkgs.Include)
Packit Service 509fd4
			} else {
Packit Service 15f37d
				assert.ElementsMatch(t, buildPackages[archLabel], buildPkgs.Include)
Packit Service 509fd4
			}
Packit Service 509fd4
		}
Packit Service 509fd4
	}
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestImageType_Name(t *testing.T) {
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	imgMap := []struct {
Packit Service 509fd4
		arch     string
Packit Service 509fd4
		imgNames []string
Packit Service 509fd4
	}{
Packit Service 509fd4
		{
Packit Service 509fd4
			arch: "x86_64",
Packit Service 509fd4
			imgNames: []string{
Packit Service 509fd4
				"ami",
Packit Service 509fd4
				"qcow2",
Packit Service 509fd4
				"openstack",
Packit Service 509fd4
				"vhd",
Packit Service 509fd4
				"vmdk",
Packit Service 509fd4
			},
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			arch: "aarch64",
Packit Service 509fd4
			imgNames: []string{
Packit Service 509fd4
				"ami",
Packit Service 509fd4
				"qcow2",
Packit Service 509fd4
				"openstack",
Packit Service 509fd4
			},
Packit Service 509fd4
		},
Packit Service 509fd4
	}
Packit Service 509fd4
	for _, mapping := range imgMap {
Packit Service 509fd4
		arch, err := distro.GetArch(mapping.arch)
Packit Service 509fd4
		if assert.NoError(t, err) {
Packit Service 509fd4
			for _, imgName := range mapping.imgNames {
Packit Service 509fd4
				imgType, err := arch.GetImageType(imgName)
Packit Service 509fd4
				if assert.NoError(t, err) {
Packit Service 509fd4
					assert.Equalf(t, imgName, imgType.Name(), "arch: %s", mapping.arch)
Packit Service 509fd4
				}
Packit Service 509fd4
			}
Packit Service 509fd4
		}
Packit Service 509fd4
	}
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestImageType_Size(t *testing.T) {
Packit Service 509fd4
	const gigaByte = 1024 * 1024 * 1024
Packit Service 509fd4
	sizeMap := []struct {
Packit Service 509fd4
		name       string
Packit Service 509fd4
		inputSize  uint64
Packit Service 509fd4
		outputSize uint64
Packit Service 509fd4
	}{
Packit Service 509fd4
		{
Packit Service 509fd4
			name:       "ami",
Packit Service 509fd4
			inputSize:  6*gigaByte + 1,
Packit Service 509fd4
			outputSize: 6*gigaByte + 1,
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:       "ami",
Packit Service 509fd4
			inputSize:  0,
Packit Service 509fd4
			outputSize: 6 * gigaByte,
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:       "vhd",
Packit Service 509fd4
			inputSize:  10 * gigaByte,
Packit Service 509fd4
			outputSize: 10 * gigaByte,
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:       "vhd",
Packit Service 509fd4
			inputSize:  10*gigaByte - 1,
Packit Service 509fd4
			outputSize: 10 * gigaByte,
Packit Service 509fd4
		},
Packit Service 509fd4
	}
Packit Service 509fd4
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	arch, err := distro.GetArch("x86_64")
Packit Service 509fd4
	if assert.NoError(t, err) {
Packit Service 509fd4
		for _, mapping := range sizeMap {
Packit Service 509fd4
			imgType, err := arch.GetImageType(mapping.name)
Packit Service 509fd4
			if assert.NoError(t, err) {
Packit Service 509fd4
				size := imgType.Size(mapping.inputSize)
Packit Service 509fd4
				assert.Equalf(t, mapping.outputSize, size, "Image type: %s, input size: %d, expected: %d, got: %d",
Packit Service 509fd4
					mapping.name, mapping.inputSize, mapping.outputSize, size)
Packit Service 509fd4
			}
Packit Service 509fd4
		}
Packit Service 509fd4
	}
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestImageType_BasePackages(t *testing.T) {
Packit Service 509fd4
	pkgMaps := []struct {
Packit Service 509fd4
		name               string
Packit Service 509fd4
		basePackages       []string
Packit Service 509fd4
		bootloaderPackages []string
Packit Service 509fd4
		excludedPackages   []string
Packit Service 509fd4
		bootable           bool
Packit Service 509fd4
	}{
Packit Service 509fd4
		{
Packit Service 509fd4
			name: "ami",
Packit Service 509fd4
			basePackages: []string{
Packit Service 509fd4
				"@Core",
Packit Service 509fd4
				"chrony",
Packit Service 509fd4
				"selinux-policy-targeted",
Packit Service 509fd4
				"langpacks-en",
Packit Service 509fd4
				"libxcrypt-compat",
Packit Service 509fd4
				"xfsprogs",
Packit Service 509fd4
				"cloud-init",
Packit Service 509fd4
				"checkpolicy",
Packit Service 509fd4
				"net-tools",
Packit Service bcdfb1
Packit Service bcdfb1
				// Default from Blueprint
Packit Service bcdfb1
				"kernel",
Packit Service 509fd4
			},
Packit Service 509fd4
			bootloaderPackages: []string{
Packit Service 509fd4
				"dracut-config-generic",
Packit Service 509fd4
				"grub2-pc",
Packit Service 509fd4
			},
Packit Service 509fd4
			excludedPackages: []string{
Packit Service 509fd4
				"dracut-config-rescue",
Packit Service 509fd4
				"geolite2-city",
Packit Service 509fd4
				"geolite2-country",
Packit Service 509fd4
				"zram-generator-defaults",
Packit Service 509fd4
			},
Packit Service 509fd4
			bootable: true,
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name: "openstack",
Packit Service 509fd4
			basePackages: []string{
Packit Service 509fd4
				"@Core",
Packit Service 509fd4
				"chrony",
Packit Service 509fd4
				"selinux-policy-targeted",
Packit Service 509fd4
				"spice-vdagent",
Packit Service 509fd4
				"qemu-guest-agent",
Packit Service 509fd4
				"xen-libs",
Packit Service 509fd4
				"langpacks-en",
Packit Service 509fd4
				"cloud-init",
Packit Service 509fd4
				"libdrm",
Packit Service bcdfb1
Packit Service bcdfb1
				// Default from Blueprint
Packit Service bcdfb1
				"kernel",
Packit Service 509fd4
			},
Packit Service 509fd4
			bootloaderPackages: []string{
Packit Service 509fd4
				"dracut-config-generic",
Packit Service 509fd4
				"grub2-pc",
Packit Service 509fd4
			},
Packit Service 509fd4
			excludedPackages: []string{
Packit Service 509fd4
				"dracut-config-rescue",
Packit Service 509fd4
				"geolite2-city",
Packit Service 509fd4
				"geolite2-country",
Packit Service 509fd4
				"zram-generator-defaults",
Packit Service 509fd4
			},
Packit Service 509fd4
			bootable: true,
Packit Service 509fd4
		},
Packit Service 509fd4
	}
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	arch, err := distro.GetArch("x86_64")
Packit Service 509fd4
	assert.NoError(t, err)
Packit Service 509fd4
Packit Service 509fd4
	for _, pkgMap := range pkgMaps {
Packit Service 509fd4
		imgType, err := arch.GetImageType(pkgMap.name)
Packit Service 509fd4
		assert.NoError(t, err)
Packit Service 15f37d
		packages := imgType.PackageSets(blueprint.Blueprint{})["packages"]
Packit Service 15f37d
		assert.NotNil(t, packages)
Packit Service 509fd4
		assert.Equalf(
Packit Service 509fd4
			t,
Packit Service 509fd4
			append(pkgMap.basePackages, pkgMap.bootloaderPackages...),
Packit Service 15f37d
			packages.Include,
Packit Service 509fd4
			"image type: %s",
Packit Service 509fd4
			pkgMap.name,
Packit Service 509fd4
		)
Packit Service 15f37d
		assert.Equalf(t, pkgMap.excludedPackages, packages.Exclude, "image type: %s", pkgMap.name)
Packit Service 509fd4
	}
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestDistro_Manifest(t *testing.T) {
Packit Service 509fd4
	distro_test_common.TestDistro_Manifest(t, "../../../test/data/manifests/", "fedora_33*", fedora33.New())
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service bcdfb1
// Check that Manifest() function returns an error for unsupported
Packit Service bcdfb1
// configurations.
Packit Service bcdfb1
func TestDistro_ManifestError(t *testing.T) {
Packit Service bcdfb1
	// Currently, the only unsupported configuration is OSTree commit types
Packit Service bcdfb1
	// with Kernel boot options
Packit Service bcdfb1
	f33distro := fedora33.New()
Packit Service bcdfb1
	bp := blueprint.Blueprint{
Packit Service bcdfb1
		Customizations: &blueprint.Customizations{
Packit Service bcdfb1
			Kernel: &blueprint.KernelCustomization{
Packit Service bcdfb1
				Append: "debug",
Packit Service bcdfb1
			},
Packit Service bcdfb1
		},
Packit Service bcdfb1
	}
Packit Service bcdfb1
Packit Service bcdfb1
	for _, archName := range f33distro.ListArches() {
Packit Service bcdfb1
		arch, _ := f33distro.GetArch(archName)
Packit Service bcdfb1
		for _, imgTypeName := range arch.ListImageTypes() {
Packit Service bcdfb1
			imgType, _ := arch.GetImageType(imgTypeName)
Packit Service 15f37d
			_, err := imgType.Manifest(bp.Customizations, distro.ImageOptions{}, nil, nil, 0)
Packit Service bcdfb1
			if imgTypeName == "fedora-iot-commit" {
Packit Service bcdfb1
				assert.EqualError(t, err, "kernel boot parameter customizations are not supported for ostree types")
Packit Service bcdfb1
			} else {
Packit Service bcdfb1
				assert.NoError(t, err)
Packit Service bcdfb1
			}
Packit Service bcdfb1
		}
Packit Service bcdfb1
	}
Packit Service bcdfb1
}
Packit Service bcdfb1
Packit Service 509fd4
func TestFedora33_ListArches(t *testing.T) {
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	arches := distro.ListArches()
Packit Service 509fd4
	assert.Equal(t, []string{"aarch64", "x86_64"}, arches)
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestFedora33_GetArch(t *testing.T) {
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	arches := []struct {
Packit Service 509fd4
		name          string
Packit Service 509fd4
		errorExpected bool
Packit Service 509fd4
	}{
Packit Service 509fd4
		{
Packit Service 509fd4
			name: "x86_64",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name: "aarch64",
Packit Service 509fd4
		},
Packit Service 509fd4
		{
Packit Service 509fd4
			name:          "foo-arch",
Packit Service 509fd4
			errorExpected: true,
Packit Service 509fd4
		},
Packit Service 509fd4
	}
Packit Service 509fd4
Packit Service 509fd4
	for _, a := range arches {
Packit Service 509fd4
		actualArch, err := distro.GetArch(a.name)
Packit Service 509fd4
		if !a.errorExpected {
Packit Service 509fd4
			assert.Equal(t, a.name, actualArch.Name())
Packit Service 509fd4
			assert.NoError(t, err)
Packit Service 509fd4
		} else {
Packit Service 509fd4
			assert.Nil(t, actualArch)
Packit Service 509fd4
			assert.Error(t, err)
Packit Service 509fd4
		}
Packit Service 509fd4
	}
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestFedora33_Name(t *testing.T) {
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	assert.Equal(t, "fedora-33", distro.Name())
Packit Service 509fd4
}
Packit Service 509fd4
Packit Service 509fd4
func TestFedora33_ModulePlatformID(t *testing.T) {
Packit Service 509fd4
	distro := fedora33.New()
Packit Service 509fd4
	assert.Equal(t, "platform:f33", distro.ModulePlatformID())
Packit Service 509fd4
}
Packit Service bcdfb1
Packit Service bcdfb1
func TestFedora33_KernelOption(t *testing.T) {
Packit Service bcdfb1
	distro_test_common.TestDistro_KernelOption(t, fedora33.New())
Packit Service bcdfb1
}