From ea1cfd256a2a4549dfff53c7d3aaf6ea02db07eb Mon Sep 17 00:00:00 2001
From: Leon <leon@apecloud.com>
Date: Thu, 28 Mar 2024 12:25:26 +0800
Subject: [PATCH] chore: refine data dump/load actions (#6820)

---
 .../v1alpha1/componentdefinition_types.go     | 10 +--
 apis/apps/v1alpha1/zz_generated.deepcopy.go   |  8 +-
 ...ps.kubeblocks.io_componentdefinitions.yaml | 27 +++---
 .../apps/componentdefinition_controller.go    |  4 +-
 ...ps.kubeblocks.io_componentdefinitions.yaml | 27 +++---
 docs/developer_docs/api-reference/cluster.md  | 10 +--
 pkg/constant/lorry.go                         |  2 +
 .../component_definition_convertor.go         |  4 +-
 pkg/controller/component/lorry_utils.go       | 10 +--
 pkg/lorry/client/client.go                    | 10 +++
 pkg/lorry/client/client_mock.go               | 28 +++++++
 pkg/lorry/client/interface.go                 |  3 +
 pkg/lorry/operations/replica/data_dump.go     | 83 +++++++++++++++++++
 pkg/lorry/operations/replica/data_load.go     | 71 ++++++++++++++++
 pkg/lorry/util/types.go                       |  3 +
 pkg/testutil/apps/constant.go                 |  4 +-
 16 files changed, 251 insertions(+), 53 deletions(-)
 create mode 100644 pkg/lorry/operations/replica/data_dump.go
 create mode 100644 pkg/lorry/operations/replica/data_load.go

diff --git a/apis/apps/v1alpha1/componentdefinition_types.go b/apis/apps/v1alpha1/componentdefinition_types.go
index b9276612468..14416c1b3b3 100644
--- a/apis/apps/v1alpha1/componentdefinition_types.go
+++ b/apis/apps/v1alpha1/componentdefinition_types.go
@@ -789,7 +789,7 @@ type ComponentLifecycleActions struct {
 	// +optional
 	Readwrite *LifecycleActionHandler `json:"readwrite,omitempty"`
 
-	// Defines the method to populate the data to create new replicas.
+	// Defines the method to dump the data from a replica.
 	// This action is typically used when a new replica needs to be constructed, such as:
 	//
 	// - scale-out
@@ -800,21 +800,21 @@ type ComponentLifecycleActions struct {
 	// This field cannot be updated.
 	//
 	// +optional
-	DataPopulate *LifecycleActionHandler `json:"dataPopulate,omitempty"`
+	DataDump *LifecycleActionHandler `json:"dataDump,omitempty"`
 
-	// Defines the method to assemble data synchronized from external before starting the service for a new replica.
+	// Defines the method to load data into a replica.
 	// This action is typically used when creating a new replica, such as:
 	//
 	// - scale-out
 	// - rebuild
 	// - clone
 	//
-	// The data will be streamed in via stdin. If any error occurs during the assembly process,
+	// The data will be streamed in via stdin. If any error occurs during the process,
 	// the action must be able to guarantee idempotence to allow for retries from the beginning.
 	// This field cannot be updated.
 	//
 	// +optional
-	DataAssemble *LifecycleActionHandler `json:"dataAssemble,omitempty"`
+	DataLoad *LifecycleActionHandler `json:"dataLoad,omitempty"`
 
 	// Defines the method to notify the replica service that there is a configuration update.
 	// This field cannot be updated.
diff --git a/apis/apps/v1alpha1/zz_generated.deepcopy.go b/apis/apps/v1alpha1/zz_generated.deepcopy.go
index fe11146171f..cfa4336bccf 100644
--- a/apis/apps/v1alpha1/zz_generated.deepcopy.go
+++ b/apis/apps/v1alpha1/zz_generated.deepcopy.go
@@ -1975,13 +1975,13 @@ func (in *ComponentLifecycleActions) DeepCopyInto(out *ComponentLifecycleActions
 		*out = new(LifecycleActionHandler)
 		(*in).DeepCopyInto(*out)
 	}
-	if in.DataPopulate != nil {
-		in, out := &in.DataPopulate, &out.DataPopulate
+	if in.DataDump != nil {
+		in, out := &in.DataDump, &out.DataDump
 		*out = new(LifecycleActionHandler)
 		(*in).DeepCopyInto(*out)
 	}
-	if in.DataAssemble != nil {
-		in, out := &in.DataAssemble, &out.DataAssemble
+	if in.DataLoad != nil {
+		in, out := &in.DataLoad, &out.DataLoad
 		*out = new(LifecycleActionHandler)
 		(*in).DeepCopyInto(*out)
 	}
diff --git a/config/crd/bases/apps.kubeblocks.io_componentdefinitions.yaml b/config/crd/bases/apps.kubeblocks.io_componentdefinitions.yaml
index 031ad6a1718..2635d683d10 100644
--- a/config/crd/bases/apps.kubeblocks.io_componentdefinitions.yaml
+++ b/config/crd/bases/apps.kubeblocks.io_componentdefinitions.yaml
@@ -524,14 +524,12 @@ spec:
                             type: integer
                         type: object
                     type: object
-                  dataAssemble:
-                    description: "Defines the method to assemble data synchronized
-                      from external before starting the service for a new replica.
-                      This action is typically used when creating a new replica, such
-                      as: \n - scale-out - rebuild - clone \n The data will be streamed
-                      in via stdin. If any error occurs during the assembly process,
-                      the action must be able to guarantee idempotence to allow for
-                      retries from the beginning. This field cannot be updated."
+                  dataDump:
+                    description: "Defines the method to dump the data from a replica.
+                      This action is typically used when a new replica needs to be
+                      constructed, such as: \n - scale-out - rebuild - clone \n It
+                      should write the valid data to stdout without including any
+                      extraneous information. This field cannot be updated."
                     properties:
                       builtinHandler:
                         description: BuiltinHandler specifies the builtin action handler
@@ -819,12 +817,13 @@ spec:
                             type: integer
                         type: object
                     type: object
-                  dataPopulate:
-                    description: "Defines the method to populate the data to create
-                      new replicas. This action is typically used when a new replica
-                      needs to be constructed, such as: \n - scale-out - rebuild -
-                      clone \n It should write the valid data to stdout without including
-                      any extraneous information. This field cannot be updated."
+                  dataLoad:
+                    description: "Defines the method to load data into a replica.
+                      This action is typically used when creating a new replica, such
+                      as: \n - scale-out - rebuild - clone \n The data will be streamed
+                      in via stdin. If any error occurs during the process, the action
+                      must be able to guarantee idempotence to allow for retries from
+                      the beginning. This field cannot be updated."
                     properties:
                       builtinHandler:
                         description: BuiltinHandler specifies the builtin action handler
diff --git a/controllers/apps/componentdefinition_controller.go b/controllers/apps/componentdefinition_controller.go
index 8e87d861a48..cc286ff0a5b 100644
--- a/controllers/apps/componentdefinition_controller.go
+++ b/controllers/apps/componentdefinition_controller.go
@@ -357,8 +357,8 @@ func (r *ComponentDefinitionReconciler) validateLifecycleActionBuiltInHandlers(l
 		{lifecycleActions.MemberLeave},
 		{lifecycleActions.Readonly},
 		{lifecycleActions.Readwrite},
-		{lifecycleActions.DataPopulate},
-		{lifecycleActions.DataAssemble},
+		{lifecycleActions.DataDump},
+		{lifecycleActions.DataLoad},
 		{lifecycleActions.Reconfigure},
 		{lifecycleActions.AccountProvision},
 	}
diff --git a/deploy/helm/crds/apps.kubeblocks.io_componentdefinitions.yaml b/deploy/helm/crds/apps.kubeblocks.io_componentdefinitions.yaml
index 031ad6a1718..2635d683d10 100644
--- a/deploy/helm/crds/apps.kubeblocks.io_componentdefinitions.yaml
+++ b/deploy/helm/crds/apps.kubeblocks.io_componentdefinitions.yaml
@@ -524,14 +524,12 @@ spec:
                             type: integer
                         type: object
                     type: object
-                  dataAssemble:
-                    description: "Defines the method to assemble data synchronized
-                      from external before starting the service for a new replica.
-                      This action is typically used when creating a new replica, such
-                      as: \n - scale-out - rebuild - clone \n The data will be streamed
-                      in via stdin. If any error occurs during the assembly process,
-                      the action must be able to guarantee idempotence to allow for
-                      retries from the beginning. This field cannot be updated."
+                  dataDump:
+                    description: "Defines the method to dump the data from a replica.
+                      This action is typically used when a new replica needs to be
+                      constructed, such as: \n - scale-out - rebuild - clone \n It
+                      should write the valid data to stdout without including any
+                      extraneous information. This field cannot be updated."
                     properties:
                       builtinHandler:
                         description: BuiltinHandler specifies the builtin action handler
@@ -819,12 +817,13 @@ spec:
                             type: integer
                         type: object
                     type: object
-                  dataPopulate:
-                    description: "Defines the method to populate the data to create
-                      new replicas. This action is typically used when a new replica
-                      needs to be constructed, such as: \n - scale-out - rebuild -
-                      clone \n It should write the valid data to stdout without including
-                      any extraneous information. This field cannot be updated."
+                  dataLoad:
+                    description: "Defines the method to load data into a replica.
+                      This action is typically used when creating a new replica, such
+                      as: \n - scale-out - rebuild - clone \n The data will be streamed
+                      in via stdin. If any error occurs during the process, the action
+                      must be able to guarantee idempotence to allow for retries from
+                      the beginning. This field cannot be updated."
                     properties:
                       builtinHandler:
                         description: BuiltinHandler specifies the builtin action handler
diff --git a/docs/developer_docs/api-reference/cluster.md b/docs/developer_docs/api-reference/cluster.md
index c32558e684c..83df35d6bc5 100644
--- a/docs/developer_docs/api-reference/cluster.md
+++ b/docs/developer_docs/api-reference/cluster.md
@@ -7724,7 +7724,7 @@ LifecycleActionHandler
 </tr>
 <tr>
 <td>
-<code>dataPopulate</code><br/>
+<code>dataDump</code><br/>
 <em>
 <a href="#apps.kubeblocks.io/v1alpha1.LifecycleActionHandler">
 LifecycleActionHandler
@@ -7733,7 +7733,7 @@ LifecycleActionHandler
 </td>
 <td>
 <em>(Optional)</em>
-<p>Defines the method to populate the data to create new replicas.
+<p>Defines the method to dump the data from a replica.
 This action is typically used when a new replica needs to be constructed, such as:</p>
 <ul>
 <li>scale-out</li>
@@ -7746,7 +7746,7 @@ This field cannot be updated.</p>
 </tr>
 <tr>
 <td>
-<code>dataAssemble</code><br/>
+<code>dataLoad</code><br/>
 <em>
 <a href="#apps.kubeblocks.io/v1alpha1.LifecycleActionHandler">
 LifecycleActionHandler
@@ -7755,14 +7755,14 @@ LifecycleActionHandler
 </td>
 <td>
 <em>(Optional)</em>
-<p>Defines the method to assemble data synchronized from external before starting the service for a new replica.
+<p>Defines the method to load data into a replica.
 This action is typically used when creating a new replica, such as:</p>
 <ul>
 <li>scale-out</li>
 <li>rebuild</li>
 <li>clone</li>
 </ul>
-<p>The data will be streamed in via stdin. If any error occurs during the assembly process,
+<p>The data will be streamed in via stdin. If any error occurs during the process,
 the action must be able to guarantee idempotence to allow for retries from the beginning.
 This field cannot be updated.</p>
 </td>
diff --git a/pkg/constant/lorry.go b/pkg/constant/lorry.go
index 42215b79989..6e487b13330 100644
--- a/pkg/constant/lorry.go
+++ b/pkg/constant/lorry.go
@@ -38,6 +38,8 @@ const (
 	ReadWriteAction     = "readwrite"
 	PostProvisionAction = "postProvision"
 	PreTerminateAction  = "preTerminate"
+	DataDumpAction      = "dataDump"
+	DataLoadAction      = "dataLoad"
 )
 
 // action envs
diff --git a/pkg/controller/component/component_definition_convertor.go b/pkg/controller/component/component_definition_convertor.go
index da501f1b8e3..211c1ec3e23 100644
--- a/pkg/controller/component/component_definition_convertor.go
+++ b/pkg/controller/component/component_definition_convertor.go
@@ -580,8 +580,8 @@ func (c *compDefLifecycleActionsConvertor) convert(args ...any) (any, error) {
 	lifecycleActions.MemberLeave = nil
 	lifecycleActions.Readonly = nil
 	lifecycleActions.Readwrite = nil
-	lifecycleActions.DataPopulate = nil
-	lifecycleActions.DataAssemble = nil
+	lifecycleActions.DataDump = nil
+	lifecycleActions.DataLoad = nil
 	lifecycleActions.Reconfigure = nil
 	lifecycleActions.AccountProvision = nil
 
diff --git a/pkg/controller/component/lorry_utils.go b/pkg/controller/component/lorry_utils.go
index c290a3c6cea..bdc86f823b2 100644
--- a/pkg/controller/component/lorry_utils.go
+++ b/pkg/controller/component/lorry_utils.go
@@ -405,8 +405,8 @@ func getBuiltinActionHandler(synthesizeComp *SynthesizedComponent) appsv1alpha1.
 		synthesizeComp.LifecycleActions.MemberLeave,
 		synthesizeComp.LifecycleActions.Readonly,
 		synthesizeComp.LifecycleActions.Readwrite,
-		synthesizeComp.LifecycleActions.DataPopulate,
-		synthesizeComp.LifecycleActions.DataAssemble,
+		synthesizeComp.LifecycleActions.DataDump,
+		synthesizeComp.LifecycleActions.DataLoad,
 		synthesizeComp.LifecycleActions.Reconfigure,
 		// synthesizeComp.LifecycleActions.AccountProvision,
 	}
@@ -438,9 +438,9 @@ func getActionCommandsWithExecImageOrContainerName(synthesizeComp *SynthesizedCo
 		constant.MemberLeaveAction:   synthesizeComp.LifecycleActions.MemberLeave,
 		constant.ReadonlyAction:      synthesizeComp.LifecycleActions.Readonly,
 		constant.ReadWriteAction:     synthesizeComp.LifecycleActions.Readwrite,
-		// "dataPopulate":     synthesizeComp.LifecycleActions.DataPopulate,
-		// "dataAssemble":     synthesizeComp.LifecycleActions.DataAssemble,
-		// "reconfigure":      synthesizeComp.LifecycleActions.Reconfigure,
+		constant.DataDumpAction:      synthesizeComp.LifecycleActions.DataDump,
+		constant.DataLoadAction:      synthesizeComp.LifecycleActions.DataLoad,
+		// "reconfigure":                synthesizeComp.LifecycleActions.Reconfigure,
 		// "accountProvision": synthesizeComp.LifecycleActions.AccountProvision,
 	}
 
diff --git a/pkg/lorry/client/client.go b/pkg/lorry/client/client.go
index 435f98202ef..37d5658bad2 100644
--- a/pkg/lorry/client/client.go
+++ b/pkg/lorry/client/client.go
@@ -243,6 +243,16 @@ func (cli *lorryClient) PreTerminate(ctx context.Context) error {
 	return err
 }
 
+func (cli *lorryClient) DataDump(ctx context.Context) error {
+	_, err := cli.Request(ctx, string(DataDumpOperation), http.MethodPost, nil)
+	return err
+}
+
+func (cli *lorryClient) DataLoad(ctx context.Context) error {
+	_, err := cli.Request(ctx, string(DataLoadOperation), http.MethodPost, nil)
+	return err
+}
+
 func (cli *lorryClient) Request(ctx context.Context, operation, method string, req map[string]any) (map[string]any, error) {
 	if cli.requester == nil {
 		return nil, errors.New("lorry client's requester must be set")
diff --git a/pkg/lorry/client/client_mock.go b/pkg/lorry/client/client_mock.go
index 689d761af2c..0f97d5e82d0 100644
--- a/pkg/lorry/client/client_mock.go
+++ b/pkg/lorry/client/client_mock.go
@@ -69,6 +69,34 @@ func (mr *MockClientMockRecorder) CreateUser(arg0, arg1, arg2, arg3 interface{})
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateUser", reflect.TypeOf((*MockClient)(nil).CreateUser), arg0, arg1, arg2, arg3)
 }
 
+// DataDump mocks base method.
+func (m *MockClient) DataDump(arg0 context.Context) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "DataDump", arg0)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// DataDump indicates an expected call of DataDump.
+func (mr *MockClientMockRecorder) DataDump(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DataDump", reflect.TypeOf((*MockClient)(nil).DataDump), arg0)
+}
+
+// DataLoad mocks base method.
+func (m *MockClient) DataLoad(arg0 context.Context) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "DataLoad", arg0)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// DataLoad indicates an expected call of DataLoad.
+func (mr *MockClientMockRecorder) DataLoad(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DataLoad", reflect.TypeOf((*MockClient)(nil).DataLoad), arg0)
+}
+
 // DeleteUser mocks base method.
 func (m *MockClient) DeleteUser(arg0 context.Context, arg1 string) error {
 	m.ctrl.T.Helper()
diff --git a/pkg/lorry/client/interface.go b/pkg/lorry/client/interface.go
index 0685ed558ea..b3ff8c4e3d0 100644
--- a/pkg/lorry/client/interface.go
+++ b/pkg/lorry/client/interface.go
@@ -45,4 +45,7 @@ type Client interface {
 	Unlock(ctx context.Context) error
 	PostProvision(ctx context.Context, componentNames, podNames, podIPs, podHostNames, podHostIPs string) error
 	PreTerminate(ctx context.Context) error
+
+	DataDump(ctx context.Context) error
+	DataLoad(ctx context.Context) error
 }
diff --git a/pkg/lorry/operations/replica/data_dump.go b/pkg/lorry/operations/replica/data_dump.go
new file mode 100644
index 00000000000..b9a615498af
--- /dev/null
+++ b/pkg/lorry/operations/replica/data_dump.go
@@ -0,0 +1,83 @@
+/*
+Copyright (C) 2022-2024 ApeCloud Co., Ltd
+
+This file is part of KubeBlocks project
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package replica
+
+import (
+	"context"
+	"encoding/json"
+	"strings"
+
+	"github.com/go-logr/logr"
+	"github.com/spf13/viper"
+
+	"github.com/apecloud/kubeblocks/pkg/constant"
+	"github.com/apecloud/kubeblocks/pkg/lorry/operations"
+	"github.com/apecloud/kubeblocks/pkg/lorry/util"
+)
+
+type dataDump struct {
+	operations.Base
+	logger  logr.Logger
+	Command []string
+}
+
+func init() {
+	err := operations.Register(strings.ToLower(string(util.DataDumpOperation)), &dataDump{})
+	if err != nil {
+		panic(err.Error())
+	}
+}
+
+func (s *dataDump) Init(_ context.Context) error {
+	actionJSON := viper.GetString(constant.KBEnvActionCommands)
+	if actionJSON != "" {
+		actionCommands := map[string][]string{}
+		err := json.Unmarshal([]byte(actionJSON), &actionCommands)
+		if err != nil {
+			s.logger.Info("get action commands failed", "error", err.Error())
+			return err
+		}
+		cmd, ok := actionCommands[constant.DataDumpAction]
+		if ok && len(cmd) > 0 {
+			s.Command = cmd
+		}
+	}
+	return nil
+}
+
+func (s *dataDump) PreCheck(ctx context.Context, req *operations.OpsRequest) error {
+	return nil
+}
+
+func (s *dataDump) Do(ctx context.Context, req *operations.OpsRequest) (*operations.OpsResponse, error) {
+	return nil, doCommonAction(ctx, s.logger, "dataDump", s.Command)
+}
+
+func doCommonAction(ctx context.Context, logger logr.Logger, action string, commands []string) error {
+	envs, err := util.GetGlobalSharedEnvs()
+	if err != nil {
+		return err
+	}
+	output, err := util.ExecCommand(ctx, commands, envs)
+	if output != "" {
+		logger.Info(action, "output", output)
+	}
+	return err
+}
diff --git a/pkg/lorry/operations/replica/data_load.go b/pkg/lorry/operations/replica/data_load.go
new file mode 100644
index 00000000000..643b4fef891
--- /dev/null
+++ b/pkg/lorry/operations/replica/data_load.go
@@ -0,0 +1,71 @@
+/*
+Copyright (C) 2022-2024 ApeCloud Co., Ltd
+
+This file is part of KubeBlocks project
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package replica
+
+import (
+	"context"
+	"encoding/json"
+	"strings"
+
+	"github.com/go-logr/logr"
+	"github.com/spf13/viper"
+
+	"github.com/apecloud/kubeblocks/pkg/constant"
+	"github.com/apecloud/kubeblocks/pkg/lorry/operations"
+	"github.com/apecloud/kubeblocks/pkg/lorry/util"
+)
+
+type dataLoad struct {
+	operations.Base
+	logger  logr.Logger
+	Command []string
+}
+
+func init() {
+	err := operations.Register(strings.ToLower(string(util.DataLoadOperation)), &dataLoad{})
+	if err != nil {
+		panic(err.Error())
+	}
+}
+
+func (s *dataLoad) Init(_ context.Context) error {
+	actionJSON := viper.GetString(constant.KBEnvActionCommands)
+	if actionJSON != "" {
+		actionCommands := map[string][]string{}
+		err := json.Unmarshal([]byte(actionJSON), &actionCommands)
+		if err != nil {
+			s.logger.Info("get action commands failed", "error", err.Error())
+			return err
+		}
+		cmd, ok := actionCommands[constant.DataLoadAction]
+		if ok && len(cmd) > 0 {
+			s.Command = cmd
+		}
+	}
+	return nil
+}
+
+func (s *dataLoad) PreCheck(ctx context.Context, req *operations.OpsRequest) error {
+	return nil
+}
+
+func (s *dataLoad) Do(ctx context.Context, req *operations.OpsRequest) (*operations.OpsResponse, error) {
+	return nil, doCommonAction(ctx, s.logger, "dataLoad", s.Command)
+}
diff --git a/pkg/lorry/util/types.go b/pkg/lorry/util/types.go
index c0b1ed73ef2..70d612cf2cd 100644
--- a/pkg/lorry/util/types.go
+++ b/pkg/lorry/util/types.go
@@ -73,6 +73,9 @@ const (
 	OperationFailed            = "Failed"
 	DefaultProbeTimeoutSeconds = 2
 
+	DataDumpOperation OperationKind = "dataDump"
+	DataLoadOperation OperationKind = "dataLoad"
+
 	// this is a general script template, which can be used for all kinds of exec request to databases.
 	DataScriptRequestTpl string = `
 		response=$(curl -s -X POST -H 'Content-Type: application/json' http://%s:3501/v1.0/bindings/%s -d '%s')
diff --git a/pkg/testutil/apps/constant.go b/pkg/testutil/apps/constant.go
index 7a7d733ffa0..c8cb5717694 100644
--- a/pkg/testutil/apps/constant.go
+++ b/pkg/testutil/apps/constant.go
@@ -349,8 +349,8 @@ var (
 			MemberLeave:      defaultLifecycleActionHandler,
 			Readonly:         defaultLifecycleActionHandler,
 			Readwrite:        defaultLifecycleActionHandler,
-			DataPopulate:     defaultLifecycleActionHandler,
-			DataAssemble:     defaultLifecycleActionHandler,
+			DataDump:         defaultLifecycleActionHandler,
+			DataLoad:         defaultLifecycleActionHandler,
 			Reconfigure:      defaultLifecycleActionHandler,
 			AccountProvision: defaultLifecycleActionHandler,
 		},