111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
// Copyright IBM Corp. 2021, 2026
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package host
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/ec2"
|
|
"github.com/aws/aws-sdk-go-v2/service/ec2/types"
|
|
)
|
|
|
|
const (
|
|
testOptionErr = "test option error"
|
|
testGetUserErr = "test error for GetUser"
|
|
testGetCallerIdentityErr = "test error for GetCallerIdentity"
|
|
testDeleteAccessKeyErr = "test error for DeleteAccessKey"
|
|
testDescribeInstancesError = "DescribeInstances error"
|
|
)
|
|
|
|
type testMockEC2State struct {
|
|
DescribeInstancesCalled bool
|
|
DescribeInstancesInputParams *ec2.DescribeInstancesInput
|
|
DescribeInstancesCallCount int
|
|
DescribeInstancesInputs []*ec2.DescribeInstancesInput
|
|
ClientRegions []string
|
|
}
|
|
|
|
func (s *testMockEC2State) Reset() {
|
|
s.DescribeInstancesCalled = false
|
|
s.DescribeInstancesInputParams = nil
|
|
s.DescribeInstancesCallCount = 0
|
|
s.DescribeInstancesInputs = nil
|
|
s.ClientRegions = nil
|
|
}
|
|
|
|
type testMockEC2 struct {
|
|
EC2API
|
|
|
|
State *testMockEC2State
|
|
Region string
|
|
|
|
DescribeInstancesOutput *ec2.DescribeInstancesOutput
|
|
DescribeInstancesError error
|
|
}
|
|
|
|
type testMockEC2Option func(m *testMockEC2) error
|
|
|
|
func testMockEC2WithDescribeInstancesOutput(o *ec2.DescribeInstancesOutput) testMockEC2Option {
|
|
return func(m *testMockEC2) error {
|
|
m.DescribeInstancesOutput = o
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testMockEC2WithDescribeInstancesError(e error) testMockEC2Option {
|
|
return func(m *testMockEC2) error {
|
|
m.DescribeInstancesError = e
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func newTestMockEC2(state *testMockEC2State, opts ...testMockEC2Option) ec2APIFunc {
|
|
return func(cfgs ...aws.Config) (EC2API, error) {
|
|
m := &testMockEC2{
|
|
State: state,
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
if err := opt(m); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
for _, cfg := range cfgs {
|
|
// Last region takes precedence
|
|
if cfg.Region != "" {
|
|
m.Region = cfg.Region
|
|
}
|
|
}
|
|
|
|
if state != nil {
|
|
state.ClientRegions = append(state.ClientRegions, m.Region)
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
}
|
|
|
|
func (m *testMockEC2) DescribeInstances(ctx context.Context, input *ec2.DescribeInstancesInput, optFns ...func(*ec2.Options)) (*ec2.DescribeInstancesOutput, error) {
|
|
if m.State != nil {
|
|
m.State.DescribeInstancesCalled = true
|
|
m.State.DescribeInstancesInputParams = input
|
|
m.State.DescribeInstancesCallCount++
|
|
m.State.DescribeInstancesInputs = append(m.State.DescribeInstancesInputs, input)
|
|
}
|
|
|
|
if m.DescribeInstancesError != nil {
|
|
return nil, m.DescribeInstancesError
|
|
}
|
|
|
|
return m.DescribeInstancesOutput, nil
|
|
}
|
|
|
|
type ec2FilterSorter []types.Filter
|
|
|
|
func (s ec2FilterSorter) Len() int { return len(s) }
|
|
func (s ec2FilterSorter) Less(i, j int) bool { return *s[i].Name < *s[j].Name }
|
|
func (s ec2FilterSorter) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
|