mirror of
https://github.com/mulgadc/spinifex.git
synced 2026-10-15 17:11:30 +00:00
Julian Sommer
57 lines
2.1 KiB
Go
57 lines
2.1 KiB
Go
//go:build integration
|
|
|
|
package integration
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/mulgadc/spinifex/spinifex/ebsmetadata"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestImage ports tests/e2e/single/image_test.go's runImage describe-images
|
|
// round-trip. The live test resolved a pre-staged gold-image AMI via
|
|
// needAMI (bootstrap-install.sh's `spx admin images import`); this tier has
|
|
// no such bootstrap, so it registers its own AMI against a hand-seeded
|
|
// snapshot via the real RegisterImage path instead — same production
|
|
// ImageServiceImpl.DescribeImages contract, without depending on a live
|
|
// cluster's pre-imported gold image.
|
|
func TestImage(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
gw := StartGateway(t)
|
|
_, store := StartImageDaemonLite(t, gw)
|
|
ec2Cli := gw.EC2Client(t)
|
|
|
|
const snapshotID = "snap-integration-test-image"
|
|
require.NoError(t, ebsmetadata.NewStore(store, testImageBucket).PutSnapshot(t.Context(), ebsmetadata.Snapshot{
|
|
SnapshotID: snapshotID,
|
|
VolumeID: "vol-integration-test-image",
|
|
VolumeSize: 8,
|
|
State: "completed",
|
|
OwnerID: gw.AccountID,
|
|
}), "seed snapshot metadata")
|
|
|
|
registerOut, err := ec2Cli.RegisterImage(&ec2.RegisterImageInput{
|
|
Name: aws.String("integration-test-ami"),
|
|
RootDeviceName: aws.String("/dev/sda1"),
|
|
BlockDeviceMappings: []*ec2.BlockDeviceMapping{{
|
|
DeviceName: aws.String("/dev/sda1"),
|
|
Ebs: &ec2.EbsBlockDevice{SnapshotId: aws.String(snapshotID)},
|
|
}},
|
|
})
|
|
require.NoError(t, err, "register-image")
|
|
amiID := aws.StringValue(registerOut.ImageId)
|
|
require.NotEmpty(t, amiID, "register-image returned empty ImageId")
|
|
|
|
byID, err := ec2Cli.DescribeImages(&ec2.DescribeImagesInput{
|
|
ImageIds: []*string{aws.String(amiID)},
|
|
})
|
|
require.NoErrorf(t, err, "describe-images %s", amiID)
|
|
require.Lenf(t, byID.Images, 1, "expected exactly 1 image for %s, got %d", amiID, len(byID.Images))
|
|
require.Equal(t, amiID, aws.StringValue(byID.Images[0].ImageId), "round-trip AMI ID mismatch")
|
|
require.NotEmpty(t, aws.StringValue(byID.Images[0].Name), "lookup-by-ID returned empty Name")
|
|
}
|