test/distro/rhel84: add test for Arch.ListImageTypes

We didn't have one so I added it.

Signed-off-by: Ondřej Budai <ondrej@budai.cz>

internal/distro/rhel84/distro_test.go

@@ -8,6 +8,7 @@ import (
 	"github.com/osbuild/osbuild-composer/internal/distro/distro_test_common"
 	"github.com/osbuild/osbuild-composer/internal/distro/rhel84"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestFilenameFromType(t *testing.T) {
@@ -393,6 +394,58 @@ func TestDistro_ManifestError(t *testing.T) {
 	}
 }
 
+func TestArchitecture_ListImageTypes(t *testing.T) {
+	imgMap := []struct {
+		arch     string
+		imgNames []string
+	}{
+		{
+			arch: "x86_64",
+			imgNames: []string{
+				"ami",
+				"qcow2",
+				"openstack",
+				"rhel-edge-commit",
+				"tar",
+				"vhd",
+				"vmdk",
+			},
+		},
+		{
+			arch: "aarch64",
+			imgNames: []string{
+				"ami",
+				"qcow2",
+				"openstack",
+				"rhel-edge-commit",
+				"tar",
+			},
+		},
+		{
+			arch: "ppc64le",
+			imgNames: []string{
+				"qcow2",
+				"tar",
+			},
+		},
+		{
+			arch: "s390x",
+			imgNames: []string{
+				"qcow2",
+				"tar",
+			},
+		},
+	}
+
+	distro := rhel84.New()
+	for _, mapping := range imgMap {
+		arch, err := distro.GetArch(mapping.arch)
+		require.NoError(t, err)
+		imageTypes := arch.ListImageTypes()
+		require.ElementsMatch(t, mapping.imgNames, imageTypes)
+	}
+}
+
 func TestRhel84_ListArches(t *testing.T) {
 	distro := rhel84.New()
 	arches := distro.ListArches()