mirror of
https://github.com/project-zot/zot.git
synced 2024-12-23 22:27:35 -05:00
49e8167dbe
- AccessControlContext now resides in a separate package from where it can be imported, along with the contextKey that will be used to set and retrieve this context value. - AccessControlContext has a new field called Username, that will be of use for future implementations in graphQL resolvers. - GlobalSearch resolver now uses this context to filter repos available to the logged user. - moved logic for uploading images in tests so that it can be used in every package - tests were added for multiple request scenarios, when zot-server requires authz on specific repos - added tests with injected errors for extended coverage - added tests for status code error injection utilities Closes https://github.com/project-zot/zot/issues/615 Signed-off-by: Alex Stan <alexandrustan96@yahoo.ro>
106 lines
1.5 KiB
Go
106 lines
1.5 KiB
Go
//go:build dev
|
|
// +build dev
|
|
|
|
// This file should be linked only in **development** mode.
|
|
|
|
package test
|
|
|
|
import (
|
|
"net/http"
|
|
"sync"
|
|
|
|
zerr "zotregistry.io/zot/errors"
|
|
"zotregistry.io/zot/pkg/log"
|
|
)
|
|
|
|
func Ok(ok bool) bool {
|
|
if !ok {
|
|
return ok
|
|
}
|
|
|
|
if injectedFailure() {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func Error(err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if injectedFailure() {
|
|
return zerr.ErrInjected
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Used to inject error status codes for coverage purposes.
|
|
// -1 will be returned in case of successful failure injection.
|
|
func ErrStatusCode(status int) int {
|
|
if !injectedFailure() {
|
|
if status == http.StatusAccepted || status == http.StatusCreated {
|
|
return status
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
return -1
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Failure injection infrastructure to cover hard-to-reach code paths.
|
|
*
|
|
**/
|
|
|
|
type inject struct {
|
|
skip int
|
|
}
|
|
|
|
//nolint:gochecknoglobals // only used by test code
|
|
var injMap sync.Map
|
|
|
|
func InjectFailure(skip int) bool {
|
|
gid := log.GoroutineID()
|
|
if gid < 0 {
|
|
panic("invalid goroutine id")
|
|
}
|
|
|
|
if _, ok := injMap.Load(gid); ok {
|
|
panic("prior incomplete fault injection")
|
|
}
|
|
|
|
injst := inject{skip: skip}
|
|
injMap.Store(gid, injst)
|
|
|
|
return true
|
|
}
|
|
|
|
func injectedFailure() bool {
|
|
gid := log.GoroutineID()
|
|
|
|
val, ok := injMap.Load(gid)
|
|
if !ok {
|
|
return false
|
|
}
|
|
|
|
injst, ok := val.(inject)
|
|
if !ok {
|
|
panic("invalid type")
|
|
}
|
|
|
|
if injst.skip == 0 {
|
|
injMap.Delete(gid)
|
|
|
|
return true
|
|
}
|
|
|
|
injst.skip--
|
|
injMap.Store(gid, injst)
|
|
|
|
return false
|
|
}
|