@finos/legend-application-studio 28.1.4 → 28.2.1

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

@@ -31,9 +31,8 @@ export class MappingExecutionQueryBuilderState extends QueryBuilderState {
     mapping: Mapping,
   ) {
     super(applicationStore, graphManagerState);
-
     this.executionMapping = mapping;
-    this.mapping = mapping;
+    this.executionContextState.mapping = mapping;
   }
 
   override get isMappingReadOnly(): boolean {

package/src/stores/editor/editor-state/element-editor-state/service/ServiceEditorState.ts

@@ -42,6 +42,7 @@ import {
   getValueSpecificationReturnType,
   type Type,
   resolveServiceQueryRawLambda,
+  PureExecution,
 } from '@finos/legend-graph';
 import { ServiceTestableState } from './testable/ServiceTestableState.js';
 import { User } from '@finos/legend-server-sdlc';
@@ -132,10 +133,13 @@ export class ServiceEditorState extends ElementEditorState {
         editorStore.sdlcServerClient.features.canCreateVersion,
     );
     this.testableState = new ServiceTestableState(editorStore, this);
-    const query = this.executionState.serviceExecutionParameters?.query;
+    const executionQuery =
+      this.service.execution instanceof PureExecution
+        ? this.service.execution.func
+        : undefined;
     // default to execution tab if query is defined
     this.selectedTab =
-      query && !isStubbed_RawLambda(query)
+      executionQuery && !isStubbed_RawLambda(executionQuery)
         ? SERVICE_TAB.EXECUTION
         : SERVICE_TAB.GENERAL;
     this.postValidationState = new ServicePostValidationsState(this);

package/src/stores/editor/editor-state/element-editor-state/service/testable/ServiceTestDataState.ts

@@ -23,6 +23,7 @@ import {
   type RawLambda,
   type DataElement,
   type Mapping,
+  type TestDataGenerationResult,
   ConnectionTestData,
   PureSingleExecution,
   PureMultiExecution,
@@ -37,6 +38,7 @@ import {
   getAllIdentifiedConnectionsFromRuntime,
   getAllIdentifiedServiceConnections,
   Database,
+  RuntimePointer,
 } from '@finos/legend-graph';
 import {
   type GeneratorFn,
@@ -47,6 +49,8 @@ import {
   guaranteeNonNullable,
   returnUndefOnError,
   getNullableFirstEntry,
+  assertType,
+  assertTrue,
 } from '@finos/legend-shared';
 import { action, flow, flowResult, makeObservable, observable } from 'mobx';
 import type { EditorStore } from '../../../../EditorStore.js';
@@ -156,6 +160,7 @@ export class ConnectionTestDataState {
   readonly connectionData: ConnectionTestData;
   readonly parametersState: ServiceTestDataParametersState;
   readonly generatingTestDataState = ActionState.create();
+  readonly generateSchemaQueryState = ActionState.create();
   useSharedModal = false;
 
   embeddedEditorState: EmbeddedDataEditorState;
@@ -167,6 +172,7 @@ export class ConnectionTestDataState {
   ) {
     makeObservable(this, {
       generatingTestDataState: observable,
+      generateSchemaQueryState: observable,
       embeddedEditorState: observable,
       useSharedModal: observable,
       anonymizeGeneratedData: observable,
@@ -175,6 +181,7 @@ export class ConnectionTestDataState {
       buildEmbeddedEditorState: action,
       generateTestData: flow,
       generateTestDataForDatabaseConnection: flow,
+      generateQuerySchemas: flow,
     });
     this.testDataState = testDataState;
     this.editorStore = testDataState.editorStore;
@@ -326,6 +333,60 @@ export class ConnectionTestDataState {
     }
   }
 
+  *generateQuerySchemas(): GeneratorFn<void> {
+    try {
+      this.generateSchemaQueryState.inProgress();
+      const connection = guaranteeNonNullable(
+        this.resolveConnectionValue(this.connectionData.connectionId),
+        `Unable to resolve connection ID '${this.connectionData.connectionId}`,
+      );
+      if (connection instanceof DatabaseConnection) {
+        const serviceExecutionParameters = guaranteeNonNullable(
+          this.testDataState.testSuiteState.testableState.serviceEditorState
+            .executionState.serviceExecutionParameters,
+        );
+        assertType(
+          serviceExecutionParameters.runtime,
+          RuntimePointer,
+          'Expected runtime type to be RuntimePointer',
+        );
+        const testDataGenerationResult =
+          (yield this.editorStore.graphManagerState.graphManager.generateTestData(
+            getExecutionQueryFromRawLambda(
+              serviceExecutionParameters.query,
+              this.parametersState.parameterStates,
+              this.editorStore.graphManagerState,
+            ),
+            serviceExecutionParameters.mapping.path,
+            serviceExecutionParameters.runtime.packageableRuntime.value.path,
+            this.editorStore.graphManagerState.graph,
+          )) as TestDataGenerationResult;
+        assertTrue(
+          testDataGenerationResult.data.length >= 1,
+          'Expected generated data to at least have one data',
+        );
+        service_setConnectionTestDataEmbeddedData(
+          this.connectionData,
+          guaranteeNonNullable(testDataGenerationResult.data[0]),
+          this.editorStore.changeDetectionState.observerContext,
+        );
+        this.embeddedEditorState = new EmbeddedDataEditorState(
+          this.testDataState.editorStore,
+          this.connectionData.testData,
+        );
+      }
+      this.generateSchemaQueryState.pass();
+    } catch (error) {
+      assertErrorThrown(error);
+      this.editorStore.applicationStore.notificationService.notifyError(
+        `Unable to generate query schemas: ${error.message}`,
+      );
+      this.generateSchemaQueryState.fail();
+    } finally {
+      this.generateSchemaQueryState.complete();
+    }
+  }
+
   changeEmbeddedData(val: EmbeddedData): void {
     service_setConnectionTestDataEmbeddedData(
       this.connectionData,

package/src/stores/editor/editor-state/project-configuration-editor-state/ProjectConfigurationEditorState.ts

@@ -200,22 +200,6 @@ export class ProjectConfigurationEditorState extends EditorState {
         .map((v) => StoreProjectData.serialization.fromJson(v))
         .forEach((project) => this.projects.set(project.coordinates, project));
 
-      // Update the legacy dependency to newer format (using group ID and artifact ID instead of just project ID)
-      this.projectConfiguration?.projectDependencies.forEach(
-        (dependency): void => {
-          if (!dependency.isLegacyDependency) {
-            return;
-          }
-          const project = Array.from(this.projects.values()).find(
-            (e) => e.projectId === dependency.projectId,
-          );
-          // re-write to new format
-          if (project) {
-            dependency.setProjectId(project.coordinates);
-          }
-        },
-      );
-
       // fetch the versions for the dependency projects
       for (const dep of this.projectConfiguration?.projectDependencies ?? []) {
         const project = this.projects.get(dep.projectId);