image-builder: update upload API to match latest images

This commit updates the upload API to match the latest
version in the images library.
This commit is contained in:
Michael Vogt 2025-08-07 09:43:22 +02:00 committed by Achilleas Koutsou
parent eef98870f4
commit 6169260dcd
2 changed files with 9 additions and 3 deletions

View file

@ -102,7 +102,9 @@ func TestUploadWithAWSMock(t *testing.T) {
assert.Equal(t, bucketName, "aws-bucket-2")
assert.Equal(t, amiName, "aws-ami-3")
expectedBootMode := platform.BOOT_HYBRID
assert.Equal(t, &awscloud.UploaderOptions{TargetArch: tc.expectedUploadArch, BootMode: &expectedBootMode}, uploadOpts)
targetArch, err := arch.FromString(tc.expectedUploadArch)
assert.NoError(t, err)
assert.Equal(t, &awscloud.UploaderOptions{TargetArch: targetArch, BootMode: &expectedBootMode}, uploadOpts)
assert.Equal(t, 0, fa.checkCalls)
assert.Equal(t, 1, fa.uploadAndRegisterCalls)
@ -199,7 +201,7 @@ func TestBuildAndUploadWithAWSMock(t *testing.T) {
assert.Equal(t, bucketName, "aws-bucket-2")
assert.Equal(t, amiName, "aws-ami-3")
expectedBootMode := platform.BOOT_HYBRID
assert.Equal(t, &awscloud.UploaderOptions{BootMode: &expectedBootMode, TargetArch: arch.Current().String()}, uploadOpts)
assert.Equal(t, &awscloud.UploaderOptions{BootMode: &expectedBootMode, TargetArch: arch.Current()}, uploadOpts)
assert.Equal(t, 1, fa.checkCalls)
assert.Equal(t, 1, fa.uploadAndRegisterCalls)
assert.Equal(t, "fake-img-raw\n", fa.uploadAndRegisterRead.String())