/* Copyright (c) Edgeless Systems GmbH SPDX-License-Identifier: AGPL-3.0-only */ package gcp import ( "context" "errors" "testing" "github.com/edgelesssys/constellation/internal/cloud/metadata" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestList(t *testing.T) { err := errors.New("some err") uid := "1234" instancesGenerator := func() *[]metadata.InstanceMetadata { return &[]metadata.InstanceMetadata{ { Name: "someInstance", ProviderID: "gce://someProject/someZone/someInstance", VPCIP: "192.0.2.0", }, } } testCases := map[string]struct { client stubGCPClient instancesGenerator func() *[]metadata.InstanceMetadata instancesMutator func(*[]metadata.InstanceMetadata) wantErr bool wantInstances []metadata.InstanceMetadata }{ "retrieve works": { client: stubGCPClient{ projectID: "someProjectID", zone: "someZone", retrieveInstanceMetadaValues: map[string]string{ "constellation-uid": uid, }, }, instancesGenerator: instancesGenerator, wantInstances: []metadata.InstanceMetadata{ { Name: "someInstance", ProviderID: "gce://someProject/someZone/someInstance", VPCIP: "192.0.2.0", }, }, }, "retrieve error is detected": { client: stubGCPClient{ projectID: "someProjectID", zone: "someZone", retrieveInstanceMetadaValues: map[string]string{ "constellation-uid": uid, }, retrieveInstancesErr: err, }, instancesGenerator: instancesGenerator, wantErr: true, }, "project metadata retrieval error is detected": { client: stubGCPClient{ retrieveProjectIDErr: err, }, instancesGenerator: instancesGenerator, wantErr: true, }, "zone retrieval error is detected": { client: stubGCPClient{ retrieveZoneErr: err, }, instancesGenerator: instancesGenerator, wantErr: true, }, } for name, tc := range testCases { t.Run(name, func(t *testing.T) { assert := assert.New(t) require := require.New(t) tc.client.retrieveInstancesValues = *tc.instancesGenerator() if tc.instancesMutator != nil { tc.instancesMutator(&tc.client.retrieveInstancesValues) } metadata := New(&tc.client) instances, err := metadata.List(context.Background()) if tc.wantErr { assert.Error(err) return } require.NoError(err) assert.ElementsMatch(tc.wantInstances, instances) }) } } func TestSelf(t *testing.T) { err := errors.New("some err") uid := "1234" testCases := map[string]struct { client stubGCPClient wantErr bool wantInstance metadata.InstanceMetadata }{ "retrieve works": { client: stubGCPClient{ projectID: "someProjectID", zone: "someZone", retrieveInstanceValue: metadata.InstanceMetadata{ Name: "someInstance", ProviderID: "gce://someProject/someZone/someInstance", VPCIP: "192.0.2.0", }, }, wantInstance: metadata.InstanceMetadata{ Name: "someInstance", ProviderID: "gce://someProject/someZone/someInstance", VPCIP: "192.0.2.0", }, }, "retrieve error is detected": { client: stubGCPClient{ projectID: "someProjectID", zone: "someZone", retrieveInstanceMetadaValues: map[string]string{ "constellation-uid": uid, }, retrieveInstanceErr: err, }, wantErr: true, }, "project id retrieval error is detected": { client: stubGCPClient{ retrieveProjectIDErr: err, }, wantErr: true, }, "zone retrieval error is detected": { client: stubGCPClient{ retrieveZoneErr: err, }, wantErr: true, }, "instance name retrieval error is detected": { client: stubGCPClient{ retrieveInstanceNameErr: err, }, wantErr: true, }, } for name, tc := range testCases { t.Run(name, func(t *testing.T) { assert := assert.New(t) require := require.New(t) cloud := New(&tc.client) instance, err := cloud.Self(context.Background()) if tc.wantErr { assert.Error(err) return } require.NoError(err) assert.Equal(tc.wantInstance, instance) }) } } func TestGetInstance(t *testing.T) { err := errors.New("some err") testCases := map[string]struct { providerID string client stubGCPClient wantErr bool wantInstance metadata.InstanceMetadata }{ "retrieve works": { providerID: "gce://someProject/someZone/someInstance", client: stubGCPClient{ retrieveInstanceValue: metadata.InstanceMetadata{ Name: "someInstance", ProviderID: "gce://someProject/someZone/someInstance", VPCIP: "192.0.2.0", }, }, wantInstance: metadata.InstanceMetadata{ Name: "someInstance", ProviderID: "gce://someProject/someZone/someInstance", VPCIP: "192.0.2.0", }, }, "retrieve error is detected": { providerID: "gce://someProject/someZone/someInstance", client: stubGCPClient{ retrieveInstanceErr: err, }, wantErr: true, }, "malformed providerID with too many fields is detected": { providerID: "gce://someProject/someZone/someInstance/tooMany/fields", wantErr: true, }, "malformed providerID with too few fields is detected": { providerID: "gce://someProject", wantErr: true, }, } for name, tc := range testCases { t.Run(name, func(t *testing.T) { assert := assert.New(t) require := require.New(t) cloud := New(&tc.client) instance, err := cloud.GetInstance(context.Background(), tc.providerID) if tc.wantErr { assert.Error(err) return } require.NoError(err) assert.Equal(tc.wantInstance, instance) }) } } type stubGCPClient struct { retrieveUIDValue string retrieveUIDErr error retrieveInstanceValue metadata.InstanceMetadata retrieveInstanceErr error retrieveInstancesValues []metadata.InstanceMetadata retrieveInstancesErr error retrieveInstanceMetadaValues map[string]string retrieveInstanceMetadataErr error retrieveSubnetworkAliasErr error projectID string zone string instanceName string loadBalancerIP string retrieveProjectIDErr error retrieveZoneErr error retrieveInstanceNameErr error setInstanceMetadataErr error unsetInstanceMetadataErr error retrieveLoadBalancerErr error instanceMetadataProjects []string instanceMetadataZones []string instanceMetadataInstanceNames []string instanceMetadataKeys []string instanceMetadataValues []string unsetMetadataProjects []string unsetMetadataZones []string unsetMetadataInstanceNames []string unsetMetadataKeys []string } func (s *stubGCPClient) RetrieveInstances(ctx context.Context, project, zone string) ([]metadata.InstanceMetadata, error) { return s.retrieveInstancesValues, s.retrieveInstancesErr } func (s *stubGCPClient) RetrieveInstance(ctx context.Context, project, zone string, instanceName string) (metadata.InstanceMetadata, error) { return s.retrieveInstanceValue, s.retrieveInstanceErr } func (s *stubGCPClient) RetrieveInstanceMetadata(attr string) (string, error) { return s.retrieveInstanceMetadaValues[attr], s.retrieveInstanceMetadataErr } func (s *stubGCPClient) RetrieveProjectID() (string, error) { return s.projectID, s.retrieveProjectIDErr } func (s *stubGCPClient) RetrieveZone() (string, error) { return s.zone, s.retrieveZoneErr } func (s *stubGCPClient) RetrieveInstanceName() (string, error) { return s.instanceName, s.retrieveInstanceNameErr } func (s *stubGCPClient) RetrieveLoadBalancerEndpoint(ctx context.Context, project, zone string) (string, error) { return s.loadBalancerIP, s.retrieveLoadBalancerErr } func (s *stubGCPClient) UID() (string, error) { return s.retrieveUIDValue, s.retrieveUIDErr } func (s *stubGCPClient) SetInstanceMetadata(ctx context.Context, project, zone, instanceName, key, value string) error { s.instanceMetadataProjects = append(s.instanceMetadataProjects, project) s.instanceMetadataZones = append(s.instanceMetadataZones, zone) s.instanceMetadataInstanceNames = append(s.instanceMetadataInstanceNames, instanceName) s.instanceMetadataKeys = append(s.instanceMetadataKeys, key) s.instanceMetadataValues = append(s.instanceMetadataValues, value) return s.setInstanceMetadataErr } func (s *stubGCPClient) UnsetInstanceMetadata(ctx context.Context, project, zone, instanceName, key string) error { s.unsetMetadataProjects = append(s.unsetMetadataProjects, project) s.unsetMetadataZones = append(s.unsetMetadataZones, zone) s.unsetMetadataInstanceNames = append(s.unsetMetadataInstanceNames, instanceName) s.unsetMetadataKeys = append(s.unsetMetadataKeys, key) return s.unsetInstanceMetadataErr } func (s *stubGCPClient) RetrieveSubnetworkAliasCIDR(ctx context.Context, project, zone, instanceName string) (string, error) { return "", s.retrieveSubnetworkAliasErr }