@finos/legend-application-studio 27.1.0 → 27.1.1

package/src/stores/editor/utils/TestableUtils.ts

@@ -28,6 +28,7 @@ import {
   type TestAssertion,
   type AtomicTest,
   type Class,
+  type EmbeddedDataVisitor,
   ExternalFormatData,
   RelationalCSVData,
   ConnectionTestData,
@@ -36,6 +37,14 @@ import {
   RelationalCSVDataTable,
   type INTERNAL__UnknownConnection,
   getAllIdentifiedConnectionsFromRuntime,
+  ModelStoreData,
+  ModelEmbeddedData,
+  PackageableElementExplicitReference,
+  type DataElementReference,
+  type INTERNAL__UnknownEmbeddedData,
+  type TestResult,
+  TestExecuted,
+  TestExecutionStatus,
 } from '@finos/legend-graph';
 import {
   assertTrue,
@@ -58,6 +67,7 @@ export const validateTestableId = (
   id: string | undefined,
   possibleIds: string[] | undefined,
 ): string | undefined => {
+  // undefined in this case means user has yet to write anything so we shouldnt show any error message
   if (id === undefined) {
     return undefined;
   }
@@ -101,6 +111,64 @@ export const createEmbeddedDataFromClass = (
   return data;
 };
 
+export const createBareModelStoreData = (
+  _class: Class,
+  editorStore: EditorStore,
+): ModelStoreData => {
+  const embeddedData = createEmbeddedDataFromClass(_class, editorStore);
+  const modelStoreData = new ModelStoreData();
+  const modelData = new ModelEmbeddedData();
+  modelData.data = embeddedData;
+  modelData.model = PackageableElementExplicitReference.create(_class);
+  modelStoreData.modelData = [modelData];
+  return modelStoreData;
+};
+
+export class EmbeddedDataCreatorFromEmbeddedData
+  implements EmbeddedDataVisitor<EmbeddedData>
+{
+  visit_EmbeddedData(data: EmbeddedData): EmbeddedData {
+    throw new Error('Method not implemented.');
+  }
+  visit_INTERNAL__UnknownEmbeddedData(
+    data: INTERNAL__UnknownEmbeddedData,
+  ): EmbeddedData {
+    throw new Error('Method not implemented.');
+  }
+  visit_ExternalFormatData(data: ExternalFormatData): EmbeddedData {
+    const val = new ExternalFormatData();
+    val.contentType = data.contentType;
+    val.data = '';
+    return val;
+  }
+  visit_ModelStoreData(data: ModelStoreData): EmbeddedData {
+    const val = new ModelStoreData();
+    val.modelData = data.modelData
+      ?.map((e) => {
+        if (e instanceof ModelEmbeddedData) {
+          const v = new ModelEmbeddedData();
+          v.model = PackageableElementExplicitReference.create(e.model.value);
+          v.data = e.data.accept_EmbeddedDataVisitor(
+            new EmbeddedDataCreatorFromEmbeddedData(),
+          );
+          return v;
+        }
+        return undefined;
+      })
+      .filter(isNonNullable);
+    return val;
+  }
+  visit_DataElementReference(data: DataElementReference): EmbeddedData {
+    return data.dataElement.value.data.accept_EmbeddedDataVisitor(
+      new EmbeddedDataCreatorFromEmbeddedData(),
+    );
+  }
+  visit_RelationalCSVData(data: RelationalCSVData): EmbeddedData {
+    const val = new RelationalCSVData();
+    return val;
+  }
+}
+
 // NOTE: this will all move to `engine` once engine support generating test data for all connections
 // Throws if unable to generate test data
 export class TEMPORARY__EmbeddedDataConnectionVisitor
@@ -251,3 +319,8 @@ export const TEMPORARY__createRelationalDataFromCSV = (
   });
   return data;
 };
+
+// test result
+export const isTestPassing = (testResult: TestResult): boolean =>
+  testResult instanceof TestExecuted &&
+  testResult.testExecutionStatus === TestExecutionStatus.PASS;

package/src/stores/graph-modifier/DSL_Mapping_GraphModifierHelper.ts

@@ -78,11 +78,10 @@ import {
   PrimitiveType,
   type MappingTestSuite,
   observe_MappingTestSuite,
-  type MappingQueryTestSuite,
-  type MappingQueryTest,
-  type MappingDataTestSuite,
-  type MappingDataTest,
   type StoreTestData,
+  type MappingTest,
+  type EmbeddedData,
+  observe_EmbeddedData,
 } from '@finos/legend-graph';
 import {
   addUniqueEntry,
@@ -203,25 +202,29 @@ export const mapping_deleteTestSuite = action(
 );
 
 export const mappingTestable_setQuery = action(
-  (test: MappingQueryTest | MappingQueryTestSuite, value: RawLambda): void => {
+  (test: MappingTestSuite, value: RawLambda): void => {
     test.func = observe_RawLambda(value);
   },
 );
 
 export const mappingTestable_deleteStoreTestData = action(
-  (
-    dataHolder: MappingDataTestSuite | MappingDataTest,
-    val: StoreTestData,
-  ): void => {
+  (dataHolder: MappingTest, val: StoreTestData): void => {
     deleteEntry(dataHolder.storeTestData, val);
   },
 );
 
-export const mappingTestable_addStoreTestData = action(
+export const mappingTestable_setEmbeddedData = action(
   (
-    dataHolder: MappingDataTestSuite | MappingDataTest,
-    val: StoreTestData,
+    store: StoreTestData,
+    embeddedData: EmbeddedData,
+    observerContext: ObserverContext,
   ): void => {
+    store.data = observe_EmbeddedData(embeddedData, observerContext);
+  },
+);
+
+export const mappingTestable_addStoreTestData = action(
+  (dataHolder: MappingTest, val: StoreTestData): void => {
     addUniqueEntry(dataHolder.storeTestData, val);
   },
 );

package/tsconfig.json

@@ -112,6 +112,7 @@
     "./src/stores/editor/editor-state/element-editor-state/mapping/UnsupportedInstanceSetImplementationState.ts",
     "./src/stores/editor/editor-state/element-editor-state/mapping/relational/RelationalInstanceSetImplementationState.ts",
     "./src/stores/editor/editor-state/element-editor-state/mapping/testable/MappingTestableState.ts",
+    "./src/stores/editor/editor-state/element-editor-state/mapping/testable/MappingTestingHelper.ts",
     "./src/stores/editor/editor-state/element-editor-state/service/ServiceEditorState.ts",
     "./src/stores/editor/editor-state/element-editor-state/service/ServiceExecutionState.ts",
     "./src/stores/editor/editor-state/element-editor-state/service/ServicePostValidationState.ts",