fix: add tcl for testworkflow mapper

Signed-off-by: Vladislav Sukhin <[email protected]>
kubeshop · Oct 8, 2024 · 3269f3d · 3269f3d
1 parent d972b1a
commit 3269f3d
Show file tree

Hide file tree

Showing 4 changed files with 61 additions and 84 deletions.
diff --git a/pkg/api/v1/testkube/model_test_workflow_running_context_caller.go b/pkg/api/v1/testkube/model_test_workflow_running_context_caller.go
diff --git a/pkg/api/v1/testkube/model_test_workflow_running_context_caller_resource_type.go b/pkg/api/v1/testkube/model_test_workflow_running_context_caller_resource_type.go
diff --git a/pkg/mapper/testworkflows/openapi_kube.go b/pkg/mapper/testworkflows/openapi_kube.go
@@ -13,6 +13,7 @@ import (
 	testworkflowsv1 "github.com/kubeshop/testkube-operator/api/testworkflows/v1"
 	"github.com/kubeshop/testkube/internal/common"
 	"github.com/kubeshop/testkube/pkg/api/v1/testkube"
+	mappertcl "github.com/kubeshop/testkube/pkg/tcl/mappertcl/testworkflows"
 )
 
 func MapStringToIntOrString(i string) intstr.IntOrString {
@@ -1380,39 +1381,6 @@ func MapTestWorkflowSignatureAPIToKube(v testkube.TestWorkflowSignature) testwor
 	}
 }
 
-func MapTestWorkflowRunningContextInterfaceTypeAPIToKube(v testkube.TestWorkflowRunningContextInterfaceType) testworkflowsv1.TestWorkflowRunningContextInterfaceType {
-	return testworkflowsv1.TestWorkflowRunningContextInterfaceType(v)
-}
-
-func MapTestWorkflowRunningContextInterfaceAPIToKube(v testkube.TestWorkflowRunningContextInterface) testworkflowsv1.TestWorkflowRunningContextInterface {
-	return testworkflowsv1.TestWorkflowRunningContextInterface{
-		Name:  v.Name,
-		Type_: common.MapPtr(v.Type_, MapTestWorkflowRunningContextInterfaceTypeAPIToKube),
-	}
-}
-
-func MapTestWorkflowRunningContextActorTypeAPIToKube(v testkube.TestWorkflowRunningContextActorType) testworkflowsv1.TestWorkflowRunningContextActorType {
-	return testworkflowsv1.TestWorkflowRunningContextActorType(v)
-}
-
-func MapTestWorkflowRunningContextActorAPIToKube(v testkube.TestWorkflowRunningContextActor) testworkflowsv1.TestWorkflowRunningContextActor {
-	return testworkflowsv1.TestWorkflowRunningContextActor{
-		Name:          v.Name,
-		Username:      v.Username,
-		Email:         v.Email,
-		ExecutionId:   v.ExecutionId,
-		ExecutionPath: v.ExecutionPath,
-		Type_:         common.MapPtr(v.Type_, MapTestWorkflowRunningContextActorTypeAPIToKube),
-	}
-}
-
-func MapTestWorkflowRunningContextAPIToKube(v testkube.TestWorkflowRunningContext) testworkflowsv1.TestWorkflowRunningContext {
-	return testworkflowsv1.TestWorkflowRunningContext{
-		Interface_: common.MapPtr(v.Interface_, MapTestWorkflowRunningContextInterfaceAPIToKube),
-		Actor:      common.MapPtr(v.Actor, MapTestWorkflowRunningContextActorAPIToKube),
-	}
-}
-
 func MapTestWorkflowExecutionAPIToKube(v *testkube.TestWorkflowExecution) *testworkflowsv1.TestWorkflowExecutionDetails {
 	return &testworkflowsv1.TestWorkflowExecutionDetails{
 		Id:                        v.Id,
@@ -1430,7 +1398,8 @@ func MapTestWorkflowExecutionAPIToKube(v *testkube.TestWorkflowExecution) *testw
 		TestWorkflowExecutionName: v.TestWorkflowExecutionName,
 		DisableWebhooks:           v.DisableWebhooks,
 		Tags:                      v.Tags,
-		RunningContext:            common.MapPtr(v.RunningContext, MapTestWorkflowRunningContextAPIToKube),
+		// Pro edition only (tcl protected code)
+		RunningContext: common.MapPtr(v.RunningContext, mappertcl.MapTestWorkflowRunningContextAPIToKube),
 	}
 }
 
@@ -1444,15 +1413,16 @@ func MapTestWorkflowExecutionStatusAPIToKube(v *testkube.TestWorkflowExecution,
 func MapTestWorkflowExecutionAPIToKubeTestWorkflowStatusSummary(v *testkube.TestWorkflowExecution) testworkflowsv1.TestWorkflowStatusSummary {
 	return testworkflowsv1.TestWorkflowStatusSummary{
 		LatestExecution: &testworkflowsv1.TestWorkflowExecutionSummary{
-			Id:             v.Id,
-			Name:           v.Name,
-			Number:         v.Number,
-			ScheduledAt:    metav1.NewTime(v.ScheduledAt),
-			StatusAt:       metav1.NewTime(v.StatusAt),
-			Result:         common.MapPtr(v.Result, MapTestWorkflowResultAPIToKubeTestWorkflowResultSummary),
-			Workflow:       common.MapPtr(v.Workflow, MapTestWorkflowAPIToKubeTestWorkflowSummary),
-			Tags:           v.Tags,
-			RunningContext: common.MapPtr(v.RunningContext, MapTestWorkflowRunningContextAPIToKube),
+			Id:          v.Id,
+			Name:        v.Name,
+			Number:      v.Number,
+			ScheduledAt: metav1.NewTime(v.ScheduledAt),
+			StatusAt:    metav1.NewTime(v.StatusAt),
+			Result:      common.MapPtr(v.Result, MapTestWorkflowResultAPIToKubeTestWorkflowResultSummary),
+			Workflow:    common.MapPtr(v.Workflow, MapTestWorkflowAPIToKubeTestWorkflowSummary),
+			Tags:        v.Tags,
+			// Pro edition only (tcl protected code)
+			RunningContext: common.MapPtr(v.RunningContext, mappertcl.MapTestWorkflowRunningContextAPIToKube),
 		},
 	}
 }

diff --git a/pkg/tcl/mappertcl/testworkflows/openapi_kube.go b/pkg/tcl/mappertcl/testworkflows/openapi_kube.go
@@ -0,0 +1,48 @@
+// Copyright 2024 Testkube.
+//
+// Licensed as a Testkube Pro file under the Testkube Community
+// License (the "License"); you may not use this file except in compliance with
+// the License. You may obtain a copy of the License at
+//
+//     https://github.com/kubeshop/testkube/blob/main/licenses/TCL.txt
+
+package testworkflows
+
+import (
+	testworkflowsv1 "github.com/kubeshop/testkube-operator/api/testworkflows/v1"
+	"github.com/kubeshop/testkube/internal/common"
+	"github.com/kubeshop/testkube/pkg/api/v1/testkube"
+)
+
+func MapTestWorkflowRunningContextInterfaceTypeAPIToKube(v testkube.TestWorkflowRunningContextInterfaceType) testworkflowsv1.TestWorkflowRunningContextInterfaceType {
+	return testworkflowsv1.TestWorkflowRunningContextInterfaceType(v)
+}
+
+func MapTestWorkflowRunningContextInterfaceAPIToKube(v testkube.TestWorkflowRunningContextInterface) testworkflowsv1.TestWorkflowRunningContextInterface {
+	return testworkflowsv1.TestWorkflowRunningContextInterface{
+		Name:  v.Name,
+		Type_: common.MapPtr(v.Type_, MapTestWorkflowRunningContextInterfaceTypeAPIToKube),
+	}
+}
+
+func MapTestWorkflowRunningContextActorTypeAPIToKube(v testkube.TestWorkflowRunningContextActorType) testworkflowsv1.TestWorkflowRunningContextActorType {
+	return testworkflowsv1.TestWorkflowRunningContextActorType(v)
+}
+
+func MapTestWorkflowRunningContextActorAPIToKube(v testkube.TestWorkflowRunningContextActor) testworkflowsv1.TestWorkflowRunningContextActor {
+	return testworkflowsv1.TestWorkflowRunningContextActor{
+		Name:          v.Name,
+		Username:      v.Username,
+		Email:         v.Email,
+		ExecutionId:   v.ExecutionId,
+		ExecutionPath: v.ExecutionPath,
+		Type_:         common.MapPtr(v.Type_, MapTestWorkflowRunningContextActorTypeAPIToKube),
+	}
+}
+
+func MapTestWorkflowRunningContextAPIToKube(v testkube.TestWorkflowRunningContext) testworkflowsv1.TestWorkflowRunningContext {
+	return testworkflowsv1.TestWorkflowRunningContext{
+		Interface_: common.MapPtr(v.Interface_, MapTestWorkflowRunningContextInterfaceAPIToKube),
+		Actor:      common.MapPtr(v.Actor, MapTestWorkflowRunningContextActorAPIToKube),
+	}
+}