package azure import ( "testing" "github.com/edgelesssys/constellation/coordinator/attestation/vtpm" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestGetSNPAttestation(t *testing.T) { testCases := map[string]struct { tpmFunc vtpm.TPMOpenFunc errExpected bool }{ "success": { tpmFunc: vtpm.OpenSimulatedTPM, errExpected: false, }, } for name, tc := range testCases { t.Run(name, func(t *testing.T) { assert := assert.New(t) require := require.New(t) tpm, err := tc.tpmFunc() require.NoError(err) defer tpm.Close() _, err = getSNPAttestation(tpm) if tc.errExpected { assert.Error(err) } else { assert.NoError(err) } }) } }