feat: rerun client method

Signed-off-by: Vladislav Sukhin <vladislav@kubeshop.io>
kubeshop · Feb 12, 2025 · 593fc3b · 593fc3b
1 parent 0e798b8
commit 593fc3b
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 3 deletions.
diff --git a/internal/app/api/v1/server.go b/internal/app/api/v1/server.go
@@ -163,7 +163,7 @@ func (s *TestkubeAPI) Init(server server.HTTPServer) {
 	testWorkflows.Post("/:id/executions/:executionID/pause", s.PauseTestWorkflowExecutionHandler())
 	testWorkflows.Post("/:id/executions/:executionID/resume", s.ResumeTestWorkflowExecutionHandler())
 	testWorkflows.Get("/:id/executions/:executionID/logs", s.GetTestWorkflowExecutionLogsHandler())
-	testWorkflows.Post("/:id/executions/:executionID/rerun", s.ReRunTestWorkflowHandler())
+	testWorkflows.Post("/:id/executions/:executionID/rerun", s.ReRunTestWorkflowExecutionHandler())
 
 	testWorkflowExecutions := root.Group("/test-workflow-executions")
 	testWorkflowExecutions.Get("/", s.ListTestWorkflowExecutionsHandler())
@@ -182,7 +182,7 @@ func (s *TestkubeAPI) Init(server server.HTTPServer) {
 	testWorkflowExecutions.Get("/:executionID/artifacts", s.ListTestWorkflowExecutionArtifactsHandler())
 	testWorkflowExecutions.Get("/:executionID/artifacts/:filename", s.GetTestWorkflowArtifactHandler())
 	testWorkflowExecutions.Get("/:executionID/artifact-archive", s.GetTestWorkflowArtifactArchiveHandler())
-	testWorkflowExecutions.Post("/:executionID/rerun", s.ReRunTestWorkflowHandler())
+	testWorkflowExecutions.Post("/:executionID/rerun", s.ReRunTestWorkflowExecutionHandler())
 
 	testWorkflowWithExecutions := root.Group("/test-workflow-with-executions")
 	testWorkflowWithExecutions.Get("/", s.ListTestWorkflowWithExecutionsHandler())

diff --git a/internal/app/api/v1/testworkflows.go b/internal/app/api/v1/testworkflows.go
@@ -451,7 +451,7 @@ func (s *TestkubeAPI) ExecuteTestWorkflowHandler() fiber.Handler {
 }
 
 // TODO: Add metrics
-func (s *TestkubeAPI) ReRunTestWorkflowHandler() fiber.Handler {
+func (s *TestkubeAPI) ReRunTestWorkflowExecutionHandler() fiber.Handler {
 	return func(c *fiber.Ctx) (err error) {
 		ctx := c.Context()
 		executionID := c.Params("executionID")

diff --git a/pkg/api/v1/client/interface.go b/pkg/api/v1/client/interface.go
@@ -183,6 +183,7 @@ type TestWorkflowExecutionAPI interface {
 	GetTestWorkflowExecutionArtifacts(executionID string) (artifacts testkube.Artifacts, err error)
 	DownloadTestWorkflowArtifact(executionID, fileName, destination string) (artifact string, err error)
 	DownloadTestWorkflowArtifactArchive(executionID, destination string, masks []string) (archive string, err error)
+	ReRunTestWorkflowExecution(workflow string, id string, runningContext testkube.TestWorkflowRunningContext) (testkube.TestWorkflowExecution, error)
 }
 
 // TestWorkflowTemplateAPI describes test workflow api methods

diff --git a/pkg/api/v1/client/testworkflow.go b/pkg/api/v1/client/testworkflow.go
@@ -219,3 +219,19 @@ func (c TestWorkflowClient) GetTestWorkflowExecutionLogs(id string) (result []by
 	uri := c.testWorkflowTransport.GetURI("/test-workflow-executions/%s/logs", id)
 	return c.testWorkflowTransport.GetRawBody(http.MethodGet, uri, nil, nil)
 }
+
+// ReRunTestWorkflowExecution reruns selected execution
+func (c TestWorkflowClient) ReRunTestWorkflowExecution(workflow, id string, runningContext testkube.TestWorkflowRunningContext) (result testkube.TestWorkflowExecution, err error) {
+	if workflow == "" {
+		return result, fmt.Errorf("test workflow name '%s' is not valid", workflow)
+	}
+
+	uri := c.testWorkflowTransport.GetURI("/test-workflows/%s/executions/%s/rerun", workflow, id)
+
+	body, err := json.Marshal(runningContext)
+	if err != nil {
+		return result, err
+	}
+
+	return c.testWorkflowExecutionTransport.Execute(http.MethodPost, uri, body, nil)
+}