mirror of
https://github.com/project-zot/zot.git
synced 2024-12-23 22:27:35 -05:00
ba6f347d8d
Which could be imported independently. See more details: 1. "zotregistry.io/zot/pkg/test/common" - currently used as tcommon "zotregistry.io/zot/pkg/test/common" - inside pkg/test test "zotregistry.io/zot/pkg/test/common" - in tests . "zotregistry.io/zot/pkg/test/common" - in tests Decouple zb from code in test/pkg in order to keep the size small. 2. "zotregistry.io/zot/pkg/test/image-utils" - curently used as . "zotregistry.io/zot/pkg/test/image-utils" 3. "zotregistry.io/zot/pkg/test/deprecated" - curently used as "zotregistry.io/zot/pkg/test/deprecated" This one will bre replaced gradually by image-utils in the future. 4. "zotregistry.io/zot/pkg/test/signature" - (cosign + notation) use as "zotregistry.io/zot/pkg/test/signature" 5. "zotregistry.io/zot/pkg/test/auth" - (bearer + oidc) curently used as authutils "zotregistry.io/zot/pkg/test/auth" 6. "zotregistry.io/zot/pkg/test/oci-utils" - curently used as ociutils "zotregistry.io/zot/pkg/test/oci-utils" Some unused functions were removed, some were replaced, and in a few cases specific funtions were moved to the files they were used in. Added an interface for the StoreController, this reduces the number of imports of the entire image store, decreasing binary size for tests. If the zb code was still coupled with pkg/test, this would have reflected in zb size. Signed-off-by: Andrei Aaron <aaaron@luxoft.com>
176 lines
5.4 KiB
Go
176 lines
5.4 KiB
Go
//go:build userprefs
|
|
// +build userprefs
|
|
|
|
package extensions_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gorilla/mux"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"gopkg.in/resty.v1"
|
|
|
|
zerr "zotregistry.io/zot/errors"
|
|
"zotregistry.io/zot/pkg/api"
|
|
"zotregistry.io/zot/pkg/api/config"
|
|
"zotregistry.io/zot/pkg/api/constants"
|
|
"zotregistry.io/zot/pkg/extensions"
|
|
extconf "zotregistry.io/zot/pkg/extensions/config"
|
|
"zotregistry.io/zot/pkg/log"
|
|
mTypes "zotregistry.io/zot/pkg/meta/types"
|
|
test "zotregistry.io/zot/pkg/test/common"
|
|
"zotregistry.io/zot/pkg/test/mocks"
|
|
)
|
|
|
|
var ErrTestError = errors.New("TestError")
|
|
|
|
func TestAllowedMethodsHeaderUserPrefs(t *testing.T) {
|
|
defaultVal := true
|
|
|
|
Convey("Test http options response", t, func() {
|
|
conf := config.New()
|
|
port := test.GetFreePort()
|
|
conf.HTTP.Port = port
|
|
conf.Extensions = &extconf.ExtensionConfig{}
|
|
conf.Extensions.Search = &extconf.SearchConfig{}
|
|
conf.Extensions.Search.Enable = &defaultVal
|
|
conf.Extensions.Search.CVE = nil
|
|
conf.Extensions.UI = &extconf.UIConfig{}
|
|
conf.Extensions.UI.Enable = &defaultVal
|
|
|
|
baseURL := test.GetBaseURL(port)
|
|
|
|
ctlr := api.NewController(conf)
|
|
ctlr.Config.Storage.RootDirectory = t.TempDir()
|
|
|
|
ctrlManager := test.NewControllerManager(ctlr)
|
|
|
|
ctrlManager.StartAndWait(port)
|
|
defer ctrlManager.StopServer()
|
|
|
|
resp, _ := resty.R().Options(baseURL + constants.FullUserPrefs)
|
|
So(resp, ShouldNotBeNil)
|
|
So(resp.Header().Get("Access-Control-Allow-Methods"), ShouldResemble, "PUT,OPTIONS")
|
|
So(resp.StatusCode(), ShouldEqual, http.StatusNoContent)
|
|
})
|
|
}
|
|
|
|
func TestHandlers(t *testing.T) {
|
|
const UserprefsBaseURL = "http://127.0.0.1:8080/v2/_zot/ext/userprefs"
|
|
|
|
log := log.NewLogger("debug", "")
|
|
mockmetaDB := mocks.MetaDBMock{}
|
|
|
|
Convey("No repo in request", t, func() {
|
|
request := httptest.NewRequest(http.MethodGet, UserprefsBaseURL+"", strings.NewReader("My string"))
|
|
response := httptest.NewRecorder()
|
|
|
|
extensions.PutStar(response, request, mockmetaDB, log)
|
|
res := response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusBadRequest)
|
|
defer res.Body.Close()
|
|
|
|
extensions.PutBookmark(response, request, mockmetaDB, log)
|
|
res = response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusBadRequest)
|
|
defer res.Body.Close()
|
|
})
|
|
|
|
Convey("Empty repo in request", t, func() {
|
|
request := httptest.NewRequest(http.MethodGet, UserprefsBaseURL+"?repo=", strings.NewReader("My string"))
|
|
response := httptest.NewRecorder()
|
|
|
|
extensions.PutStar(response, request, mockmetaDB, log)
|
|
res := response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
defer res.Body.Close()
|
|
|
|
extensions.PutBookmark(response, request, mockmetaDB, log)
|
|
res = response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
defer res.Body.Close()
|
|
})
|
|
|
|
Convey("ToggleStarRepo different errors", t, func() {
|
|
request := httptest.NewRequest(http.MethodGet, UserprefsBaseURL+"?repo=test",
|
|
strings.NewReader("My string"))
|
|
|
|
Convey("ErrRepoMetaNotFound", func() {
|
|
mockmetaDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (mTypes.ToggleState, error) {
|
|
return mTypes.NotChanged, zerr.ErrRepoMetaNotFound
|
|
}
|
|
|
|
mockmetaDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (mTypes.ToggleState, error) {
|
|
return mTypes.NotChanged, zerr.ErrRepoMetaNotFound
|
|
}
|
|
|
|
response := httptest.NewRecorder()
|
|
extensions.PutBookmark(response, request, mockmetaDB, log)
|
|
res := response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
defer res.Body.Close()
|
|
|
|
response = httptest.NewRecorder()
|
|
extensions.PutStar(response, request, mockmetaDB, log)
|
|
res = response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
defer res.Body.Close()
|
|
})
|
|
|
|
Convey("ErrUserDataNotAllowed", func() {
|
|
request = mux.SetURLVars(request, map[string]string{
|
|
"name": "repo",
|
|
})
|
|
|
|
mockmetaDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (mTypes.ToggleState, error) {
|
|
return mTypes.NotChanged, zerr.ErrUserDataNotAllowed
|
|
}
|
|
|
|
mockmetaDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (mTypes.ToggleState, error) {
|
|
return mTypes.NotChanged, zerr.ErrUserDataNotAllowed
|
|
}
|
|
|
|
response := httptest.NewRecorder()
|
|
extensions.PutBookmark(response, request, mockmetaDB, log)
|
|
res := response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusForbidden)
|
|
defer res.Body.Close()
|
|
|
|
response = httptest.NewRecorder()
|
|
extensions.PutStar(response, request, mockmetaDB, log)
|
|
res = response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusForbidden)
|
|
defer res.Body.Close()
|
|
})
|
|
|
|
Convey("ErrUnexpectedError", func() {
|
|
request = mux.SetURLVars(request, map[string]string{
|
|
"name": "repo",
|
|
})
|
|
|
|
mockmetaDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (mTypes.ToggleState, error) {
|
|
return mTypes.NotChanged, ErrTestError
|
|
}
|
|
|
|
mockmetaDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (mTypes.ToggleState, error) {
|
|
return mTypes.NotChanged, ErrTestError
|
|
}
|
|
response := httptest.NewRecorder()
|
|
extensions.PutBookmark(response, request, mockmetaDB, log)
|
|
res := response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusInternalServerError)
|
|
defer res.Body.Close()
|
|
|
|
response = httptest.NewRecorder()
|
|
extensions.PutStar(response, request, mockmetaDB, log)
|
|
res = response.Result()
|
|
So(res.StatusCode, ShouldEqual, http.StatusInternalServerError)
|
|
defer res.Body.Close()
|
|
})
|
|
})
|
|
}
|