Add stop replication visual policy action

Signed-off-by: wzyns <me@wzyns.com>

Author: wzyns

models/interfaces.ts

@@ -360,6 +360,10 @@ export interface DeleteAction extends Action {
   delete: {};
 }
 
+export interface StopReplicationAction extends Action {
+  stop_replication: {};
+}
+
 export interface RolloverAction extends Action {
   rollover: {
     min_size?: string;

public/pages/VisualCreatePolicy/components/UIActions/StopReplicationUIAction.tsx

@@ -0,0 +1,32 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React from "react";
+import { StopReplicationAction, UIAction } from "../../../../../models/interfaces";
+import { makeId } from "../../../../utils/helpers";
+import { ActionType } from "../../utils/constants";
+
+export class StopReplicationUIAction implements UIAction<StopReplicationAction> {
+  id: string;
+  action: StopReplicationAction;
+  type = ActionType.StopReplication;
+
+  constructor(action: StopReplicationAction, id: string = makeId()) {
+    this.action = action;
+    this.id = id;
+  }
+
+  content = () => `Stop replication`;
+
+  clone = (action: StopReplicationAction) => new StopReplicationUIAction(action, this.id);
+
+  isValid = () => true;
+
+  render = (action: UIAction<StopReplicationAction>, onChangeAction: (action: UIAction<StopReplicationAction>) => void) => {
+    return <div />;
+  };
+
+  toAction = () => this.action;
+}

public/pages/VisualCreatePolicy/components/UIActions/index.ts

@@ -18,6 +18,7 @@ import RolloverUIAction from "./RolloverUIAction";
 import RollupUIAction from "./RollupUIAction";
 import ShrinkUIAction from "./ShrinkUIAction";
 import SnapshotUIAction from "./SnapshotUIAction";
+import { StopReplicationUIAction } from "./StopReplicationUIAction";
 
 export {
   AliasUIAction,
@@ -35,4 +36,5 @@ export {
   RollupUIAction,
   ShrinkUIAction,
   SnapshotUIAction,
+  StopReplicationUIAction,
 };

public/pages/VisualCreatePolicy/containers/CreateAction/CreateAction.test.tsx

@@ -26,4 +26,21 @@ describe("<CreateAction /> spec", () => {
     );
     expect(container.firstChild).toMatchSnapshot();
   });
+
+  it("includes stop replication in the action type dropdown", () => {
+    const rolloverAction: UIAction<any> = new RolloverUIAction(DEFAULT_ROLLOVER, "some_id");
+    const { container } = render(
+      <CreateAction
+        stateName="some_state"
+        actions={[rolloverAction]}
+        editAction={rolloverAction}
+        onClickCancelAction={() => {}}
+        onClickSaveAction={() => {}}
+      />
+    );
+
+    const stopReplicationOption = container.querySelector('option[value="stop_replication"]');
+    expect(stopReplicationOption).toBeInTheDocument();
+    expect(stopReplicationOption).toHaveTextContent("Stop Replication");
+  });
 });

public/pages/VisualCreatePolicy/containers/CreateAction/__snapshots__/CreateAction.test.tsx.snap

@@ -170,6 +170,11 @@ exports[`<CreateAction /> spec renders the component 1`] = `
                 >
                   Snapshot
                 </option>
+                <option
+                  value="stop_replication"
+                >
+                  Stop Replication
+                </option>
               </select>
               <div
                 class="euiFormControlLayoutIcons euiFormControlLayoutIcons--right"

public/pages/VisualCreatePolicy/containers/CreateState/CreateState.test.tsx

@@ -8,10 +8,26 @@ import "@testing-library/jest-dom/extend-expect";
 import { render } from "@testing-library/react";
 import CreateState from "./CreateState";
 import { DEFAULT_POLICY } from "../../utils/constants";
+import { Policy, State } from "../../../../../models/interfaces";
 
 describe("<CreateState /> spec", () => {
   it("renders the component", () => {
     render(<CreateState policy={DEFAULT_POLICY} onSaveState={() => {}} onCloseFlyout={() => {}} state={null} />);
     expect(document.body.children[1]).toMatchSnapshot();
   });
+
+  it("loads a state with a stop replication action", () => {
+    const state: State = {
+      name: "stop_replication",
+      actions: [{ stop_replication: {} }],
+      transitions: [],
+    };
+    const policy: Policy = {
+      ...DEFAULT_POLICY,
+      default_state: state.name,
+      states: [state],
+    };
+
+    expect(() => render(<CreateState policy={policy} onSaveState={() => {}} onCloseFlyout={() => {}} state={state} />)).not.toThrow();
+  });
 });

public/pages/VisualCreatePolicy/utils/constants.ts

@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { AliasAction, AllocationAction } from "../../../../models/interfaces";
+import { AliasAction, AllocationAction, StopReplicationAction } from "../../../../models/interfaces";
 
 export enum ActionType {
   Alias = "alias",
@@ -21,6 +21,7 @@ export enum ActionType {
   Rollup = "rollup",
   Shrink = "shrink",
   Snapshot = "snapshot",
+  StopReplication = "stop_replication",
 }
 
 export const DEFAULT_POLICY = {
@@ -194,6 +195,9 @@ export const DEFAULT_SNAPSHOT = {
     snapshot: "example-snapshot",
   },
 };
+export const DEFAULT_STOP_REPLICATION: StopReplicationAction = {
+  stop_replication: {},
+};
 
 export const actions = [
   DEFAULT_ALIAS,
@@ -211,6 +215,7 @@ export const actions = [
   DEFAULT_ROLLUP,
   DEFAULT_SHRINK,
   DEFAULT_SNAPSHOT,
+  DEFAULT_STOP_REPLICATION,
 ];
 
 export const ISM_TEMPLATE_INPUT_MAX_WIDTH = "400px";

public/pages/VisualCreatePolicy/utils/helpers.test.ts

@@ -6,7 +6,8 @@
 import { actionRepoSingleton, capitalizeFirstLetter, convertTemplatesToArray, getOrderInfo, getUpdatedPolicy } from "./helpers";
 import { Action, ISMTemplate, UIAction } from "../../../../models/interfaces";
 import { makeId } from "../../../utils/helpers";
-import { DEFAULT_POLICY } from "./constants";
+import { DEFAULT_POLICY, DEFAULT_STOP_REPLICATION } from "./constants";
+import { StopReplicationUIAction } from "../components/UIActions";
 
 test("converts all ism template formats into a list of ism templates", () => {
   expect(convertTemplatesToArray(null)).toEqual([]);
@@ -75,13 +76,22 @@ class DummyUIAction implements UIAction<DummyAction> {
 }
 
 test("action repository usage", () => {
-  expect(actionRepoSingleton.getAllActionTypes().length).toBe(15);
-  actionRepoSingleton.registerAction("dummy", DummyUIAction, DEFAULT_DUMMY);
   expect(actionRepoSingleton.getAllActionTypes().length).toBe(16);
+  actionRepoSingleton.registerAction("dummy", DummyUIAction, DEFAULT_DUMMY);
+  expect(actionRepoSingleton.getAllActionTypes().length).toBe(17);
   expect(actionRepoSingleton.getUIAction("dummy") instanceof DummyUIAction).toBe(true);
   expect(actionRepoSingleton.getUIActionFromData(DEFAULT_DUMMY) instanceof DummyUIAction).toBe(true);
 });
 
+test("action repository supports stop replication action", () => {
+  expect(actionRepoSingleton.getAllActionTypes()).toContain("stop_replication");
+  expect(actionRepoSingleton.getUIAction("stop_replication") instanceof StopReplicationUIAction).toBe(true);
+  expect(actionRepoSingleton.getUIActionFromData(DEFAULT_STOP_REPLICATION) instanceof StopReplicationUIAction).toBe(true);
+  expect(actionRepoSingleton.getUIActionFromData({ stop_replication: {} }).toAction()).toEqual({
+    stop_replication: {},
+  });
+});
+
 test("changing the default state name correctly updates default_state", () => {
   const policy = DEFAULT_POLICY;
   const updatedState = { ...policy.states[0], name: "new_hot" };

public/pages/VisualCreatePolicy/utils/helpers.ts

@@ -21,6 +21,7 @@ import {
   DEFAULT_ROLLUP,
   DEFAULT_SHRINK,
   DEFAULT_SNAPSHOT,
+  DEFAULT_STOP_REPLICATION,
 } from "./constants";
 import {
   AliasUIAction,
@@ -38,6 +39,7 @@ import {
   RollupUIAction,
   ShrinkUIAction,
   SnapshotUIAction,
+  StopReplicationUIAction,
 } from "../components/UIActions";
 
 export const getConditionContent = (transition: Transition): string => {
@@ -99,6 +101,8 @@ export const getUIAction = (actionType: string): UIAction<any> => {
       return new ShrinkUIAction(DEFAULT_SHRINK);
     case ActionType.Snapshot:
       return new SnapshotUIAction(DEFAULT_SNAPSHOT);
+    case ActionType.StopReplication:
+      return new StopReplicationUIAction(DEFAULT_STOP_REPLICATION);
     default:
       throw new Error(`Action type [${actionType}] not supported`);
   }
@@ -139,6 +143,7 @@ class ActionRepository {
     rollup: [RollupUIAction, DEFAULT_ROLLUP],
     shrink: [ShrinkUIAction, DEFAULT_SHRINK],
     snapshot: [SnapshotUIAction, DEFAULT_SNAPSHOT],
+    stop_replication: [StopReplicationUIAction, DEFAULT_STOP_REPLICATION],
   };
 
   getAllActionTypes = () => {