@finos/legend-application-studio 26.1.8 → 26.1.9

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

@@ -25,19 +25,17 @@ import {
   type RelationalDatabaseConnection,
   type XmlModelConnection,
   type Runtime,
-  type IdentifiedConnection,
   type TestAssertion,
   type AtomicTest,
   type Class,
   ExternalFormatData,
   RelationalCSVData,
   ConnectionTestData,
-  EngineRuntime,
-  RuntimePointer,
   EqualToJson,
   DEFAULT_TEST_ASSERTION_PREFIX,
   RelationalCSVDataTable,
   type INTERNAL__UnknownConnection,
+  getAllIdentifiedConnectionsFromRuntime,
 } from '@finos/legend-graph';
 import {
   assertTrue,
@@ -60,6 +58,9 @@ export const validateTestableId = (
   id: string | undefined,
   possibleIds: string[] | undefined,
 ): string | undefined => {
+  if (id === undefined) {
+    return undefined;
+  }
   if (!id) {
     return 'ID is required';
   } else if (id.includes(' ')) {
@@ -90,22 +91,6 @@ export const createDefaultEqualToJSONTestAssertion = (
   return assertion;
 };
 
-export const getAllIdentifiedConnectionsFromRuntime = (
-  runtime: Runtime,
-): IdentifiedConnection[] => {
-  const resolvedRuntimes: EngineRuntime[] = [];
-  if (runtime instanceof RuntimePointer) {
-    resolvedRuntimes.push(runtime.packageableRuntime.value.runtimeValue);
-  } else if (runtime instanceof EngineRuntime) {
-    resolvedRuntimes.push(runtime);
-  }
-  return resolvedRuntimes
-    .flatMap((e) =>
-      e.connections.map((connection) => connection.storeConnections),
-    )
-    .flat();
-};
-
 export const createEmbeddedDataFromClass = (
   _class: Class,
   editorStore: EditorStore,

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

@@ -24,10 +24,14 @@ import {
   type DataElementReference,
   type RelationalCSVDataTable,
   type RelationalCSVData,
+  type ModelData,
+  type PackageableElementReference,
+  type ModelEmbeddedData,
   observe_EmbeddedData,
   observe_DataElement,
   observe_RelationalDataTable,
   PackageableElementExplicitReference,
+  observe_ModelData,
 } from '@finos/legend-graph';
 import { addUniqueEntry, deleteEntry } from '@finos/legend-shared';
 import { action } from 'mobx';
@@ -54,12 +58,6 @@ export const externalFormatData_setData = action(
   },
 );
 
-export const modelStoreData_setInstance = action(
-  (modelStoreData: ModelStoreData, value: Map<Class, object>): void => {
-    modelStoreData.instances = value;
-  },
-);
-
 export const relationalData_setTableValues = action(
   (data: RelationalCSVDataTable, values: string): void => {
     data.values = values;
@@ -102,3 +100,44 @@ export const dataElementReference_setDataElement = action(
       );
   },
 );
+
+// Model Store
+export const modelStoreData_addModelData = action(
+  (
+    modelStoreData: ModelStoreData,
+    value: ModelData,
+    context: ObserverContext,
+  ): void => {
+    const entries = modelStoreData.modelData ?? [];
+    if (!modelStoreData.modelData) {
+      modelStoreData.modelData = entries;
+    }
+    addUniqueEntry(entries, observe_ModelData(value, context));
+  },
+);
+
+export const modelStoreData_setDataModelModel = action(
+  (modelData: ModelData, val: PackageableElementReference<Class>): void => {
+    modelData.model = val;
+  },
+);
+
+export const modelStoreData_deleteModelData = action(
+  (modelStoreData: ModelStoreData, value: ModelData): void => {
+    if (modelStoreData.modelData) {
+      deleteEntry(modelStoreData.modelData, value);
+    }
+  },
+);
+
+export const modelStoreData_setModelDataClass = action(
+  (modelData: ModelData, val: PackageableElementReference<Class>): void => {
+    modelData.model = val;
+  },
+);
+
+export const modelStoreData_setModelDataData = action(
+  (modelData: ModelEmbeddedData, val: EmbeddedData): void => {
+    modelData.data = val;
+  },
+);

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

@@ -204,7 +204,7 @@ export const mapping_deleteTestSuite = action(
 
 export const mappingTestable_setQuery = action(
   (test: MappingQueryTest | MappingQueryTestSuite, value: RawLambda): void => {
-    test.query = observe_RawLambda(value);
+    test.func = observe_RawLambda(value);
   },
 );
 

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

@@ -309,6 +309,12 @@ export const snowflakeDatasourceSpec_setQuotedIdentifiersIgnoreCase = action(
   },
 );
 
+export const snowflakeDatasourceSpec_setEnableQueryTags = action(
+  (spec: SnowflakeDatasourceSpecification, val: boolean | undefined): void => {
+    spec.enableQueryTags = val;
+  },
+);
+
 export const snowflakeDatasourceSpec_setProxyHost = action(
   (spec: SnowflakeDatasourceSpecification, val: string | undefined): void => {
     spec.proxyHost = val;