diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/AdaptiveFlowEditor.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/AdaptiveFlowEditor.test.tsx
new file mode 100644
index 0000000000..ad83d07308
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/AdaptiveFlowEditor.test.tsx
@@ -0,0 +1,27 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+import { ExtensionContext } from '@bfc/extension';
+
+import AdaptiveFlowEditor from '../../src/adaptive-flow-editor/AdaptiveFlowEditor';
+
+import { ShellApiStub } from './stubs/ShellApiStub';
+
+describe('<VisualDesigner/>', () => {
+  it('can render.', () => {
+    const visualDesigner = render(
+      <ExtensionContext.Provider
+        value={{
+          shellApi: ShellApiStub,
+          shellData: {} as any,
+          plugins: [],
+        }}
+      >
+        <AdaptiveFlowEditor />
+      </ExtensionContext.Provider>
+    );
+    expect(visualDesigner).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/components/KeyboardZone.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/components/KeyboardZone.test.tsx
new file mode 100644
index 0000000000..0625746ee8
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/components/KeyboardZone.test.tsx
@@ -0,0 +1,39 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render, fireEvent } from '@bfc/test-utils';
+
+import { KeyboardZone } from '../../../src/adaptive-flow-editor/components/KeyboardZone';
+
+describe('KeyboardZone', () => {
+  it('can be rendered.', () => {
+    const zone = render(
+      <KeyboardZone onCommand={() => undefined}>
+        <span data-testid="zone-child">children</span>
+      </KeyboardZone>
+    );
+    expect(zone).toBeTruthy();
+    expect(zone.getByTestId('zone-child')).toBeTruthy();
+  });
+
+  it('can trigger onCommand.', () => {
+    const mockOnCommand = jest.fn();
+    const zone = render(
+      <KeyboardZone onCommand={mockOnCommand}>
+        <span data-testid="zone-child">children</span>
+      </KeyboardZone>
+    ).getByTestId('keyboard-zone');
+
+    fireEvent.focus(zone);
+    fireEvent.keyDown(
+      zone,
+      new KeyboardEvent('keydown', {
+        key: 'C',
+        code: 'C',
+        ctrlKey: true,
+      })
+    );
+    expect(mockOnCommand).toHaveBeenCalled();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/ElementAttributes.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/ElementAttributes.test.tsx
new file mode 100644
index 0000000000..70471db044
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/ElementAttributes.test.tsx
@@ -0,0 +1,18 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { AttrNames } from '../../../src/adaptive-flow-editor/constants/ElementAttributes';
+
+describe('ElementAttributes', () => {
+  it('should contain necessary attrs.', () => {
+    expect(AttrNames.SelectableElement).toBeTruthy();
+    expect(AttrNames.NodeElement).toBeTruthy();
+    expect(AttrNames.EdgeMenuElement).toBeTruthy();
+    expect(AttrNames.FocusedId).toBeTruthy();
+    expect(AttrNames.InlineLinkElement).toBeTruthy();
+    expect(AttrNames.SelectedId).toBeTruthy();
+    expect(AttrNames.Tab).toBeTruthy();
+    expect(AttrNames.FocusedId).toBeTruthy();
+    expect(AttrNames.SelectionIndex).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/KeyboardCommandTypes.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/KeyboardCommandTypes.test.ts
new file mode 100644
index 0000000000..edbcb96764
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/KeyboardCommandTypes.test.ts
@@ -0,0 +1,66 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import {
+  KeyboardCommandTypes,
+  KeyboardPrimaryTypes,
+  mapShortcutToKeyboardCommand,
+} from '../../../src/adaptive-flow-editor/constants/KeyboardCommandTypes';
+
+describe('KeyboardCommandTypes', () => {
+  it('should be truthy.', () => {
+    expect(KeyboardCommandTypes).toBeTruthy();
+  });
+
+  it('should contain 3 categories.', () => {
+    expect(KeyboardCommandTypes.Cursor).toBeTruthy();
+    expect(KeyboardCommandTypes.Node).toBeTruthy();
+    expect(KeyboardCommandTypes.Operation).toBeTruthy();
+  });
+
+  it('<Cursor> should contain necessary commands.', () => {
+    const { Cursor } = KeyboardCommandTypes;
+
+    expect(Cursor.MoveUp).toBeTruthy();
+    expect(Cursor.MoveDown).toBeTruthy();
+    expect(Cursor.MoveLeft).toBeTruthy();
+    expect(Cursor.MoveRight).toBeTruthy();
+    expect(Cursor.ShortMoveUp).toBeTruthy();
+    expect(Cursor.ShortMoveDown).toBeTruthy();
+    expect(Cursor.ShortMoveLeft).toBeTruthy();
+    expect(Cursor.ShortMoveRight).toBeTruthy();
+    expect(Cursor.MovePrevious).toBeTruthy();
+    expect(Cursor.MoveNext).toBeTruthy();
+  });
+
+  it('<Node> should contain necessary actions.', () => {
+    const { Node } = KeyboardCommandTypes;
+
+    expect(Node.Delete).toBeTruthy();
+    expect(Node.Copy).toBeTruthy();
+    expect(Node.Cut).toBeTruthy();
+    expect(Node.Paste).toBeTruthy();
+  });
+
+  it('<Operation> should contain necessary operations.', () => {
+    const { Operation } = KeyboardCommandTypes;
+
+    expect(Operation.Undo).toBeTruthy();
+    expect(Operation.Redo).toBeTruthy();
+  });
+});
+
+describe('KeyboardPrimaryTypes', () => {
+  it('should contain 3 types.', () => {
+    expect(KeyboardPrimaryTypes.Cursor).toBeTruthy();
+    expect(KeyboardPrimaryTypes.Node).toBeTruthy();
+    expect(KeyboardPrimaryTypes.Operation).toBeTruthy();
+  });
+});
+
+describe('mapShortcutToKeyboardCommand', () => {
+  expect(mapShortcutToKeyboardCommand('Windows.Control.Z')).toEqual({
+    area: 'Operation',
+    command: 'undo',
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/MenuTypes.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/MenuTypes.test.ts
new file mode 100644
index 0000000000..07c2f13ac2
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/MenuTypes.test.ts
@@ -0,0 +1,12 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { MenuTypes, MenuEventTypes } from '../../../src/adaptive-flow-editor/constants/MenuTypes';
+
+describe('MenuTypes', () => {
+  it('should contain necessary keys.', () => {
+    expect(MenuTypes.EdgeMenu).toBeTruthy();
+    expect(MenuTypes.NodeMenu).toBeTruthy();
+    expect(MenuEventTypes.Paste).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/ScreenReaderMessage.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/ScreenReaderMessage.test.ts
new file mode 100644
index 0000000000..b477462737
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/ScreenReaderMessage.test.ts
@@ -0,0 +1,24 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { ScreenReaderMessage } from '../../../src/adaptive-flow-editor/constants/ScreenReaderMessage';
+
+describe('ScreenReaderMessage', () => {
+  it('should contain necessary messages.', () => {
+    expect(ScreenReaderMessage.EventFocused).toBeTruthy();
+    expect(ScreenReaderMessage.ActionFocused).toBeTruthy();
+    expect(ScreenReaderMessage.ActionUnfocused).toBeTruthy();
+    expect(ScreenReaderMessage.RangeSelection).toBeTruthy();
+    expect(ScreenReaderMessage.DialogOpened).toBeTruthy();
+    expect(ScreenReaderMessage.ActionDeleted).toBeTruthy();
+    expect(ScreenReaderMessage.ActionsDeleted).toBeTruthy();
+    expect(ScreenReaderMessage.ActionCreated).toBeTruthy();
+    expect(ScreenReaderMessage.ActionsCreated).toBeTruthy();
+    expect(ScreenReaderMessage.EventCreated).toBeTruthy();
+    expect(ScreenReaderMessage.ActionsCopied).toBeTruthy();
+    expect(ScreenReaderMessage.ActionsCut).toBeTruthy();
+    expect(ScreenReaderMessage.ActionsMoved).toBeTruthy();
+    expect(ScreenReaderMessage.ActionUndo).toBeTruthy();
+    expect(ScreenReaderMessage.ActionRedo).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/editorConfig.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/editorConfig.test.ts
new file mode 100644
index 0000000000..38abeef675
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/constants/editorConfig.test.ts
@@ -0,0 +1,17 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { EditorConfig } from '../../../src/adaptive-flow-editor/constants/editorConfig';
+
+describe('EditorConfig', () => {
+  it('should cover several features.', () => {
+    expect(EditorConfig).toBeTruthy();
+    expect(EditorConfig.features).toBeDefined();
+
+    expect(EditorConfig.features.showEvents).toBeDefined();
+    expect(EditorConfig.features.arrowNavigation).toBeDefined();
+    expect(EditorConfig.features.tabNavigation).toBeDefined();
+    expect(EditorConfig.features.keyboardNodeEditing).toBeDefined();
+    expect(EditorConfig.features.keyboardOperationEditing).toBeDefined();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/contexts/NodeRendererContext.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/contexts/NodeRendererContext.test.tsx
new file mode 100644
index 0000000000..40a9108bca
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/contexts/NodeRendererContext.test.tsx
@@ -0,0 +1,50 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React, { useContext } from 'react';
+import { render } from '@bfc/test-utils';
+import { DialogFactory } from '@bfc/shared';
+
+import { NodeRendererContext } from '../../../src/adaptive-flow-editor/contexts/NodeRendererContext';
+
+describe('NodeRendererContext', () => {
+  const CtxtConsumer = () => {
+    const { focusedId, focusedEvent, focusedTab, clipboardActions, dialogFactory, customSchemas } = useContext(
+      NodeRendererContext
+    );
+
+    return (
+      <div>
+        <span data-testid="focusedId-value">{focusedId}</span>
+        <span data-testid="focusedEvent-value">{focusedEvent}</span>
+        <span data-testid="focusedTab-value">{focusedTab}</span>
+        <span data-testid="clipboardAction-length">{clipboardActions.length}</span>
+        <span data-testid="dialogFactory-existence">{'' + !!dialogFactory}</span>
+        <span data-testid="customSchemas-length">{customSchemas.length}</span>
+      </div>
+    );
+  };
+  it('can be consumed.', () => {
+    const ele = render(
+      <NodeRendererContext.Provider
+        value={{
+          focusedId: 'id1',
+          focusedEvent: 'event1',
+          focusedTab: 'tab1',
+          clipboardActions: ['action1'],
+          dialogFactory: new DialogFactory(),
+          customSchemas: [{}],
+        }}
+      >
+        <CtxtConsumer />
+      </NodeRendererContext.Provider>
+    );
+
+    expect(ele.getByTestId('focusedId-value').textContent).toEqual('id1');
+    expect(ele.getByTestId('focusedEvent-value').textContent).toEqual('event1');
+    expect(ele.getByTestId('focusedTab-value').textContent).toEqual('tab1');
+    expect(ele.getByTestId('clipboardAction-length').textContent).toEqual('1');
+    expect(ele.getByTestId('dialogFactory-existence').textContent).toEqual('true');
+    expect(ele.getByTestId('customSchemas-length').textContent).toEqual('1');
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/contexts/SelectionContext.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/contexts/SelectionContext.test.tsx
new file mode 100644
index 0000000000..9f3e15288a
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/contexts/SelectionContext.test.tsx
@@ -0,0 +1,45 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React, { useContext } from 'react';
+import { render } from '@bfc/test-utils';
+
+import { SelectionContext } from '../../../src/adaptive-flow-editor/contexts/SelectionContext';
+
+describe('SelectionContext', () => {
+  const ContextConsumer = () => {
+    const { getNodeIndex, getSelectableIds, selectedIds, setSelectedIds, selectableElements } = useContext(
+      SelectionContext
+    );
+    return (
+      <div>
+        <span data-testid="getNodeIndex-result">{getNodeIndex('')}</span>
+        <span data-testid="getSelectableIds-result">{getSelectableIds().join(',')}</span>
+        <span data-testid="selectedIds-str">{selectedIds.join(',')}</span>
+        <span data-testid="setSelectedIds-existence">{'' + !!setSelectedIds}</span>
+        <span data-testid="selectableElements-length">{selectableElements.length}</span>
+      </div>
+    );
+  };
+  it('can be provided.', () => {
+    const ele = render(
+      <SelectionContext.Provider
+        value={{
+          getNodeIndex: () => 1,
+          getSelectableIds: () => ['a', 'b', 'c'],
+          selectedIds: ['a'],
+          setSelectedIds: () => null,
+          selectableElements: [],
+        }}
+      >
+        <ContextConsumer />
+      </SelectionContext.Provider>
+    );
+
+    expect(ele.getByTestId('getNodeIndex-result').textContent).toEqual('1');
+    expect(ele.getByTestId('getSelectableIds-result').textContent).toEqual('a,b,c');
+    expect(ele.getByTestId('selectedIds-str').textContent).toEqual('a');
+    expect(ele.getByTestId('setSelectedIds-existence').textContent).toEqual('true');
+    expect(ele.getByTestId('selectableElements-length').textContent).toEqual('0');
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/hooks/useEditorEventApi.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/hooks/useEditorEventApi.test.ts
new file mode 100644
index 0000000000..64bfc2d68b
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/hooks/useEditorEventApi.test.ts
@@ -0,0 +1,50 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { renderHook } from '@bfc/test-utils/lib/hooks';
+
+import { useEditorEventApi } from '../../../src/adaptive-flow-editor/hooks/useEditorEventApi';
+import { ShellApiStub } from '../stubs/ShellApiStub';
+import { defaultRendererContextValue } from '../../../src/adaptive-flow-editor/contexts/NodeRendererContext';
+import { defaultSelectionContextValue } from '../../../src/adaptive-flow-editor/contexts/SelectionContext';
+import { NodeEventTypes } from '../../../src/adaptive-flow-renderer/constants/NodeEventTypes';
+
+describe('useSelectionEffect', () => {
+  const hook = renderHook(() =>
+    useEditorEventApi(
+      {
+        path: '',
+        data: {},
+        nodeContext: { ...defaultRendererContextValue, focusedId: 'a' },
+        selectionContext: { ...defaultSelectionContextValue, selectedIds: ['a'] },
+      },
+      ShellApiStub
+    )
+  ).result.current;
+
+  it('returns necessary apis.', () => {
+    expect(hook.handleEditorEvent).toBeTruthy();
+  });
+
+  it('can handle events.', () => {
+    const { handleEditorEvent } = hook;
+
+    handleEditorEvent('event.view.focus' as NodeEventTypes, {});
+    handleEditorEvent('event.view.ctrl-click' as NodeEventTypes, {});
+    handleEditorEvent('event.view.shift-click' as NodeEventTypes, {});
+    handleEditorEvent('event.view.focus-event' as NodeEventTypes, {});
+    handleEditorEvent('event.view.move-cursor' as NodeEventTypes, {});
+    handleEditorEvent('event.nav.opendialog' as NodeEventTypes, {});
+    handleEditorEvent('event.data.delete' as NodeEventTypes, {});
+    handleEditorEvent('event.data.insert' as NodeEventTypes, {});
+    handleEditorEvent('event.data.copy-selection' as NodeEventTypes, {});
+    handleEditorEvent('event.data.cut-selection' as NodeEventTypes, {});
+    handleEditorEvent('event.data.paste-selection' as NodeEventTypes, {});
+    handleEditorEvent('event.data.move-selection' as NodeEventTypes, {});
+    handleEditorEvent('event.data.delete-selection' as NodeEventTypes, {});
+    handleEditorEvent('event.data.paste-selection--keyboard' as NodeEventTypes, {});
+    handleEditorEvent('event.data.paste-selection--menu' as NodeEventTypes, {});
+    handleEditorEvent('event.operation.undo' as NodeEventTypes, {});
+    handleEditorEvent('event.operation.redo' as NodeEventTypes, {});
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/EdgeMenu.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/EdgeMenu.test.tsx
new file mode 100644
index 0000000000..177479e59a
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/EdgeMenu.test.tsx
@@ -0,0 +1,45 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+import { DialogGroup } from '@bfc/shared';
+
+import { EdgeMenu } from '../../../src/adaptive-flow-editor/renderers/EdgeMenu';
+import { createActionMenu } from '../../../src/adaptive-flow-editor/renderers/EdgeMenu/createSchemaMenu';
+
+describe('<EdgeMenu>', () => {
+  it('can render.', () => {
+    const menu = render(<EdgeMenu id="test" onClick={() => undefined} />);
+    expect(menu).toBeTruthy();
+  });
+});
+
+describe('createActionMenu()', () => {
+  it('options.enablePaste should control Paste button state.', () => {
+    const menuItems1 = createActionMenu(() => null, { isSelfHosted: false, enablePaste: true });
+    expect(menuItems1.findIndex((x) => x.key === 'Paste')).toEqual(0);
+    expect(menuItems1[0].disabled).toBeFalsy();
+
+    const menuItems2 = createActionMenu(() => null, { isSelfHosted: false, enablePaste: false });
+    expect(menuItems2[0].disabled).toBeTruthy();
+  });
+
+  it('should return builtin $kinds.', () => {
+    const menuItemsHosted = createActionMenu(() => null, { isSelfHosted: true, enablePaste: true });
+    expect(menuItemsHosted.findIndex((x) => x.key === DialogGroup.RESPONSE)).toBeTruthy();
+  });
+
+  it('should show custom actions as last item.', () => {
+    const menuItemsWithoutCustomActions = createActionMenu(() => null, { isSelfHosted: false, enablePaste: false }, []);
+    expect(menuItemsWithoutCustomActions.findIndex((x) => x.key === 'Custom Actions')).toEqual(-1);
+
+    const customActions = [
+      [{ title: 'Custom1', description: 'Custom1', $ref: 'Group1.Custom1' }],
+      [{ title: 'Custom2', description: 'Custom2', $ref: 'Group2.Custom2' }],
+    ];
+    const withCustomActions = createActionMenu(() => null, { isSelfHosted: false, enablePaste: false }, customActions);
+    expect(withCustomActions.findIndex((x) => x.key === 'Custom Actions')).toEqual(withCustomActions.length - 1);
+    expect(withCustomActions[withCustomActions.length - 1].subMenuProps?.items.length).toEqual(3); // 2 action labels + 1 sep line
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/ElementWrapper.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/ElementWrapper.test.tsx
new file mode 100644
index 0000000000..81a2850a7b
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/ElementWrapper.test.tsx
@@ -0,0 +1,21 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { ElementWrapper } from '../../../src/adaptive-flow-editor/renderers/ElementWrapper';
+
+describe('<ElementWrapper>', () => {
+  it('can render.', () => {
+    const ele = render(
+      <ElementWrapper nodeId="test" tagId="1">
+        <span data-testid="wrapped-content">Content</span>
+      </ElementWrapper>
+    );
+
+    expect(ele).toBeTruthy();
+    expect(ele.getByTestId('wrapped-content')).toBeTruthy();
+    expect(ele.getByTestId('wrapped-content').textContent).toEqual('Content');
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/NodeWrapper.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/NodeWrapper.test.tsx
new file mode 100644
index 0000000000..8ae45c2f02
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/renderers/NodeWrapper.test.tsx
@@ -0,0 +1,30 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render, fireEvent } from '@bfc/test-utils';
+import ExtensionContext from '@bfc/extension/lib/extensionContext';
+
+import { ActionNodeWrapper } from '../../../src/adaptive-flow-editor/renderers/NodeWrapper';
+import { ShellApiStub } from '../stubs/ShellApiStub';
+
+describe('<ActionNodeWrapper>', () => {
+  it('can render.', () => {
+    const mockOnEvent = jest.fn();
+    const ele = render(
+      <ExtensionContext.Provider
+        value={{
+          shellApi: ShellApiStub,
+          shellData: {} as any,
+          plugins: [],
+        }}
+      >
+        <ActionNodeWrapper data={{}} id="test" onEvent={mockOnEvent} />
+      </ExtensionContext.Provider>
+    );
+    expect(ele.getByTestId('ActionNodeWrapper')).toBeTruthy();
+
+    fireEvent.click(ele.getByTestId('ActionNodeWrapper'));
+    expect(mockOnEvent).toHaveBeenCalled();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/stubs/ShellApiStub.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/stubs/ShellApiStub.ts
new file mode 100644
index 0000000000..71a0939388
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/stubs/ShellApiStub.ts
@@ -0,0 +1,45 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { ShellApi } from '@bfc/shared';
+
+const fn = () => ({} as any);
+const fnList = () => [] as any[];
+const fnPromise = () => Promise.resolve({} as any);
+
+export const ShellApiStub: ShellApi = {
+  getDialog: fn,
+  saveDialog: fn,
+  saveData: fn,
+  navTo: fn,
+  onFocusSteps: fn,
+  onFocusEvent: fn,
+  onSelect: fn,
+  getLgTemplates: fnList,
+  copyLgTemplate: fnPromise,
+  addLgTemplate: fnPromise,
+  updateLgTemplate: fnPromise,
+  removeLgTemplate: fnPromise,
+  removeLgTemplates: fnPromise,
+  getLuIntent: fn,
+  getLuIntents: fnList,
+  addLuIntent: fnPromise,
+  updateLuIntent: fnPromise,
+  removeLuIntent: fn,
+  updateRegExIntent: fn,
+  createDialog: fnPromise,
+  addCoachMarkRef: fn,
+  onCopy: fn,
+  undo: fn,
+  redo: fn,
+  updateUserSettings: fn,
+  addSkillDialog: fnPromise,
+  announce: fn,
+  displayManifestModal: fn,
+};
+
+describe('ShellApiStub', () => {
+  it('be truthy.', () => {
+    expect(ShellApiStub).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/NodeIndexGetter.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/NodeIndexGetter.test.ts
new file mode 100644
index 0000000000..ebda36e5e4
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/NodeIndexGetter.test.ts
@@ -0,0 +1,25 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { NodeIndexGenerator } from '../../../src/adaptive-flow-editor/utils/NodeIndexGetter';
+
+describe('NodeIndexGetter', () => {
+  it('can work e2e.', () => {
+    const n = new NodeIndexGenerator();
+
+    const aId = n.getNodeIndex('a');
+    expect(aId).toEqual(0);
+    expect(n.getItemList()).toEqual([{ key: 'a' }]);
+
+    const bId = n.getNodeIndex('b');
+    expect(bId).toEqual(1);
+    expect(n.getItemList()).toEqual([{ key: 'a' }, { key: 'b' }]);
+
+    const bId2 = n.getNodeIndex('b');
+    expect(bId2).toEqual(1);
+    expect(n.getItemList()).toEqual([{ key: 'a' }, { key: 'b' }]);
+
+    n.reset();
+    expect(n.getItemList()).toEqual([]);
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/calculateRangeSelection.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/calculateRangeSelection.test.ts
new file mode 100644
index 0000000000..85457c6b55
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/calculateRangeSelection.test.ts
@@ -0,0 +1,11 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { calculateRangeSelection } from '../../../src/adaptive-flow-editor/utils/calculateRangeSelection';
+
+describe('calculateRangeSelection()', () => {
+  it('could pick correct range.', () => {
+    expect(calculateRangeSelection('b', 'd', ['a', 'b', 'c', 'd'])).toEqual(['b', 'c', 'd']);
+    expect(calculateRangeSelection('d', 'b', ['a', 'b', 'c', 'd'])).toEqual(['b', 'c', 'd']);
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/cursorTracker/index.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/cursorTracker/index.test.ts
new file mode 100644
index 0000000000..6e83131758
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/cursorTracker/index.test.ts
@@ -0,0 +1,46 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { moveCursor } from '../../../../src/adaptive-flow-editor/utils/cursorTracker';
+import { KeyboardCommandTypes } from '../../../../src/adaptive-flow-editor/constants/KeyboardCommandTypes';
+
+describe('moveCursor', () => {
+  it('returns undfined when no selectableElements.', () => {
+    expect(moveCursor([], 'test', KeyboardCommandTypes.Cursor.MovePrevious)).toEqual({
+      selected: 'test',
+      focused: undefined,
+    });
+  });
+
+  it('can handle Tab move.', () => {
+    expect(
+      moveCursor(
+        [
+          {
+            selectedId: 'test',
+            focusedId: 'test-focused',
+            bounds: { left: 0, top: 0, right: 10, bottom: 10 },
+          } as any,
+        ],
+        'test',
+        KeyboardCommandTypes.Cursor.MoveNext
+      )
+    ).toEqual({ focused: 'test-focused', selected: 'test', tab: '' });
+  });
+
+  it('can handle Arrow move.', () => {
+    expect(
+      moveCursor(
+        [
+          {
+            selectedId: 'test',
+            focusedId: 'test-focused',
+            bounds: { left: 0, top: 0, right: 10, bottom: 10 },
+          } as any,
+        ],
+        'test',
+        KeyboardCommandTypes.Cursor.MoveLeft
+      )
+    ).toEqual({ focused: 'test-focused', selected: 'test', tab: '' });
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/cursorTracker/type.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/cursorTracker/type.test.ts
new file mode 100644
index 0000000000..f2a8ab5f92
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/cursorTracker/type.test.ts
@@ -0,0 +1,18 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import {
+  SelectorElement,
+  Direction,
+  BoundRect,
+  Axle,
+} from '../../../../src/adaptive-flow-editor/utils/cursorTracker/type';
+
+describe('types', () => {
+  it('should be declared.', () => {
+    expect(SelectorElement).toBeTruthy();
+    expect(Direction).toBeTruthy();
+    expect(BoundRect).toBeTruthy();
+    expect(Axle).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.test.ts
new file mode 100644
index 0000000000..1c0d4b5658
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.test.ts
@@ -0,0 +1,34 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { mapKeyboardCommandToEditorEvent } from '../../../src/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent';
+import {
+  KeyboardPrimaryTypes,
+  KeyboardCommandTypes,
+} from '../../../src/adaptive-flow-editor/constants/KeyboardCommandTypes';
+import { NodeEventTypes } from '../../../src/adaptive-flow-renderer/constants/NodeEventTypes';
+
+describe('mapKeyboardCommandToEditorEvent()', () => {
+  it('can map event to correct result.', () => {
+    const validationChart = {
+      [KeyboardPrimaryTypes.Node]: {
+        [KeyboardCommandTypes.Node.Delete]: NodeEventTypes.DeleteSelection,
+        [KeyboardCommandTypes.Node.Copy]: NodeEventTypes.CopySelection,
+        [KeyboardCommandTypes.Node.Cut]: NodeEventTypes.CutSelection,
+        [KeyboardCommandTypes.Node.Paste]: NodeEventTypes.PasteSelection,
+      },
+      [KeyboardPrimaryTypes.Operation]: {
+        [KeyboardCommandTypes.Operation.Undo]: NodeEventTypes.Undo,
+        [KeyboardCommandTypes.Operation.Redo]: NodeEventTypes.Redo,
+      },
+    };
+
+    Object.keys(validationChart).forEach((area) => {
+      const subchart = validationChart[area];
+      Object.keys(subchart).forEach((command) => {
+        const resultType = subchart[command];
+        expect(mapKeyboardCommandToEditorEvent({ area, command })?.type).toEqual(resultType);
+      });
+    });
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/mergePluginConfig.test.ts b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/mergePluginConfig.test.ts
new file mode 100644
index 0000000000..e6f444199c
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-editor/utils/mergePluginConfig.test.ts
@@ -0,0 +1,34 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { mergePluginConfig } from '../../../src/adaptive-flow-editor/utils/mergePluginConfig';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('mergePluginConfig()', () => {
+  it('can generate correct config.', () => {
+    const plugins: any = [
+      {
+        visualSchema: {
+          widgets: { widget1: 'w1' },
+          schema: { [AdaptiveKinds.IfCondition]: 'widget1' },
+        },
+      },
+      {
+        visualSchema: {
+          widgets: { widget2: 'w2' },
+          schema: { [AdaptiveKinds.SwitchCondition]: 'widget2' },
+        },
+      },
+    ];
+    expect(mergePluginConfig(...plugins)).toMatchObject({
+      widgets: {
+        widget1: 'w1',
+        widget2: 'w2',
+      },
+      schema: {
+        [AdaptiveKinds.IfCondition]: 'widget1',
+        [AdaptiveKinds.SwitchCondition]: 'widget2',
+      },
+    });
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/ActionCard.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/ActionCard.test.tsx
new file mode 100644
index 0000000000..0ff082eae3
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/ActionCard.test.tsx
@@ -0,0 +1,31 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { ActionCard } from '../../../src/adaptive-flow-renderer/widgets';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('ActionCard', () => {
+  it('can be rendered.', () => {
+    const card = render(<ActionCard data={{ $kind: AdaptiveKinds.SendActivity }} id="test" onEvent={() => null} />);
+    expect(card).toBeTruthy();
+  });
+
+  it('can be rendered with injected content.', () => {
+    const card = render(
+      <ActionCard
+        body={<span data-testid="test-body">Body</span>}
+        data={{ $kind: AdaptiveKinds.SendActivity }}
+        footer={<span data-testid="test-footer">Footer</span>}
+        header={<span data-testid="test-header">Header</span>}
+        id="test"
+        onEvent={() => null}
+      />
+    );
+    expect(card.getByTestId('test-header')).toBeTruthy();
+    expect(card.getByTestId('test-body')).toBeTruthy();
+    expect(card.getByTestId('test-footer')).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/DialogRef.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/DialogRef.test.tsx
new file mode 100644
index 0000000000..477089175c
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/DialogRef.test.tsx
@@ -0,0 +1,36 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { DialogRef } from '../../../src/adaptive-flow-renderer/widgets';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('DialogRef', () => {
+  it('can be rendered.', () => {
+    const dialogRef = render(
+      <DialogRef data={{ $kind: AdaptiveKinds.BeginDialog }} dialog="" id="test" onEvent={() => null} />
+    );
+    expect(dialogRef).toBeTruthy();
+  });
+
+  it('can ref string dialog value correctly.', () => {
+    const dialogRef = render(
+      <DialogRef data={{ $kind: AdaptiveKinds.BeginDialog }} dialog="test-dialog" id="test" onEvent={() => null} />
+    );
+    expect(dialogRef.queryAllByText('test-dialog')).toHaveLength(1);
+  });
+
+  it('can ref object dialog value correctly.', () => {
+    const dialogRef = render(
+      <DialogRef
+        data={{ $kind: AdaptiveKinds.BeginDialog }}
+        dialog={{ $ref: 'test-dialog-obj' }}
+        id="test"
+        onEvent={() => null}
+      />
+    );
+    expect(dialogRef.queryAllByText('test-dialog-obj')).toHaveLength(1);
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/ForeachWidget.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/ForeachWidget.test.tsx
new file mode 100644
index 0000000000..c9f3f24d09
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/ForeachWidget.test.tsx
@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { ForeachWidget } from '../../../src/adaptive-flow-renderer/widgets';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('ForeachWidget', () => {
+  it('can be rendered correctly.', () => {
+    const foreachNode = render(
+      <ForeachWidget
+        data={{ $kind: AdaptiveKinds.Foreach }}
+        id="test"
+        loop={<span data-testid="test-loop">Loop Head</span>}
+        onEvent={() => null}
+      />
+    );
+    expect(foreachNode).toBeTruthy();
+    expect(foreachNode.getByTestId('test-loop')).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/IfConditionWidget.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/IfConditionWidget.test.tsx
new file mode 100644
index 0000000000..a6a872cf52
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/IfConditionWidget.test.tsx
@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { IfConditionWidget } from '../../../src/adaptive-flow-renderer/widgets';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('IfConditionWidget', () => {
+  it('can be rendered correctly.', () => {
+    const ifCondition = render(
+      <IfConditionWidget
+        data={{ $kind: AdaptiveKinds.IfCondition }}
+        id="test"
+        judgement={<span data-testid="test-judgement">Condition Judgement</span>}
+        onEvent={() => null}
+      />
+    );
+    expect(ifCondition).toBeTruthy();
+    expect(ifCondition.getByTestId('test-judgement')).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/PromptWidget.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/PromptWidget.test.tsx
new file mode 100644
index 0000000000..5d3c89a208
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/PromptWidget.test.tsx
@@ -0,0 +1,25 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { PromptWidget } from '../../../src/adaptive-flow-renderer/widgets';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('PromptWidget', () => {
+  it('can be rendered correctly.', () => {
+    const promptNode = render(
+      <PromptWidget
+        botAsks={<span data-testid="test-botAsks">BotAsks</span>}
+        data={{ $kind: AdaptiveKinds.TextInput }}
+        id="test"
+        userInput={<span data-testid="test-userInput">UserInput</span>}
+        onEvent={() => null}
+      />
+    );
+    expect(promptNode).toBeTruthy();
+    expect(promptNode.getByTestId('test-botAsks')).toBeTruthy();
+    expect(promptNode.getByTestId('test-userInput')).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/SwitchConditionWidget.test.tsx b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/SwitchConditionWidget.test.tsx
new file mode 100644
index 0000000000..89c65e65a5
--- /dev/null
+++ b/Composer/packages/extensions/adaptive-flow/__tests__/adaptive-flow-renderer/widgets/SwitchConditionWidget.test.tsx
@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import React from 'react';
+import { render } from '@bfc/test-utils';
+
+import { SwitchConditionWidget } from '../../../src/adaptive-flow-renderer/widgets';
+import { AdaptiveKinds } from '../../../src/adaptive-flow-renderer/constants/AdaptiveKinds';
+
+describe('SwitchConditionWidget', () => {
+  it('can be rendered correctly.', () => {
+    const switchCondition = render(
+      <SwitchConditionWidget
+        data={{ $kind: AdaptiveKinds.SwitchCondition }}
+        id="test"
+        judgement={<span data-testid="test-judgement">Condition Judgement</span>}
+        onEvent={() => null}
+      />
+    );
+    expect(switchCondition).toBeTruthy();
+    expect(switchCondition.getByTestId('test-judgement')).toBeTruthy();
+  });
+});
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/AdaptiveFlowEditor.tsx b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/AdaptiveFlowEditor.tsx
index 0118d98d45..810167bdb5 100644
--- a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/AdaptiveFlowEditor.tsx
+++ b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/AdaptiveFlowEditor.tsx
@@ -21,7 +21,7 @@ import { mergePluginConfig } from './utils/mergePluginConfig';
 import { getCustomSchema } from './utils/getCustomSchema';
 import { SelectionContext } from './contexts/SelectionContext';
 import { KeyboardZone } from './components/KeyboardZone';
-import { mapKeyboardCommandToEditorEvent } from './utils/mapKeyboardCommandToEditorEvent.ts';
+import { mapKeyboardCommandToEditorEvent } from './utils/mapKeyboardCommandToEditorEvent';
 import { useSelectionEffect } from './hooks/useSelectionEffect';
 import { useEditorEventApi } from './hooks/useEditorEventApi';
 import {
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/components/KeyboardZone.tsx b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/components/KeyboardZone.tsx
index 788e76d8ae..7f97598d45 100644
--- a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/components/KeyboardZone.tsx
+++ b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/components/KeyboardZone.tsx
@@ -62,7 +62,7 @@ export const KeyboardZone = React.forwardRef<HTMLDivElement, KeyboardZoneProps>(
   };
 
   return (
-    <div ref={ref} css={styles} data-test-id="keyboard-zone" tabIndex={0} onKeyDown={handleKeyDown}>
+    <div ref={ref} css={styles} data-testid="keyboard-zone" tabIndex={0} onKeyDown={handleKeyDown}>
       {children}
     </div>
   );
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/NodeRendererContext.ts b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/NodeRendererContext.ts
index 4a8cc7d126..d78c9e64e5 100644
--- a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/NodeRendererContext.ts
+++ b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/NodeRendererContext.ts
@@ -13,11 +13,12 @@ export interface NodeRendererContextValue {
   customSchemas: OBISchema[];
 }
 
-export const NodeRendererContext = React.createContext<NodeRendererContextValue>({
+export const defaultRendererContextValue = {
   focusedId: '',
   focusedEvent: '',
   focusedTab: '',
   clipboardActions: [],
-  dialogFactory: new DialogFactory(),
+  dialogFactory: new DialogFactory({}),
   customSchemas: [],
-});
+};
+export const NodeRendererContext = React.createContext<NodeRendererContextValue>(defaultRendererContextValue);
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/SelectionContext.ts b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/SelectionContext.ts
index a7b576af26..c3a48ee572 100644
--- a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/SelectionContext.ts
+++ b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/contexts/SelectionContext.ts
@@ -13,10 +13,11 @@ export interface SelectionContextData {
   selectableElements: SelectorElement[];
 }
 
-export const SelectionContext = React.createContext<SelectionContextData>({
+export const defaultSelectionContextValue = {
   getNodeIndex: (_: string): number => 0,
   getSelectableIds: () => [],
   selectedIds: [] as string[],
   setSelectedIds: () => null,
   selectableElements: [],
-});
+};
+export const SelectionContext = React.createContext<SelectionContextData>(defaultSelectionContextValue);
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/hooks/useWindowDimensions.ts b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/hooks/useWindowDimensions.ts
deleted file mode 100644
index 6707cce97a..0000000000
--- a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/hooks/useWindowDimensions.ts
+++ /dev/null
@@ -1,25 +0,0 @@
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-
-import { useState, useEffect, useRef } from 'react';
-import debounce from 'lodash/debounce';
-
-const getWindowDimensions = () => {
-  const { innerWidth: width, innerHeight: height } = window;
-  return {
-    width,
-    height,
-  };
-};
-
-export const useWindowDimensions = () => {
-  const [windowDimensions, setWindowDimensions] = useState(getWindowDimensions());
-  const handleResize = useRef(debounce(() => setWindowDimensions(getWindowDimensions()), 200)).current;
-
-  useEffect(() => {
-    window.addEventListener('resize', handleResize);
-    return () => window.removeEventListener('resize', handleResize);
-  }, []);
-
-  return windowDimensions;
-};
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/renderers/NodeWrapper.tsx b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/renderers/NodeWrapper.tsx
index 3255718da8..c8799f64df 100644
--- a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/renderers/NodeWrapper.tsx
+++ b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/renderers/NodeWrapper.tsx
@@ -74,6 +74,7 @@ export const ActionNodeWrapper: FC<NodeWrapperProps> = ({ id, tab, data, onEvent
           ${!nodeFocused && nodeBorderHoveredStyle}
         }
       `}
+      data-testid="ActionNodeWrapper"
       {...declareElementAttributes(selectableId, id)}
       aria-label={generateSDKTitle(data, '', tab)}
       onClick={(e) => {
diff --git a/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.ts.ts b/Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.ts
similarity index 100%
rename from Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.ts.ts
rename to Composer/packages/extensions/adaptive-flow/src/adaptive-flow-editor/utils/mapKeyboardCommandToEditorEvent.ts
diff --git a/Composer/packages/extensions/extension/src/index.ts b/Composer/packages/extensions/extension/src/index.ts
index 3ecebc1a7c..931425d29c 100644
--- a/Composer/packages/extensions/extension/src/index.ts
+++ b/Composer/packages/extensions/extension/src/index.ts
@@ -1,9 +1,12 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { Extension } from './components';
+import extensionContext from './extensionContext';
 
 export * from './components';
 export * from './hooks';
 export * from './types';
 
 export default Extension;
+
+export const ExtensionContext = extensionContext;