portainer/api/internal/testhelpers/request_bouncer.go

package testhelpers

import (
	"net/http"

	portainer "github.com/portainer/portainer/api"
	"github.com/portainer/portainer/api/dataservices"
)

type testRequestBouncer struct{}

// NewTestRequestBouncer creates new mock for requestBouncer
func NewTestRequestBouncer() *testRequestBouncer {
	return &testRequestBouncer{}
}

func (testRequestBouncer) AuthenticatedAccess(h http.Handler) http.Handler {
	return h
}

func (testRequestBouncer) AdminAccess(h http.Handler) http.Handler {
	return h
}

func (testRequestBouncer) RestrictedAccess(h http.Handler) http.Handler {
	return h
}

func (testRequestBouncer) PublicAccess(h http.Handler) http.Handler {
	return h
}

func (testRequestBouncer) TeamLeaderAccess(h http.Handler) http.Handler {
	return h
}

func (testRequestBouncer) EdgeComputeOperation(h http.Handler) http.Handler {
	return h
}

func (testRequestBouncer) AuthorizedEndpointOperation(r *http.Request, endpoint *portainer.Endpoint) error {
	return nil
}

func (testRequestBouncer) AuthorizedEdgeEndpointOperation(r *http.Request, endpoint *portainer.Endpoint) error {
	return nil
}

func (testRequestBouncer) TrustedEdgeEnvironmentAccess(tx dataservices.DataStoreTx, endpoint *portainer.Endpoint) error {
	return nil
}

func (testRequestBouncer) CookieAuthLookup(r *http.Request) (*portainer.TokenData, error) {
	return nil, nil
}

func (testRequestBouncer) JWTAuthLookup(r *http.Request) (*portainer.TokenData, error) {
	return nil, nil
}

func (testRequestBouncer) RevokeJWT(jti string) {}

// AddTestSecurityCookie adds a security cookie to the request
func AddTestSecurityCookie(r *http.Request, jwt string) {
	r.AddCookie(&http.Cookie{
		Name:  portainer.AuthCookieKey,
		Value: jwt,
	})
}