fixL move mapper to tcl

Signed-off-by: Vladislav Sukhin <vladislav@kubeshop.io>
kubeshop · Oct 16, 2024 · 9dc607f · 9dc607f
1 parent c3a077a
commit 9dc607f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/pkg/mapper/cdevents/mapper.go b/pkg/mapper/cdevents/mapper.go
@@ -9,6 +9,7 @@ import (
 	cdevents "github.com/cdevents/sdk-go/pkg/api"
 
 	"github.com/kubeshop/testkube/pkg/api/v1/testkube"
+	"github.com/kubeshop/testkube/pkg/tcl/testworkflowstcl/mapper"
 )
 
 // MapTestkubeEventToCDEvent maps OpenAPI spec Event to CDEvent CDEventReader
@@ -526,7 +527,8 @@ func MapTestkubeEventQueuedTestWorkflowTestToCDEvent(event testkube.Event, clust
 		if event.TestWorkflowExecution.RunningContext != nil && event.TestWorkflowExecution.RunningContext.Actor != nil &&
 			event.TestWorkflowExecution.RunningContext.Actor.Type_ != nil {
 			ev.SetSubjectTrigger(&cdevents.TestCaseRunQueuedSubjectContentTrigger{
-				Type: MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
+				// Pro edition only (tcl protected code)
+				Type: mapper.MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
 			})
 		}
 	}
@@ -572,7 +574,8 @@ func MapTestkubeEventQueuedTestWorkflowTestSuiteToCDEvent(event testkube.Event,
 		if event.TestWorkflowExecution.RunningContext != nil && event.TestWorkflowExecution.RunningContext.Actor != nil &&
 			event.TestWorkflowExecution.RunningContext.Actor.Type_ != nil {
 			ev.SetSubjectTrigger(&cdevents.TestSuiteRunQueuedSubjectContentTrigger{
-				Type: MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
+				// Pro edition only (tcl protected code)
+				Type: mapper.MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
 			})
 		}
 	}
@@ -619,7 +622,8 @@ func MapTestkubeEventStartTestWorkflowTestToCDEvent(event testkube.Event, cluste
 		if event.TestWorkflowExecution.RunningContext != nil && event.TestWorkflowExecution.RunningContext.Actor != nil &&
 			event.TestWorkflowExecution.RunningContext.Actor.Type_ != nil {
 			ev.SetSubjectTrigger(&cdevents.TestCaseRunStartedSubjectContentTrigger{
-				Type: MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
+				// Pro edition only (tcl protected code)
+				Type: mapper.MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
 			})
 		}
 	}
@@ -665,7 +669,8 @@ func MapTestkubeEventStartTestWorkflowTestSuiteToCDEvent(event testkube.Event, c
 		if event.TestWorkflowExecution.RunningContext != nil && event.TestWorkflowExecution.RunningContext.Actor != nil &&
 			event.TestWorkflowExecution.RunningContext.Actor.Type_ != nil {
 			ev.SetSubjectTrigger(&cdevents.TestSuiteRunStartedSubjectContentTrigger{
-				Type: MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
+				// Pro edition only (tcl protected code)
+				Type: mapper.MapTestkubeTestWorkflowRunningContextActorToCDEventTiggerType(*event.TestWorkflowExecution.RunningContext.Actor.Type_),
 			})
 		}
 	}

diff --git a/pkg/tcl/testworkflowstcl/mapper/cdevents_test.go b/pkg/tcl/testworkflowstcl/mapper/cdevents_test.go
@@ -0,0 +1,14 @@
+package mapper
+
+import (
+	"github.com/kubeshop/testkube/internal/common"
+	"github.com/kubeshop/testkube/pkg/api/v1/testkube"
+)
+
+func NewMockRunningContext() *testkube.TestWorkflowRunningContext {
+	return &testkube.TestWorkflowRunningContext{
+		Actor: &testkube.TestWorkflowRunningContextActor{
+			Type_: common.Ptr(testkube.CRON_TestWorkflowRunningContextActorType),
+		},
+	}
+}