2023-04-24 13:13:15 -05:00
|
|
|
//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"
|
2023-05-10 12:09:53 -05:00
|
|
|
"gopkg.in/resty.v1"
|
2023-04-24 13:13:15 -05:00
|
|
|
|
|
|
|
zerr "zotregistry.io/zot/errors"
|
2023-05-10 12:09:53 -05:00
|
|
|
"zotregistry.io/zot/pkg/api"
|
|
|
|
"zotregistry.io/zot/pkg/api/config"
|
|
|
|
"zotregistry.io/zot/pkg/api/constants"
|
2023-04-24 13:13:15 -05:00
|
|
|
"zotregistry.io/zot/pkg/extensions"
|
2023-05-10 12:09:53 -05:00
|
|
|
extconf "zotregistry.io/zot/pkg/extensions/config"
|
2023-04-24 13:13:15 -05:00
|
|
|
"zotregistry.io/zot/pkg/log"
|
|
|
|
"zotregistry.io/zot/pkg/meta/repodb"
|
2023-05-10 12:09:53 -05:00
|
|
|
"zotregistry.io/zot/pkg/test"
|
2023-04-24 13:13:15 -05:00
|
|
|
"zotregistry.io/zot/pkg/test/mocks"
|
|
|
|
)
|
|
|
|
|
|
|
|
var ErrTestError = errors.New("TestError")
|
|
|
|
|
2023-05-10 12:09:53 -05:00
|
|
|
func TestAllowedMethodsHeader(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{
|
|
|
|
Search: &extconf.SearchConfig{
|
|
|
|
BaseConfig: extconf.BaseConfig{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.FullUserPreferencesPrefix)
|
|
|
|
So(resp, ShouldNotBeNil)
|
|
|
|
So(resp.Header().Get("Access-Control-Allow-Methods"), ShouldResemble, "HEAD,GET,POST,PUT,OPTIONS")
|
|
|
|
So(resp.StatusCode(), ShouldEqual, http.StatusNoContent)
|
|
|
|
})
|
|
|
|
}
|
2023-04-24 13:13:15 -05:00
|
|
|
|
|
|
|
func TestHandlers(t *testing.T) {
|
2023-05-10 12:09:53 -05:00
|
|
|
const UserprefsBaseURL = "http://127.0.0.1:8080/v2/_zot/ext/userprefs"
|
|
|
|
|
2023-04-24 13:13:15 -05:00
|
|
|
log := log.NewLogger("debug", "")
|
|
|
|
mockrepoDB := mocks.RepoDBMock{}
|
|
|
|
|
|
|
|
Convey("No repo in request", t, func() {
|
2023-04-27 02:09:46 -05:00
|
|
|
request := httptest.NewRequest(http.MethodGet, UserprefsBaseURL+"", strings.NewReader("My string"))
|
2023-04-24 13:13:15 -05:00
|
|
|
response := httptest.NewRecorder()
|
|
|
|
|
|
|
|
extensions.PutStar(response, request, mockrepoDB, log)
|
|
|
|
res := response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusBadRequest)
|
|
|
|
defer res.Body.Close()
|
|
|
|
|
|
|
|
extensions.PutBookmark(response, request, mockrepoDB, log)
|
|
|
|
res = response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusBadRequest)
|
|
|
|
defer res.Body.Close()
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("Empty repo in request", t, func() {
|
2023-04-27 02:09:46 -05:00
|
|
|
request := httptest.NewRequest(http.MethodGet, UserprefsBaseURL+"?repo=", strings.NewReader("My string"))
|
2023-04-24 13:13:15 -05:00
|
|
|
response := httptest.NewRecorder()
|
|
|
|
|
|
|
|
extensions.PutStar(response, request, mockrepoDB, log)
|
|
|
|
res := response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
|
|
defer res.Body.Close()
|
|
|
|
|
|
|
|
extensions.PutBookmark(response, request, mockrepoDB, log)
|
|
|
|
res = response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
|
|
defer res.Body.Close()
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("ToggleStarRepo different errors", t, func() {
|
2023-04-27 02:09:46 -05:00
|
|
|
request := httptest.NewRequest(http.MethodGet, UserprefsBaseURL+"?repo=test",
|
2023-04-24 13:13:15 -05:00
|
|
|
strings.NewReader("My string"))
|
|
|
|
|
|
|
|
Convey("ErrRepoMetaNotFound", func() {
|
|
|
|
mockrepoDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (repodb.ToggleState, error) {
|
|
|
|
return repodb.NotChanged, zerr.ErrRepoMetaNotFound
|
|
|
|
}
|
|
|
|
|
|
|
|
mockrepoDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (repodb.ToggleState, error) {
|
|
|
|
return repodb.NotChanged, zerr.ErrRepoMetaNotFound
|
|
|
|
}
|
|
|
|
|
|
|
|
response := httptest.NewRecorder()
|
|
|
|
extensions.PutBookmark(response, request, mockrepoDB, log)
|
|
|
|
res := response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
|
|
defer res.Body.Close()
|
|
|
|
|
|
|
|
response = httptest.NewRecorder()
|
|
|
|
extensions.PutStar(response, request, mockrepoDB, 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",
|
|
|
|
})
|
|
|
|
|
|
|
|
mockrepoDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (repodb.ToggleState, error) {
|
|
|
|
return repodb.NotChanged, zerr.ErrUserDataNotAllowed
|
|
|
|
}
|
|
|
|
|
|
|
|
mockrepoDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (repodb.ToggleState, error) {
|
|
|
|
return repodb.NotChanged, zerr.ErrUserDataNotAllowed
|
|
|
|
}
|
|
|
|
|
|
|
|
response := httptest.NewRecorder()
|
|
|
|
extensions.PutBookmark(response, request, mockrepoDB, log)
|
|
|
|
res := response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusForbidden)
|
|
|
|
defer res.Body.Close()
|
|
|
|
|
|
|
|
response = httptest.NewRecorder()
|
|
|
|
extensions.PutStar(response, request, mockrepoDB, 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",
|
|
|
|
})
|
|
|
|
|
|
|
|
mockrepoDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (repodb.ToggleState, error) {
|
|
|
|
return repodb.NotChanged, ErrTestError
|
|
|
|
}
|
|
|
|
|
|
|
|
mockrepoDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (repodb.ToggleState, error) {
|
|
|
|
return repodb.NotChanged, ErrTestError
|
|
|
|
}
|
|
|
|
response := httptest.NewRecorder()
|
|
|
|
extensions.PutBookmark(response, request, mockrepoDB, log)
|
|
|
|
res := response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusInternalServerError)
|
|
|
|
defer res.Body.Close()
|
|
|
|
|
|
|
|
response = httptest.NewRecorder()
|
|
|
|
extensions.PutStar(response, request, mockrepoDB, log)
|
|
|
|
res = response.Result()
|
|
|
|
So(res.StatusCode, ShouldEqual, http.StatusInternalServerError)
|
|
|
|
defer res.Body.Close()
|
|
|
|
})
|
|
|
|
})
|
|
|
|
}
|