constellation/internal/license/license_integration_test.go

//go:build integration

package license

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestCheckQuotaIntegration(t *testing.T) {
	testCases := map[string]struct {
		license   string
		action    Action
		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()

			req := CheckQuotaRequest{
				Action:  tc.action,
				License: tc.license,
			}
			resp, err := client.CheckQuota(context.Background(), req)

			if tc.wantError {
				assert.Error(err)
				return
			}
			assert.NoError(err)
			assert.Equal(tc.wantQuota, resp.Quota)
		})
	}
}