Git-Mirrors/constellation
1
0
Fork 0
mirror of https://github.com/edgelesssys/constellation.git synced 2025-05-12 11:12:27 -04:00
Code Issues Projects Releases Packages Wiki Activity

config: dynamic attestation configuration through S3 backed API (#1808)

Browse source
This commit is contained in:
Adrian Stobbe 2023-05-25 17:43:44 +01:00 • committed by GitHub
parent 25211dc154
commit 0a6e5ec02e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
92 changed files with 1020 additions and 302 deletions
Download patch file Download diff file Expand all files Collapse all files

218
internal/api/fetcher/versionapi_test.go Normal file
View file

@ -0,0 +1,218 @@
/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package fetcher
import (
"bytes"
"context"
"encoding/json"
"io"
"net/http"
"testing"
"github.com/edgelesssys/constellation/v2/internal/api/versionsapi"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"go.uber.org/goleak"
)
func TestMain(m *testing.M) {
goleak.VerifyTestMain(m)
}
func TestFetchVersionList(t *testing.T) {
require := require.New(t)
majorList := func() *versionsapi.List {
return &versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMajor,
Base: "v1",
Kind: versionsapi.VersionKindImage,
Versions: []string{"v1.0", "v1.1", "v1.2"},
}
}
minorList := func() *versionsapi.List {
return &versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMinor,
Base: "v1.1",
Kind: versionsapi.VersionKindImage,
Versions: []string{"v1.1.0", "v1.1.1", "v1.1.2"},
}
}
majorListJSON, err := json.Marshal(majorList())
require.NoError(err)
minorListJSON, err := json.Marshal(minorList())
require.NoError(err)
inconsistentList := majorList()
inconsistentList.Base = "v2"
inconsistentListJSON, err := json.Marshal(inconsistentList)
require.NoError(err)
testCases := map[string]struct {
list versionsapi.List
serverPath string
serverResp *http.Response
wantList versionsapi.List
wantErr bool
}{
"major list fetched": {
list: versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMajor,
Base: "v1",
Kind: versionsapi.VersionKindImage,
},
serverPath: "/constellation/v1/ref/test-ref/stream/nightly/versions/major/v1/image.json",
serverResp: &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewBuffer(majorListJSON)),
},
wantList: *majorList(),
},
"minor list fetched": {
list: versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMinor,
Base: "v1.1",
Kind: versionsapi.VersionKindImage,
},
serverPath: "/constellation/v1/ref/test-ref/stream/nightly/versions/minor/v1.1/image.json",
serverResp: &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewBuffer(minorListJSON)),
},
wantList: *minorList(),
},
"list does not exist": {
list: versionsapi.List{
Ref: "another-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMajor,
Base: "v1",
Kind: versionsapi.VersionKindImage,
},
wantErr: true,
},
"invalid list requested": {
list: versionsapi.List{
Ref: "",
Stream: "unknown",
Granularity: versionsapi.GranularityMajor,
Base: "v1",
Kind: versionsapi.VersionKindImage,
},
wantErr: true,
},
"unexpected error code": {
list: versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMajor,
Base: "v1",
Kind: versionsapi.VersionKindImage,
},
serverPath: "/constellation/v1/ref/test-ref/stream/nightly/versions/major/v1/image.json",
serverResp: &http.Response{
StatusCode: http.StatusInternalServerError,
Body: io.NopCloser(bytes.NewBufferString("Internal Server Error")),
},
wantErr: true,
},
"invalid json returned": {
list: versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMajor,
Base: "v1",
Kind: versionsapi.VersionKindImage,
},
serverPath: "/constellation/v1/ref/test-ref/stream/nightly/versions/major/v1/image.json",
serverResp: &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewBufferString("invalid json")),
},
wantErr: true,
},
"invalid list returned": {
list: versionsapi.List{
Ref: "test-ref",
Stream: "nightly",
Granularity: versionsapi.GranularityMajor,
Base: "v2",
Kind: versionsapi.VersionKindImage,
},
serverPath: "/constellation/v1/ref/test-ref/stream/nightly/versions/major/v2/image.json",
serverResp: &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewBuffer(inconsistentListJSON)),
},
wantErr: true,
},
// TODO(katexochen): Remove or find strategy to implement this check in a generic way
// "response does not match request": {
// list: versionsapi.List{
// Ref: "test-ref",
// Stream: "nightly",
// Granularity: versionsapi.GranularityMajor,
// Base: "v3",
// Kind: versionsapi.VersionKindImage,
// },
// serverPath: "/constellation/v1/ref/test-ref/stream/nightly/versions/major/v3/image.json",
// serverResp: &http.Response{
// StatusCode: http.StatusOK,
// Body: io.NopCloser(bytes.NewBuffer(minorListJSON)),
// },
// wantErr: 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 {
if req.URL.Path != tc.serverPath {
return &http.Response{
StatusCode: http.StatusNotFound,
Body: io.NopCloser(bytes.NewBufferString("Not found.")),
}
}
return tc.serverResp
})
fetcher := VersionAPIFetcher{&fetcher{httpc: client}}
list, err := fetcher.FetchVersionList(context.Background(), tc.list)
if tc.wantErr {
assert.Error(err)
return
}
assert.NoError(err)
assert.Equal(tc.wantList, list)
})
}
}
type roundTripFunc func(req *http.Request) *http.Response
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) *http.Client {
return &http.Client{
Transport: fn,
}
}