snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/tests/workflows/main_test.go

@@ -0,0 +1,200 @@
+package workflows
+
+import (
+	"context"
+	"fmt"
+
+	// "fmt"
+	"path"
+	"testing"
+
+	"github.com/cortezaproject/corteza/server/app"
+	"github.com/cortezaproject/corteza/server/automation/service"
+	autTypes "github.com/cortezaproject/corteza/server/automation/types"
+	"github.com/cortezaproject/corteza/server/pkg/auth"
+	"github.com/cortezaproject/corteza/server/pkg/dal"
+	"github.com/cortezaproject/corteza/server/pkg/envoy"
+	"github.com/cortezaproject/corteza/server/pkg/envoy/csv"
+	"github.com/cortezaproject/corteza/server/pkg/envoy/directory"
+	"github.com/cortezaproject/corteza/server/pkg/envoy/json"
+	envoyStore "github.com/cortezaproject/corteza/server/pkg/envoy/store"
+	"github.com/cortezaproject/corteza/server/pkg/envoy/yaml"
+	"github.com/cortezaproject/corteza/server/pkg/eventbus"
+	"github.com/cortezaproject/corteza/server/pkg/expr"
+	"github.com/cortezaproject/corteza/server/pkg/id"
+	"github.com/cortezaproject/corteza/server/pkg/logger"
+	"github.com/cortezaproject/corteza/server/store"
+	sysTypes "github.com/cortezaproject/corteza/server/system/types"
+	"github.com/cortezaproject/corteza/server/tests/helpers"
+	"github.com/stretchr/testify/require"
+)
+
+var (
+	defApp   *app.CortezaApp
+	defStore store.Storer
+	defDal   dal.FullService
+	eventBus = eventbus.New()
+)
+
+func init() {
+	helpers.RecursiveDotEnvLoad()
+}
+
+func TestMain(m *testing.M) {
+	logger.SetDefault(logger.MakeDebugLogger())
+	ctx := context.Background()
+
+	defApp = helpers.NewIntegrationTestApp(ctx, func(app *app.CortezaApp) (err error) {
+		// some test suites require action-log enabled
+		app.Opt.ActionLog.WorkflowFunctionsEnabled = true
+
+		// trace assertions in this suite expect every executed step to be
+		// recorded; the default only keeps the last iteration of each loop
+		app.Opt.Workflow.StackTraceFull = true
+		defStore = app.Store
+		eventbus.Set(eventBus)
+
+		return nil
+	})
+
+	defDal = dal.Service()
+
+	if err := defApp.Activate(ctx); err != nil {
+		panic(fmt.Errorf("could not activate corteza: %v", err))
+	}
+
+	if err := defApp.InitExpr(ctx); err != nil {
+		panic(fmt.Errorf("could not init expressions: %v", err))
+	}
+
+	m.Run()
+}
+
+func cleanup(t *testing.T) {
+	var (
+		ctx = context.Background()
+	)
+
+	if err := defStore.TruncateAutomationWorkflows(ctx); err != nil {
+		t.Fatalf("failed to decode scenario data: %v", err)
+	}
+}
+
+func truncateRecords(ctx context.Context) error {
+	models, err := defDal.SearchModels(ctx)
+	if err != nil {
+		return err
+	}
+	for _, model := range models {
+		err = defDal.Truncate(ctx, model.ToFilter(), nil)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func loadScenario(ctx context.Context, t *testing.T) {
+	loadScenarioWithName(ctx, t, "S"+t.Name()[4:])
+}
+
+// 1st step in migration to workflow testdata w/o number prefix
+//
+// When all old scenarios are renamed, replace it with loadScenario.
+func loadNewScenario(ctx context.Context, t *testing.T) {
+	loadScenarioWithName(ctx, t, t.Name()[5:])
+}
+
+func loadScenarioWithName(ctx context.Context, t *testing.T, scenario string) {
+	var (
+		err error
+	)
+
+	cleanup(t)
+
+	decoded, err := directory.Decode(
+		ctx,
+		path.Join("testdata", scenario),
+		yaml.Decoder(),
+		csv.Decoder(),
+		json.Decoder(),
+	)
+	if err != nil {
+		t.Fatalf("failed to decode scenario data: %v", err)
+	}
+
+	storeEnc := envoyStore.NewStoreEncoder(defStore, dal.Service(), &envoyStore.EncoderConfig{})
+
+	b := envoy.NewBuilder(storeEnc)
+	g, err := b.Build(ctx, decoded...)
+	if err != nil {
+		t.Fatalf("failed to build structure graph: %v", err)
+	}
+
+	if err = envoy.Encode(ctx, g, storeEnc); err != nil {
+		t.Fatalf("failed to build structure graph: %v", err)
+	}
+
+	// Reload and register workflows
+	if err = service.DefaultWorkflow.Load(ctx); err != nil {
+		t.Fatalf("failed to reload workflows: %v", err)
+	}
+}
+
+func bypassRBAC(ctx context.Context) context.Context {
+	u := &sysTypes.User{
+		ID: id.Next(),
+	}
+
+	if err := defStore.CreateUser(ctx, u); err != nil {
+		panic(err)
+	}
+
+	u.SetRoles(auth.BypassRoles().IDs()...)
+	return auth.SetIdentityToContext(ctx, u)
+}
+
+func execWorkflow(ctx context.Context, name string, p autTypes.WorkflowExecParams) (*expr.Vars, uint64, autTypes.Stacktrace, error) {
+	wf, err := defStore.LookupAutomationWorkflowByHandle(ctx, name)
+	if err != nil {
+		return nil, 0, nil, err
+	}
+
+	return service.DefaultWorkflow.Exec(ctx, wf.ID, p)
+}
+
+func mustExecWorkflow(ctx context.Context, t *testing.T, name string, p autTypes.WorkflowExecParams) (vars *expr.Vars, strace autTypes.Stacktrace) {
+	var err error
+	vars, _, strace, err = execWorkflow(ctx, name, p)
+	if err != nil {
+		if issues, is := err.(autTypes.WorkflowIssueSet); is {
+			for _, i := range issues {
+				t.Logf("issue: %s", i.Description)
+				t.Logf("       %v", i.Culprit)
+			}
+		}
+
+		t.Fatalf("could not exec %q: %v", name, err)
+
+	}
+
+	return
+}
+
+func addRoleMember(ctx context.Context, req *require.Assertions, r string, uu ...string) {
+	role, err := store.LookupRoleByHandle(ctx, defStore, r)
+	req.NoError(err)
+
+	rr := make([]*sysTypes.RoleMember, len(uu))
+	for i, u := range uu {
+		usr, err := store.LookupUserByHandle(ctx, defStore, u)
+		req.NoError(err)
+
+		rr[i] = &sysTypes.RoleMember{
+			RoleID:   role.ID,
+			Resource: fmt.Sprintf("corteza::system:user/%d", usr.ID),
+		}
+	}
+
+	req.NoError(store.CreateRoleMember(ctx, defStore, rr...))
+}