constellation/internal/license/license_integration_test.go

55 lines
900 B
Go
Raw Normal View History

//go:build integration
/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package license
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
)
func TestQuotaCheckIntegration(t *testing.T) {
testCases := map[string]struct {
license string
wantQuota int
wantError bool
}{
"OSS license has quota 8": {
license: CommunityLicense,
wantQuota: 8,
},
"Empty license assumes community": {
license: "",
wantQuota: 8,
},
}
for name, tc := range testCases {
t.Run(name, func(t *testing.T) {
assert := assert.New(t)
client := NewClient()
req := QuotaCheckRequest{
Action: test,
License: tc.license,
}
resp, err := client.QuotaCheck(context.Background(), req)
if tc.wantError {
assert.Error(err)
return
}
assert.NoError(err)
assert.Equal(tc.wantQuota, resp.Quota)
})
}
}