b91f77a554
Co-authored-by: Chaim Lev-Ari <chiptus@users.noreply.github.com> Co-authored-by: Chaim Lev-Ari <chaim.lev-ari@portainer.io>
43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package workflows
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
gittypes "github.com/portainer/portainer/api/git/types"
|
|
ce "github.com/portainer/portainer/api/gitops/workflows"
|
|
"github.com/portainer/portainer/api/http/security"
|
|
|
|
"github.com/segmentio/encoding/json"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// buildWorkflowsReq creates an HTTP GET request with security context pre-populated.
|
|
func buildWorkflowsReq(t *testing.T, userID portainer.UserID, role portainer.UserRole, query string) *http.Request {
|
|
t.Helper()
|
|
req := httptest.NewRequest(http.MethodGet, "/gitops/workflows?"+query, nil)
|
|
ctx := security.StoreTokenData(req, &portainer.TokenData{ID: userID})
|
|
req = req.WithContext(ctx)
|
|
ctx = security.StoreRestrictedRequestContext(req, &security.RestrictedRequestContext{
|
|
UserID: userID,
|
|
IsAdmin: security.IsAdminRole(role),
|
|
})
|
|
return req.WithContext(ctx)
|
|
}
|
|
|
|
// decodeWorkflows decodes a 200 JSON response into a slice of ce.Workflow.
|
|
func decodeWorkflows(t *testing.T, rr *httptest.ResponseRecorder) []ce.Workflow {
|
|
t.Helper()
|
|
require.Equal(t, http.StatusOK, rr.Code, "unexpected status: %s", rr.Body.String())
|
|
var items []ce.Workflow
|
|
require.NoError(t, json.NewDecoder(rr.Body).Decode(&items))
|
|
return items
|
|
}
|
|
|
|
// gitConfig is a convenience constructor for test RepoConfigs.
|
|
func gitConfig(url string) *gittypes.RepoConfig {
|
|
return &gittypes.RepoConfig{URL: url, ConfigFilePath: "docker-compose.yml"}
|
|
}
|