mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-09-21 16:56:03 +00:00
65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
|
//go:build integration
|
||
|
|
||
|
package license
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
)
|
||
|
|
||
|
func TestCheckQuotaIntegration(t *testing.T) {
|
||
|
testCases := map[string]struct {
|
||
|
license string
|
||
|
action string
|
||
|
wantQuota int
|
||
|
wantError bool
|
||
|
}{
|
||
|
"ES license has quota 256": {
|
||
|
license: "***REMOVED***",
|
||
|
action: test,
|
||
|
wantQuota: 256,
|
||
|
},
|
||
|
"OSS license has quota 8": {
|
||
|
license: CommunityLicense,
|
||
|
action: test,
|
||
|
wantQuota: 8,
|
||
|
},
|
||
|
"OSS license missing action": {
|
||
|
license: CommunityLicense,
|
||
|
action: "",
|
||
|
wantQuota: 8,
|
||
|
},
|
||
|
"Empty license assumes community": {
|
||
|
license: "",
|
||
|
action: test,
|
||
|
wantQuota: 8,
|
||
|
},
|
||
|
"Empty request": {
|
||
|
license: "",
|
||
|
action: "",
|
||
|
wantQuota: 8,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for name, tc := range testCases {
|
||
|
t.Run(name, func(t *testing.T) {
|
||
|
assert := assert.New(t)
|
||
|
|
||
|
client := NewClient()
|
||
|
|
||
|
resp, err := client.CheckQuota(CheckQuotaRequest{
|
||
|
Action: tc.action,
|
||
|
License: tc.license,
|
||
|
})
|
||
|
|
||
|
if tc.wantError {
|
||
|
assert.Error(err)
|
||
|
return
|
||
|
}
|
||
|
assert.NoError(err)
|
||
|
assert.Equal(tc.wantQuota, resp.Quota)
|
||
|
})
|
||
|
}
|
||
|
}
|