mirror of
https://github.com/edgelesssys/constellation.git
synced 2025-08-07 14:32:17 -04:00
17aecaaf5f
* constellation-lib: refactor init RPC Signed-off-by: Moritz Sanft <58110325+msanft@users.noreply.github.com> * constellation-lib: pass io.Writer for collecting logs Signed-off-by: Moritz Sanft <58110325+msanft@users.noreply.github.com> * constellation-lib: add init test Signed-off-by: Moritz Sanft <58110325+msanft@users.noreply.github.com> * constellation-lib: bin dialer to struct Signed-off-by: Moritz Sanft <58110325+msanft@users.noreply.github.com> * constellation-lib: set service CIDR on init Signed-off-by: Moritz Sanft <58110325+msanft@users.noreply.github.com> --------- Signed-off-by: Moritz Sanft <58110325+msanft@users.noreply.github.com>
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package constellation
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/cloud/cloudprovider"
|
|
"github.com/edgelesssys/constellation/v2/internal/crypto"
|
|
"github.com/edgelesssys/constellation/v2/internal/license"
|
|
"github.com/edgelesssys/constellation/v2/internal/logger"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCheckLicense(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
licenseChecker *stubLicenseChecker
|
|
wantErr bool
|
|
}{
|
|
"success": {
|
|
licenseChecker: &stubLicenseChecker{},
|
|
wantErr: false,
|
|
},
|
|
"check license error": {
|
|
licenseChecker: &stubLicenseChecker{checkLicenseErr: assert.AnError},
|
|
wantErr: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
a := &Applier{licenseChecker: tc.licenseChecker, log: logger.NewTest(t)}
|
|
_, err := a.CheckLicense(context.Background(), cloudprovider.Unknown, license.CommunityLicense)
|
|
if tc.wantErr {
|
|
require.Error(err)
|
|
} else {
|
|
require.NoError(err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
type stubLicenseChecker struct {
|
|
checkLicenseErr error
|
|
}
|
|
|
|
func (c *stubLicenseChecker) CheckLicense(context.Context, cloudprovider.Provider, string) (license.QuotaCheckResponse, error) {
|
|
return license.QuotaCheckResponse{}, c.checkLicenseErr
|
|
}
|
|
|
|
func TestGenerateMasterSecret(t *testing.T) {
|
|
assert := assert.New(t)
|
|
a := &Applier{log: logger.NewTest(t)}
|
|
sec, err := a.GenerateMasterSecret()
|
|
assert.NoError(err)
|
|
assert.Len(sec.Key, crypto.MasterSecretLengthDefault)
|
|
assert.Len(sec.Key, crypto.RNGLengthDefault)
|
|
}
|
|
|
|
func TestGenerateMeasurementSalt(t *testing.T) {
|
|
assert := assert.New(t)
|
|
a := &Applier{log: logger.NewTest(t)}
|
|
salt, err := a.GenerateMeasurementSalt()
|
|
assert.NoError(err)
|
|
assert.Len(salt, crypto.RNGLengthDefault)
|
|
}
|