@finos/legend-application-studio 23.1.6 → 23.1.8

package/src/components/editor/side-bar/Explorer.tsx

@@ -106,10 +106,9 @@ import {
   getElementRootPackage,
 } from '@finos/legend-graph';
 import {
-  EDITOR_LANGUAGE,
+  CODE_EDITOR_LANGUAGE,
   getPackageableElementOptionFormatter,
   type PackageableElementOption,
-  TextInputEditor,
   useApplicationStore,
 } from '@finos/legend-application';
 import { PACKAGEABLE_ELEMENT_TYPE } from '../../../stores/editor/shared/ModelClassifierUtils.js';
@@ -124,6 +123,7 @@ import {
   CLASS_MOCK_DATA_GENERATION_FORMAT,
   createMockDataForClassWithFormat,
 } from '../../../stores/editor/shared/MockDataUtils.js';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const ElementRenamer = observer(() => {
   const editorStore = useEditorStore();
@@ -266,16 +266,16 @@ const GENERATION_DEFAULT_DEPTH = 100;
 const GENERATION_DEFAULT_FORMAT = CLASS_MOCK_DATA_GENERATION_FORMAT.JSON;
 const getMockDataEditorLanguage = (
   format: CLASS_MOCK_DATA_GENERATION_FORMAT,
-): EDITOR_LANGUAGE => {
+): CODE_EDITOR_LANGUAGE => {
   switch (format) {
     case CLASS_MOCK_DATA_GENERATION_FORMAT.JSON:
-      return EDITOR_LANGUAGE.JSON;
+      return CODE_EDITOR_LANGUAGE.JSON;
     case CLASS_MOCK_DATA_GENERATION_FORMAT.XML:
-      return EDITOR_LANGUAGE.XML;
+      return CODE_EDITOR_LANGUAGE.XML;
     case CLASS_MOCK_DATA_GENERATION_FORMAT.YAML:
-      return EDITOR_LANGUAGE.YAML;
+      return CODE_EDITOR_LANGUAGE.YAML;
     default:
-      return EDITOR_LANGUAGE.TEXT;
+      return CODE_EDITOR_LANGUAGE.TEXT;
   }
 };
 
@@ -416,7 +416,7 @@ const SampleDataGenerator = observer(() => {
           />
         </div>
         <div className="sample-data-generator__result">
-          <TextInputEditor
+          <CodeEditor
             inputValue={resultText}
             isReadOnly={true}
             hideGutter={true}

package/src/components/editor/side-bar/WorkflowManager.tsx

@@ -48,8 +48,7 @@ import { LEGEND_STUDIO_TEST_ID } from '../../../application/LegendStudioTesting.
 import { flowResult } from 'mobx';
 import { WorkflowJobStatus, WorkflowStatus } from '@finos/legend-server-sdlc';
 import {
-  EDITOR_LANGUAGE,
-  TextInputEditor,
+  CODE_EDITOR_LANGUAGE,
   useApplicationStore,
 } from '@finos/legend-application';
 import {
@@ -66,6 +65,7 @@ import {
   guaranteeType,
   isNonNullable,
 } from '@finos/legend-shared';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const getWorkflowStatusIcon = (
   workflowStatus: WorkflowStatus,
@@ -238,10 +238,10 @@ const WorkflowJobLogsViewer = observer(
             </ModalHeaderActions>
           </ModalHeader>
           <ModalBody>
-            <TextInputEditor
+            <CodeEditor
               inputValue={logs}
               isReadOnly={true}
-              language={EDITOR_LANGUAGE.TEXT}
+              language={CODE_EDITOR_LANGUAGE.TEXT}
               showMiniMap={true}
             />
           </ModalBody>

package/src/components/editor/side-bar/testable/GlobalTestRunner.tsx

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { EDITOR_LANGUAGE, TextInputEditor } from '@finos/legend-application';
+import { CODE_EDITOR_LANGUAGE } from '@finos/legend-application';
 import {
   type TreeData,
   type TreeNodeContainerProps,
@@ -78,6 +78,7 @@ import { TextDiffView } from '../../../shared/DiffView.js';
 import { getElementTypeIcon } from '../../../shared/ElementIconUtils.js';
 import { UnsupportedEditorPanel } from '../../edit-panel/UnsupportedElementEditor.js';
 import { useEditorStore } from '../../EditorStoreProvider.js';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 export const getTestableResultIcon = (
   testableResult: TESTABLE_RESULT,
@@ -189,26 +190,26 @@ const TestFailViewer = observer(
           <ModalHeader title={id} />
           <ModalBody>
             {failure instanceof TestError && (
-              <TextInputEditor
+              <CodeEditor
                 inputValue={failure.error}
                 isReadOnly={true}
-                language={EDITOR_LANGUAGE.TEXT}
+                language={CODE_EDITOR_LANGUAGE.TEXT}
                 showMiniMap={true}
               />
             )}
             {failure instanceof EqualToJsonAssertFail && (
               <TextDiffView
-                language={EDITOR_LANGUAGE.JSON}
+                language={CODE_EDITOR_LANGUAGE.JSON}
                 from={failure.expected}
                 to={failure.actual}
               />
             )}
             {failure instanceof AssertFail &&
               !(failure instanceof EqualToJsonAssertFail) && (
-                <TextInputEditor
+                <CodeEditor
                   inputValue={failure.message ?? ''}
                   isReadOnly={true}
-                  language={EDITOR_LANGUAGE.TEXT}
+                  language={CODE_EDITOR_LANGUAGE.TEXT}
                 />
               )}
           </ModalBody>

package/src/components/extensions/Core_LegendStudioApplicationPlugin.tsx

@@ -21,6 +21,8 @@ import {
   type ContextualDocumentationEntry,
   type SettingConfigurationEntry,
   collectSettingConfigurationEntriesFromConfig,
+  type LegendApplicationSetup,
+  setupPureLanguageService,
 } from '@finos/legend-application';
 import packageJson from '../../../package.json';
 import { LEGEND_STUDIO_APPLICATION_NAVIGATION_CONTEXT_KEY } from '../../application/LegendStudioApplicationNavigationContext.js';
@@ -31,6 +33,7 @@ import {
 import { LegendStudioApplicationPlugin } from '../../stores/LegendStudioApplicationPlugin.js';
 import { LEGEND_STUDIO_COMMAND_CONFIG } from '../../application/LegendStudioCommand.js';
 import { LEGEND_STUDIO_SETTING_CONFIG } from '../../application/LegendStudioSetting.js';
+import type { LegendStudioApplicationStore } from '../../stores/LegendStudioBaseStore.js';
 
 export class Core_LegendStudioApplicationPlugin extends LegendStudioApplicationPlugin {
   static NAME = packageJson.extensions.applicationStudioPlugin;
@@ -39,6 +42,24 @@ export class Core_LegendStudioApplicationPlugin extends LegendStudioApplicationP
     super(Core_LegendStudioApplicationPlugin.NAME, packageJson.version);
   }
 
+  override getExtraApplicationSetups(): LegendApplicationSetup[] {
+    return [
+      async (applicationStore) => {
+        setupPureLanguageService(
+          // TODO: move these inside of `DSL_LegendStudioApplicationPlugin_Extension`
+          (applicationStore as LegendStudioApplicationStore).pluginManager
+            .getPureGraphManagerPlugins()
+            .flatMap((plugin) => plugin.getExtraPureGrammarKeywords?.() ?? []),
+          (applicationStore as LegendStudioApplicationStore).pluginManager
+            .getPureGraphManagerPlugins()
+            .flatMap(
+              (plugin) => plugin.getExtraPureGrammarParserNames?.() ?? [],
+            ),
+        );
+      },
+    ];
+  }
+
   override getExtraKeyedCommandConfigEntries(): KeyedCommandConfigEntry[] {
     return collectKeyedCommandConfigEntriesFromConfig(
       LEGEND_STUDIO_COMMAND_CONFIG,

package/src/components/shared/DiffView.tsx

@@ -18,16 +18,12 @@ import { useEffect, useState, useRef } from 'react';
 import { observer } from 'mobx-react-lite';
 import { editor as monacoEditorAPI } from 'monaco-editor';
 import {
-  EDITOR_THEME,
-  EDITOR_LANGUAGE,
+  CODE_EDITOR_THEME,
+  CODE_EDITOR_LANGUAGE,
   TAB_SIZE,
   useApplicationStore,
 } from '@finos/legend-application';
-import {
-  disposeDiffEditor,
-  getBaseTextEditorOptions,
-  useResizeDetector,
-} from '@finos/legend-art';
+import { useResizeDetector } from '@finos/legend-art';
 import {
   isString,
   stringifyLosslessJSON,
@@ -35,10 +31,14 @@ import {
   tryToFormatLosslessJSONString,
 } from '@finos/legend-shared';
 import { useEditorStore } from '../editor/EditorStoreProvider.js';
+import {
+  disposeDiffCodeEditor,
+  getBaseCodeEditorOptions,
+} from '@finos/legend-lego/code-editor';
 
 export const TextDiffView = observer(
   (props: {
-    language: EDITOR_LANGUAGE;
+    language: CODE_EDITOR_LANGUAGE;
     from?: string | undefined;
     to?: string | undefined;
   }) => {
@@ -63,8 +63,8 @@ export const TextDiffView = observer(
       if (!editor && editorRef.current) {
         const element = editorRef.current;
         const _editor = monacoEditorAPI.createDiffEditor(element, {
-          ...getBaseTextEditorOptions(),
-          theme: EDITOR_THEME.LEGEND,
+          ...getBaseCodeEditorOptions(),
+          theme: CODE_EDITOR_THEME.LEGEND,
           readOnly: true,
         });
         setEditor(_editor);
@@ -83,15 +83,15 @@ export const TextDiffView = observer(
     useEffect(
       () => (): void => {
         if (editor) {
-          disposeDiffEditor(editor);
+          disposeDiffCodeEditor(editor);
         }
       },
       [editor],
     ); // dispose editor
 
     return (
-      <div ref={ref} className="text-editor__container">
-        <div className="text-editor__body" ref={editorRef} />
+      <div ref={ref} className="code-editor__container">
+        <div className="code-editor__body" ref={editorRef} />
       </div>
     );
   },
@@ -121,7 +121,7 @@ export const JsonDiffView = observer(
 
     return (
       <TextDiffView
-        language={EDITOR_LANGUAGE.JSON}
+        language={CODE_EDITOR_LANGUAGE.JSON}
         from={formatJSONLikeValue(from, Boolean(lossless))}
         to={formatJSONLikeValue(to, Boolean(lossless))}
       />

package/src/components/workspace-setup/CreateProjectModal.tsx

@@ -34,14 +34,12 @@ import {
 import { LEGEND_STUDIO_TEST_ID } from '../../application/LegendStudioTesting.js';
 import { isNumber } from '@finos/legend-shared';
 import { flowResult } from 'mobx';
-import {
-  DocumentationLink,
-  useConditionedApplicationNavigationContext,
-} from '@finos/legend-application';
+import { useConditionedApplicationNavigationContext } from '@finos/legend-application';
 import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../application/LegendStudioDocumentation.js';
 import { LEGEND_STUDIO_APPLICATION_NAVIGATION_CONTEXT_KEY } from '../../application/LegendStudioApplicationNavigationContext.js';
 import { useLegendStudioApplicationStore } from '../LegendStudioBaseStoreProvider.js';
 import { useWorkspaceSetupStore } from './WorkspaceSetup.js';
+import { DocumentationLink } from '@finos/legend-application/components';
 
 enum CREATE_PROJECT_MODAL_TAB {
   CREATE = 'CREATE',

package/src/components/workspace-setup/CreateWorkspaceModal.tsx

@@ -28,12 +28,10 @@ import {
 } from '@finos/legend-art';
 import { flowResult } from 'mobx';
 import { type Project, WorkspaceType } from '@finos/legend-server-sdlc';
-import {
-  useApplicationStore,
-  DocumentationLink,
-} from '@finos/legend-application';
+import { useApplicationStore } from '@finos/legend-application';
 import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../application/LegendStudioDocumentation.js';
 import { useWorkspaceSetupStore } from './WorkspaceSetup.js';
+import { DocumentationLink } from '@finos/legend-application/components';
 
 export const CreateWorkspaceModal = observer(
   (props: { selectedProject: Project }) => {

package/src/components/workspace-setup/WorkspaceSetup.tsx

@@ -34,7 +34,6 @@ import {
 } from '../../application/LegendStudioNavigation.js';
 import { flowResult } from 'mobx';
 import {
-  DocumentationLink,
   useApplicationNavigationContext,
   useParams,
 } from '@finos/legend-application';
@@ -57,6 +56,7 @@ import {
 import { debounce, guaranteeNonNullable } from '@finos/legend-shared';
 import { WorkspaceSetupStore } from '../../stores/workspace-setup/WorkspaceSetupStore.js';
 import { useSDLCServerClient } from '@finos/legend-server-sdlc';
+import { DocumentationLink } from '@finos/legend-application/components';
 
 const WorkspaceSetupStoreContext = createContext<
   WorkspaceSetupStore | undefined

package/src/index.ts

@@ -80,7 +80,7 @@ export * from './stores/editor/EditorStoreTestUtils.js';
 
 export { PostProcessorEditorState } from './stores/editor/editor-state/element-editor-state/connection/PostProcessorEditorState.js';
 export { MappingExecutionState } from './stores/editor/editor-state/element-editor-state/mapping/MappingExecutionState.js';
-export { MappingTestState } from './stores/editor/editor-state/element-editor-state/mapping/MappingTestState.js';
+export { DEPRECATED__MappingTestState } from './stores/editor/editor-state/element-editor-state/mapping/MappingTestState.js';
 export {
   ConnectionValueState,
   RelationalDatabaseConnectionValueState,

package/src/stores/editor/EditorGraphState.ts

@@ -91,14 +91,12 @@ import {
   createGraphBuilderReport,
   ExecutionEnvironmentInstance,
 } from '@finos/legend-graph';
-import {
-  LegendApplicationTelemetryHelper,
-  type TabState,
-} from '@finos/legend-application';
+import { LegendApplicationTelemetryHelper } from '@finos/legend-application';
 import { CONFIGURATION_EDITOR_TAB } from './editor-state/project-configuration-editor-state/ProjectConfigurationEditorState.js';
 import { PACKAGEABLE_ELEMENT_TYPE } from './shared/ModelClassifierUtils.js';
 import { LEGEND_STUDIO_APP_EVENT } from '../../application/LegendStudioEvent.js';
 import { LEGEND_STUDIO_SETTING_KEY } from '../../application/LegendStudioSetting.js';
+import type { TabState } from '@finos/legend-application/components';
 
 export enum GraphBuilderStatus {
   SUCCEEDED = 'SUCCEEDED',

package/src/stores/editor/EditorTabManagerState.ts

@@ -14,7 +14,6 @@
  * limitations under the License.
  */
 
-import { TabManagerState } from '@finos/legend-application';
 import {
   Association,
   Class,
@@ -57,6 +56,7 @@ import { GenerationSpecificationEditorState } from './editor-state/GenerationSpe
 import { UnsupportedElementEditorState } from './editor-state/UnsupportedElementEditorState.js';
 import type { EditorStore } from './EditorStore.js';
 import type { DSL_LegendStudioApplicationPlugin_Extension } from '../LegendStudioApplicationPlugin.js';
+import { TabManagerState } from '@finos/legend-application/components';
 
 export class EditorTabManagerState extends TabManagerState {
   readonly editorStore: EditorStore;

package/src/stores/editor/GraphEditFormModeState.ts

@@ -14,7 +14,6 @@
  * limitations under the License.
  */
 
-import type { TabState } from '@finos/legend-application';
 import {
   type PackageableElement,
   isElementReadOnly,
@@ -55,6 +54,7 @@ import {
 import { ElementEditorState } from './editor-state/element-editor-state/ElementEditorState.js';
 import { LegendStudioTelemetryHelper } from '../../application/LegendStudioTelemetryHelper.js';
 import { GraphEditorMode } from './GraphEditorMode.js';
+import type { TabState } from '@finos/legend-application/components';
 
 export class GraphEditFormModeState extends GraphEditorMode {
   *initialize(): GeneratorFn<void> {

package/src/stores/editor/editor-state/EditorState.ts

@@ -14,8 +14,8 @@
  * limitations under the License.
  */
 
+import { TabState } from '@finos/legend-application/components';
 import type { EditorStore } from '../EditorStore.js';
-import { TabState } from '@finos/legend-application';
 
 export abstract class EditorState extends TabState {
   readonly editorStore: EditorStore;

package/src/stores/editor/editor-state/FileGenerationViewerState.ts

@@ -18,7 +18,7 @@ import type { EditorStore } from '../EditorStore.js';
 import { EditorState } from './EditorState.js';
 import { observable, makeObservable, computed } from 'mobx';
 import type { FileSystem_File } from '../shared/FileSystemTreeUtils.js';
-import { EDITOR_LANGUAGE, TAB_SIZE } from '@finos/legend-application';
+import { CODE_EDITOR_LANGUAGE, TAB_SIZE } from '@finos/legend-application';
 import { returnUndefOnError } from '@finos/legend-shared';
 
 export const getTextContent = (
@@ -26,7 +26,7 @@ export const getTextContent = (
   format: string | undefined,
 ): string => {
   switch (format?.toLowerCase()) {
-    case EDITOR_LANGUAGE.JSON:
+    case CODE_EDITOR_LANGUAGE.JSON:
       return (
         returnUndefOnError(() =>
           JSON.stringify(JSON.parse(content), undefined, TAB_SIZE),
@@ -39,14 +39,14 @@ export const getTextContent = (
 
 export const getEditorLanguageForFormat = (
   format?: string,
-): EDITOR_LANGUAGE => {
+): CODE_EDITOR_LANGUAGE => {
   switch (format?.toLowerCase()) {
-    case EDITOR_LANGUAGE.JAVA:
-      return EDITOR_LANGUAGE.JAVA;
-    case EDITOR_LANGUAGE.JSON:
-      return EDITOR_LANGUAGE.JSON;
+    case CODE_EDITOR_LANGUAGE.JAVA:
+      return CODE_EDITOR_LANGUAGE.JAVA;
+    case CODE_EDITOR_LANGUAGE.JSON:
+      return CODE_EDITOR_LANGUAGE.JSON;
     default:
-      return EDITOR_LANGUAGE.TEXT;
+      return CODE_EDITOR_LANGUAGE.TEXT;
   }
 };
 

package/src/stores/editor/editor-state/GrammarTextEditorState.ts

@@ -44,9 +44,9 @@ import {
   PURE_CONNECTION_NAME,
   ExecutionEnvironmentInstance,
 } from '@finos/legend-graph';
-import type { TextEditorPosition } from '@finos/legend-art';
 import { generatePackageableElementTreeNodeDataLabel } from '../shared/PackageTreeUtils.js';
 import { LEGEND_STUDIO_SETTING_KEY } from '../../../application/LegendStudioSetting.js';
+import type { CodeEditorPosition } from '@finos/legend-lego/code-editor';
 
 const getGrammarElementTypeLabelRegexString = (
   typeLabel: string,
@@ -71,7 +71,7 @@ export class GrammarTextEditorState {
   graphGrammarText = '';
   currentElementLabelRegexString?: string | undefined;
   wrapText: boolean;
-  forcedCursorPosition?: TextEditorPosition | undefined;
+  forcedCursorPosition?: CodeEditorPosition | undefined;
 
   constructor(editorStore: EditorStore) {
     makeObservable(this, {
@@ -105,7 +105,7 @@ export class GrammarTextEditorState {
     this.wrapText = val;
   }
 
-  setForcedCursorPosition(position: TextEditorPosition | undefined): void {
+  setForcedCursorPosition(position: CodeEditorPosition | undefined): void {
     this.forcedCursorPosition = position;
   }
 

package/src/stores/editor/editor-state/element-editor-state/mapping/MappingEditorState.ts

@@ -31,7 +31,7 @@ import { PureInstanceSetImplementationState } from './PureInstanceSetImplementat
 import { ElementEditorState } from '../../../editor-state/element-editor-state/ElementEditorState.js';
 import {
   MAPPING_TEST_EDITOR_TAB_TYPE,
-  MappingTestState,
+  DEPRECATED__MappingTestState,
   TEST_RESULT,
 } from './MappingTestState.js';
 import { createMockDataForMappingElementSource } from '../../../shared/MockDataUtils.js';
@@ -62,6 +62,7 @@ import {
   type InputData,
   type Type,
   type EmbeddedSetImplementation,
+  type ExecutionResult,
   getAllClassMappings,
   GRAPH_MANAGER_EVENT,
   fromElementPathToMappingElementId,
@@ -582,7 +583,7 @@ export class MappingEditorState extends ElementEditorState {
   mappingExplorerTreeData: TreeData<MappingExplorerTreeNodeData>;
   newMappingElementSpec?: MappingElementSpec | undefined;
 
-  mappingTestStates: MappingTestState[] = [];
+  mappingTestStates: DEPRECATED__MappingTestState[] = [];
   isRunningAllTests = false;
   allTestRunTime = 0;
 
@@ -622,7 +623,7 @@ export class MappingEditorState extends ElementEditorState {
     });
 
     this.mappingTestStates = this.mapping.test.map(
-      (t) => new MappingTestState(editorStore, t, this),
+      (t) => new DEPRECATED__MappingTestState(editorStore, t, this),
     );
     this.mappingExplorerTreeData = getMappingElementTreeData(
       this.mapping,
@@ -674,7 +675,7 @@ export class MappingEditorState extends ElementEditorState {
 
   *openTab(tabState: MappingEditorTabState): GeneratorFn<void> {
     if (tabState !== this.currentTabState) {
-      if (tabState instanceof MappingTestState) {
+      if (tabState instanceof DEPRECATED__MappingTestState) {
         yield flowResult(this.openTest(tabState.test));
       } else if (tabState instanceof MappingElementState) {
         this.openMappingElement(tabState.mappingElement, false);
@@ -1184,7 +1185,7 @@ export class MappingEditorState extends ElementEditorState {
             tabState.mappingElement.id.value,
           );
           return this.createMappingElementState(mappingElement);
-        } else if (tabState instanceof MappingTestState) {
+        } else if (tabState instanceof DEPRECATED__MappingTestState) {
           return mappingEditorState.mappingTestStates.find(
             (testState) => testState.test.name === tabState.test.name,
           );
@@ -1209,16 +1210,16 @@ export class MappingEditorState extends ElementEditorState {
           tabState instanceof MappingElementState &&
           tabState.mappingElement === currentlyOpenedMappingElement,
       );
-    } else if (this.currentTabState instanceof MappingTestState) {
+    } else if (this.currentTabState instanceof DEPRECATED__MappingTestState) {
       const currentlyOpenedMappingTest =
         mappingEditorState.mappingTestStates.find(
           (testState) =>
-            this.currentTabState instanceof MappingTestState &&
+            this.currentTabState instanceof DEPRECATED__MappingTestState &&
             testState.test.name === this.currentTabState.test.name,
         )?.test;
       mappingEditorState.currentTabState = this.openedTabStates.find(
         (tabState) =>
-          tabState instanceof MappingTestState &&
+          tabState instanceof DEPRECATED__MappingTestState &&
           tabState.test === currentlyOpenedMappingTest,
       );
     } else {
@@ -1350,7 +1351,8 @@ export class MappingEditorState extends ElementEditorState {
     const isOpened = Boolean(
       this.openedTabStates.find(
         (tabState) =>
-          tabState instanceof MappingTestState && tabState.test === test,
+          tabState instanceof DEPRECATED__MappingTestState &&
+          tabState.test === test,
       ),
     );
     const testState = this.mappingTestStates.find(
@@ -1363,14 +1365,16 @@ export class MappingEditorState extends ElementEditorState {
     if (
       !this.openedTabStates.find(
         (tabState) =>
-          tabState instanceof MappingTestState && tabState.test === test,
+          tabState instanceof DEPRECATED__MappingTestState &&
+          tabState.test === test,
       )
     ) {
       addUniqueEntry(this.openedTabStates, testState);
     }
     this.currentTabState = this.openedTabStates.find(
       (tabState) =>
-        tabState instanceof MappingTestState && tabState.test === test,
+        tabState instanceof DEPRECATED__MappingTestState &&
+        tabState.test === test,
     );
     yield flowResult(
       testState.onTestStateOpen(
@@ -1402,28 +1406,55 @@ export class MappingEditorState extends ElementEditorState {
   }
 
   *runTests(): GeneratorFn<void> {
-    const startTime = Date.now();
-    this.isRunningAllTests = true;
-    this.mappingTestStates.forEach((testState) =>
-      testState.resetTestRunStatus(),
-    );
-    yield Promise.all(
-      this.mappingTestStates.map((testState: MappingTestState) => {
-        // run non-skip tests, and reset all skipped tests
-        if (!testState.isSkipped) {
-          return testState.runTest();
-        }
-        testState.resetTestRunStatus();
-        return undefined;
-      }),
-    );
-    this.isRunningAllTests = false;
-    this.allTestRunTime = Date.now() - startTime;
+    try {
+      const startTime = Date.now();
+      this.isRunningAllTests = true;
+      this.mappingTestStates.forEach((testState) =>
+        testState.resetTestRunStatus(),
+      );
+      const input = this.mappingTestStates
+        .map((testState: DEPRECATED__MappingTestState) => {
+          // run non-skip tests, and reset all skipped tests
+          if (!testState.isSkipped) {
+            testState.setIsRunningTest(true);
+            return {
+              test: testState.test,
+              runtime: testState.inputDataState.runtime,
+              handleResult: (val: ExecutionResult) =>
+                testState.handleResult(val),
+              handleError: (error: Error) =>
+                testState.handleError(error, undefined),
+            };
+          }
+          testState.resetTestRunStatus();
+          return undefined;
+        })
+        .filter(isNonNullable);
+      yield this.editorStore.graphManagerState.graphManager.DEPRECATED__runLegacyMappingTests(
+        input,
+        this.mapping,
+        this.editorStore.graphManagerState.graph,
+        {
+          useLosslessParse: true,
+        },
+      );
+      this.editorStore.applicationStore.notificationService.notifySuccess(
+        'All mapping tests have been run',
+      );
+      this.allTestRunTime = Date.now() - startTime;
+    } catch (error) {
+      assertErrorThrown(error);
+      this.editorStore.applicationStore.notificationService.notifyError(
+        `Error running all mapping tests: ${error.message}`,
+      );
+    } finally {
+      this.isRunningAllTests = false;
+    }
   }
 
   *addTest(test: DEPRECATED__MappingTest): GeneratorFn<void> {
     this.mappingTestStates.push(
-      new MappingTestState(this.editorStore, test, this),
+      new DEPRECATED__MappingTestState(this.editorStore, test, this),
     );
     mapping_addTest(
       this.mapping,
@@ -1437,7 +1468,8 @@ export class MappingEditorState extends ElementEditorState {
     const matchMappingTestState = (
       tabState: MappingEditorTabState | undefined,
     ): boolean =>
-      tabState instanceof MappingTestState && tabState.test === test;
+      tabState instanceof DEPRECATED__MappingTestState &&
+      tabState.test === test;
     mapping_deleteTest(this.mapping, test);
     if (this.currentTabState && matchMappingTestState(this.currentTabState)) {
       yield flowResult(this.closeTab(this.currentTabState));
@@ -1505,7 +1537,7 @@ export class MappingEditorState extends ElementEditorState {
     );
     // open the test
     this.mappingTestStates.push(
-      new MappingTestState(this.editorStore, newTest, this),
+      new DEPRECATED__MappingTestState(this.editorStore, newTest, this),
     );
     yield flowResult(this.openTest(newTest));
   }

package/src/stores/editor/editor-state/element-editor-state/mapping/MappingExecutionState.ts

@@ -105,7 +105,6 @@ import {
 import {
   ActionAlertActionType,
   ActionAlertType,
-  ExecutionPlanState,
   TAB_SIZE,
 } from '@finos/legend-application';
 import {
@@ -133,6 +132,7 @@ import {
   LambdaEditorState,
   QueryBuilderTelemetryHelper,
   QUERY_BUILDER_EVENT,
+  ExecutionPlanState,
 } from '@finos/legend-query-builder';
 import { MappingEditorTabState } from './MappingTabManagerState.js';
 

package/src/stores/editor/editor-state/element-editor-state/mapping/MappingTabManagerState.ts

@@ -14,6 +14,6 @@
  * limitations under the License.
  */
 
-import { TabState } from '@finos/legend-application';
+import { TabState } from '@finos/legend-application/components';
 
 export abstract class MappingEditorTabState extends TabState {}