constellation/internal/attestation/gcp/issuer_test.go
Thomas Tendyck bd63aa3c6b add license headers
sed -i '1i/*\nCopyright (c) Edgeless Systems GmbH\n\nSPDX-License-Identifier: AGPL-3.0-only\n*/\n' `grep -rL --include='*.go' 'DO NOT EDIT'`
gofumpt -w .
2022-09-05 09:17:25 +02:00

103 lines
2.3 KiB
Go

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package gcp
import (
"encoding/json"
"errors"
"io"
"testing"
"github.com/google/go-tpm-tools/proto/attest"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestGetGCEInstanceInfo(t *testing.T) {
testCases := map[string]struct {
client fakeMetadataClient
wantErr bool
}{
"success": {
client: fakeMetadataClient{
projectIDString: "projectID",
instanceNameString: "instanceName",
zoneString: "zone",
},
wantErr: false,
},
"projectID error": {
client: fakeMetadataClient{
projectIDString: "projectID",
instanceNameString: "instanceName",
zoneString: "zone",
projecIDErr: errors.New("error"),
},
wantErr: true,
},
"instanceName error": {
client: fakeMetadataClient{
projectIDString: "projectID",
instanceNameString: "instanceName",
zoneString: "zone",
instanceNameErr: errors.New("error"),
},
wantErr: true,
},
"zone error": {
client: fakeMetadataClient{
projectIDString: "projectID",
instanceNameString: "instanceName",
zoneString: "zone",
zoneErr: errors.New("error"),
},
wantErr: true,
},
}
for name, tc := range testCases {
t.Run(name, func(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
var tpm io.ReadWriteCloser
out, err := getGCEInstanceInfo(tc.client)(tpm)
if tc.wantErr {
assert.Error(err)
} else {
assert.NoError(err)
var info attest.GCEInstanceInfo
require.NoError(json.Unmarshal(out, &info))
assert.Equal(tc.client.projectIDString, info.ProjectId)
assert.Equal(tc.client.instanceNameString, info.InstanceName)
assert.Equal(tc.client.zoneString, info.Zone)
}
})
}
}
type fakeMetadataClient struct {
projectIDString string
instanceNameString string
zoneString string
projecIDErr error
instanceNameErr error
zoneErr error
}
func (c fakeMetadataClient) projectID() (string, error) {
return c.projectIDString, c.projecIDErr
}
func (c fakeMetadataClient) instanceName() (string, error) {
return c.instanceNameString, c.instanceNameErr
}
func (c fakeMetadataClient) zone() (string, error) {
return c.zoneString, c.zoneErr
}