61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
package workflows
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/dataservices"
|
|
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"}
|
|
}
|
|
|
|
// createGitStack creates Source, Workflow, and Stack records with WorkflowID properly wired.
|
|
func createGitStack(t *testing.T, tx dataservices.DataStoreTx, stack *portainer.Stack) {
|
|
t.Helper()
|
|
|
|
if stack.GitConfig != nil {
|
|
src := &portainer.Source{GitConfig: stack.GitConfig, Type: portainer.SourceTypeGit}
|
|
require.NoError(t, tx.Source().Create(src))
|
|
|
|
wf := &portainer.Workflow{SourceIDs: []portainer.SourceID{src.ID}}
|
|
require.NoError(t, tx.Workflow().Create(wf))
|
|
|
|
stack.WorkflowID = wf.ID
|
|
}
|
|
|
|
require.NoError(t, tx.Stack().Create(stack))
|
|
}
|