constellation/internal/license/license_test.go
Fabian Kammel 45beec15f5 AB#2360 enterprise build tag (#397)
* enterprise build switch to disable license checking in default (OSS) version
* remove community license quota
* empty image references on OSS build in config
Signed-off-by: Fabian Kammel <fk@edgeless.systems>
2022-08-25 14:06:29 +02:00

95 lines
2.2 KiB
Go

package license
import (
"bytes"
"context"
"io"
"net/http"
"testing"
"github.com/stretchr/testify/assert"
)
// roundTripFunc .
type roundTripFunc func(req *http.Request) *http.Response
// RoundTrip .
func (f roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
return f(req), nil
}
// newTestClient returns *http.Client with Transport replaced to avoid making real calls.
func newTestClient(fn roundTripFunc) *Client {
return &Client{
httpClient: &http.Client{
Transport: fn,
},
}
}
func TestQuotaCheck(t *testing.T) {
testCases := map[string]struct {
license string
serverResponse string
serverResponseCode int
serverResponseContent string
wantQuota int
wantError bool
}{
"success": {
license: "0c0a6558-f8af-4063-bf61-92e7ac4cb052",
serverResponse: "{\"quota\":256}",
serverResponseCode: http.StatusOK,
serverResponseContent: "application/json",
wantQuota: 256,
},
"404": {
serverResponseCode: http.StatusNotFound,
wantError: true,
},
"HTML not JSON": {
serverResponseCode: http.StatusOK,
serverResponseContent: "text/html",
wantError: true,
},
"promise JSON but actually HTML": {
serverResponseCode: http.StatusOK,
serverResponse: "<html><head></head></html>",
serverResponseContent: "application/json",
wantError: true,
},
}
for name, tc := range testCases {
t.Run(name, func(t *testing.T) {
assert := assert.New(t)
client := newTestClient(func(req *http.Request) *http.Response {
r := &http.Response{
StatusCode: tc.serverResponseCode,
Body: io.NopCloser(bytes.NewBufferString(tc.serverResponse)),
Header: make(http.Header),
}
r.Header.Set("Content-Type", tc.serverResponseContent)
return r
})
resp, err := client.QuotaCheck(context.Background(), QuotaCheckRequest{
Action: test,
License: tc.license,
})
if tc.wantError {
assert.Error(err)
return
}
assert.NoError(err)
assert.Equal(tc.wantQuota, resp.Quota)
})
}
}
func Test_licenseURL(t *testing.T) {
assert.Equal(t, "https://license.confidential.cloud/api/v1/license", licenseURL().String())
}