From 79acce46211d417daeae3db26347721e5868b806 Mon Sep 17 00:00:00 2001 From: Bjorn Neergaard Date: Wed, 27 Sep 2023 11:46:47 -0600 Subject: [PATCH] integration: use mediatype helpers Signed-off-by: Bjorn Neergaard --- integration/client/export_test.go | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/integration/client/export_test.go b/integration/client/export_test.go index 59ef7193e..c13b2c48c 100644 --- a/integration/client/export_test.go +++ b/integration/client/export_test.go @@ -116,8 +116,7 @@ func TestExportDockerManifest(t *testing.T) { // test single-platform export var result ocispec.Descriptor err = images.Walk(ctx, images.HandlerFunc(func(ctx context.Context, desc ocispec.Descriptor) ([]ocispec.Descriptor, error) { - switch desc.MediaType { - case images.MediaTypeDockerSchema2Manifest, ocispec.MediaTypeImageManifest: + if images.IsManifestType(desc.MediaType) { p, err := content.ReadBlob(ctx, client.ContentStore(), desc) if err != nil { return nil, err @@ -132,7 +131,7 @@ func TestExportDockerManifest(t *testing.T) { result = desc } return nil, nil - case images.MediaTypeDockerSchema2ManifestList, ocispec.MediaTypeImageIndex: + } else if images.IsIndexType(desc.MediaType) { p, err := content.ReadBlob(ctx, client.ContentStore(), desc) if err != nil { return nil, err