@finos/legend-graph 6.0.1 → 8.0.1

package/src/graphManager/AbstractPureGraphManager.ts

@@ -17,51 +17,54 @@
 import type {
   ExecutionResult,
   EXECUTION_SERIALIZATION_FORMAT,
-} from './action/execution/ExecutionResult';
-import type { ServiceRegistrationResult } from './action/service/ServiceRegistrationResult';
-import type { Service } from '../models/metamodels/pure/packageableElements/service/Service';
+} from './action/execution/ExecutionResult.js';
+import type { ServiceRegistrationResult } from './action/service/ServiceRegistrationResult.js';
+import type { Service } from '../models/metamodels/pure/packageableElements/service/Service.js';
 import type {
   ImportConfigurationDescription,
   ImportMode,
-} from './action/generation/ImportConfigurationDescription';
-import type { FileGenerationSpecification } from '../models/metamodels/pure/packageableElements/fileGeneration/FileGenerationSpecification';
-import type { GenerationOutput } from './action/generation/GenerationOutput';
-import type { ServiceTestResult } from './action/service/ServiceTestResult';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import type { PureModel, CoreModel, SystemModel } from '../graph/PureModel';
-import type { Mapping } from '../models/metamodels/pure/packageableElements/mapping/Mapping';
-import type { Runtime } from '../models/metamodels/pure/packageableElements/runtime/Runtime';
-import type { DependencyManager } from '../graph/DependencyManager';
-import type { Class } from '../models/metamodels/pure/packageableElements/domain/Class';
-import type { RawLambda } from '../models/metamodels/pure/rawValueSpecification/RawLambda';
+} from './action/generation/ImportConfigurationDescription.js';
+import type { FileGenerationSpecification } from '../models/metamodels/pure/packageableElements/fileGeneration/FileGenerationSpecification.js';
+import type { GenerationOutput } from './action/generation/GenerationOutput.js';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import type { PureModel, CoreModel, SystemModel } from '../graph/PureModel.js';
+import type { Mapping } from '../models/metamodels/pure/packageableElements/mapping/Mapping.js';
+import type { Runtime } from '../models/metamodels/pure/packageableElements/runtime/Runtime.js';
+import type { DependencyManager } from '../graph/DependencyManager.js';
+import type { Class } from '../models/metamodels/pure/packageableElements/domain/Class.js';
+import type { RawLambda } from '../models/metamodels/pure/rawValueSpecification/RawLambda.js';
 import type {
   GenerationConfigurationDescription,
   GenerationMode,
-} from './action/generation/GenerationConfigurationDescription';
-import type { ValueSpecification } from '../models/metamodels/pure/valueSpecification/ValueSpecification';
-import type { RawValueSpecification } from '../models/metamodels/pure/rawValueSpecification/RawValueSpecification';
-import type { ServiceExecutionMode } from './action/service/ServiceExecutionMode';
-import type { TEMPORARY__AbstractEngineConfig } from './action/TEMPORARY__AbstractEngineConfig';
-import type { DatabaseBuilderInput } from './action/generation/DatabaseBuilderInput';
-import type { RawRelationalOperationElement } from '../models/metamodels/pure/packageableElements/store/relational/model/RawRelationalOperationElement';
+} from './action/generation/GenerationConfigurationDescription.js';
+import type { ValueSpecification } from '../models/metamodels/pure/valueSpecification/ValueSpecification.js';
+import type { RawValueSpecification } from '../models/metamodels/pure/rawValueSpecification/RawValueSpecification.js';
+import type { ServiceExecutionMode } from './action/service/ServiceExecutionMode.js';
+import type { TEMPORARY__AbstractEngineConfig } from './action/TEMPORARY__AbstractEngineConfig.js';
+import type { DatabaseBuilderInput } from './action/generation/DatabaseBuilderInput.js';
+import type { RawRelationalOperationElement } from '../models/metamodels/pure/packageableElements/store/relational/model/RawRelationalOperationElement.js';
 import type {
   ExecutionPlan,
   RawExecutionPlan,
-} from '../models/metamodels/pure/executionPlan/ExecutionPlan';
-import type { ExecutionNode } from '../models/metamodels/pure/executionPlan/nodes/ExecutionNode';
+} from '../models/metamodels/pure/executionPlan/ExecutionPlan.js';
+import type { ExecutionNode } from '../models/metamodels/pure/executionPlan/nodes/ExecutionNode.js';
 import type {
+  ActionState,
   Log,
   ServerClientConfig,
   TracerService,
 } from '@finos/legend-shared';
-import type { LightQuery, Query } from './action/query/Query';
+import type { LightQuery, Query } from './action/query/Query.js';
 import type { Entity } from '@finos/legend-model-storage';
-import type { GraphPluginManager } from '../GraphPluginManager';
-import type { QuerySearchSpecification } from './action/query/QuerySearchSpecification';
-import type { ExternalFormatDescription } from './action/externalFormat/ExternalFormatDescription';
-import type { ConfigurationProperty } from '../models/metamodels/pure/packageableElements/fileGeneration/ConfigurationProperty';
-import type { GraphBuilderReport } from './GraphBuilderReport';
-import type { ModelGenerationConfiguration } from '../models/ModelGenerationConfiguration';
+import type { GraphPluginManager } from '../GraphPluginManager.js';
+import type { QuerySearchSpecification } from './action/query/QuerySearchSpecification.js';
+import type { ExternalFormatDescription } from './action/externalFormat/ExternalFormatDescription.js';
+import type { ConfigurationProperty } from '../models/metamodels/pure/packageableElements/fileGeneration/ConfigurationProperty.js';
+import type { GraphBuilderReport } from './GraphBuilderReport.js';
+import type { ModelGenerationConfiguration } from '../models/ModelGenerationConfiguration.js';
+import type { DEPRECATED__ServiceTestResult } from './action/service/DEPRECATED__ServiceTestResult.js';
+import type { RunTestsTestableInput } from '../models/metamodels/pure/test/result/RunTestsTestableInput.js';
+import type { TestResult } from '../models/metamodels/pure/test/result/TestResult.js';
 
 export interface TEMPORARY__EngineSetupConfig {
   env: string;
@@ -125,6 +128,7 @@ export abstract class AbstractPureGraphManager {
   abstract buildSystem(
     coreModel: CoreModel,
     systemModel: SystemModel,
+    buildState: ActionState,
     options?: GraphBuilderOptions,
   ): Promise<GraphBuilderReport>;
 
@@ -134,6 +138,7 @@ export abstract class AbstractPureGraphManager {
   abstract buildGraph(
     graph: PureModel,
     entities: Entity[],
+    buildState: ActionState,
     options?: GraphBuilderOptions,
   ): Promise<GraphBuilderReport>;
 
@@ -150,12 +155,14 @@ export abstract class AbstractPureGraphManager {
     systemModel: SystemModel,
     dependencyManager: DependencyManager,
     dependencyEntitiesMap: Map<string, Entity[]>,
+    buildState: ActionState,
     options?: GraphBuilderOptions,
   ): Promise<GraphBuilderReport>;
 
   abstract buildGenerations(
     graph: PureModel,
     generationEntities: Map<string, Entity[]>,
+    buildState: ActionState,
     options?: GraphBuilderOptions,
   ): Promise<GraphBuilderReport>;
 
@@ -172,10 +179,9 @@ export abstract class AbstractPureGraphManager {
   abstract pureCodeToLambda(
     lambda: string,
     lambdaId?: string,
-  ): Promise<RawLambda | undefined>;
+  ): Promise<RawLambda>;
   abstract lambdaToPureCode(
     lambda: RawLambda,
-    lambdaId?: string,
     pretty?: boolean,
   ): Promise<string>;
   abstract lambdasToPureCode(
@@ -187,7 +193,7 @@ export abstract class AbstractPureGraphManager {
   abstract pureCodeToRelationalOperationElement(
     operation: string,
     operationId: string,
-  ): Promise<RawRelationalOperationElement | undefined>;
+  ): Promise<RawRelationalOperationElement>;
   abstract relationalOperationElementToPureCode(
     operations: Map<string, RawRelationalOperationElement>,
   ): Promise<Map<string, string>>;
@@ -209,7 +215,14 @@ export abstract class AbstractPureGraphManager {
     options?: { keepSourceInformation?: boolean },
   ): Promise<string>;
 
-  // ------------------------------------------- ValueSpecification  -------------------------------------------
+  // ------------------------------------------- Test  -------------------------------------------
+
+  abstract runTests(
+    inputs: RunTestsTestableInput[],
+    graph: PureModel,
+  ): Promise<TestResult[]>;
+
+  // ------------------------------------------- Value Specification  -------------------------------------------
 
   abstract buildValueSpecification(
     valueSpecificationJson: Record<PropertyKey, unknown>,
@@ -333,15 +346,19 @@ export abstract class AbstractPureGraphManager {
     executionMode: ServiceExecutionMode,
     version: string | undefined,
   ): Promise<ServiceRegistrationResult>;
-  abstract runServiceTests(
-    service: Service,
-    graph: PureModel,
-  ): Promise<ServiceTestResult[]>;
   abstract activateService(
     serviceUrl: string,
     serviceId: string,
   ): Promise<void>;
 
+  /**
+   * @deprecated
+   */
+  abstract runLegacyServiceTests(
+    service: Service,
+    graph: PureModel,
+  ): Promise<DEPRECATED__ServiceTestResult[]>;
+
   // ------------------------------------------- Query -------------------------------------------
 
   abstract searchQueries(
@@ -355,6 +372,7 @@ export abstract class AbstractPureGraphManager {
   abstract deleteQuery(queryId: string): Promise<void>;
 
   // ------------------------------------------- Legend Query -------------------------------------
+
   abstract buildGraphForCreateQuerySetup(
     graph: PureModel,
     entities: Entity[],
@@ -382,13 +400,6 @@ export abstract class AbstractPureGraphManager {
   // displaying, or for interacting with SDLC server.
   // As such, most of these methods will only deal with `string` or `plain object`
 
-  /**
-   * Prune source information from protocol object
-   *
-   * NOTE: if we did this right initially, it is as easy as walking through the object and prune
-   * any field with key `sourceInformation`, but we have introduced many specific source information
-   * fields, such as `elementSourceInformation in connection, so we need to handle them all.
-   */
   abstract pruneSourceInformation(object: object): Record<PropertyKey, unknown>;
   abstract elementToEntity(
     element: PackageableElement,

package/src/graphManager/CorePureGraphManagerPlugin.ts

@@ -15,12 +15,14 @@
  */
 
 import packageJson from '../../package.json';
-import { PureGraphManagerPlugin } from './PureGraphManagerPlugin';
-import { CORE_PURE_PATH } from '../MetaModelConst';
+import { PureGraphManagerPlugin } from './PureGraphManagerPlugin.js';
+import { CORE_PURE_PATH } from '../MetaModelConst.js';
 
 export class CorePureGraphManagerPlugin extends PureGraphManagerPlugin {
+  static NAME = packageJson.extensions.pureGraphManagerPlugin;
+
   constructor() {
-    super(packageJson.extensions.pureGraphManagerPlugin, packageJson.version);
+    super(CorePureGraphManagerPlugin.NAME, packageJson.version);
   }
 
   override getExtraExposedSystemElementPath(): string[] {

package/src/graphManager/DSLExternalFormat_GraphManagerHelper.ts

@@ -15,10 +15,10 @@
  */
 
 import { guaranteeNonNullable, guaranteeType } from '@finos/legend-shared';
-import type { BasicModel } from '../graph/BasicModel';
-import type { PureModel } from '../graph/PureModel';
-import { SchemaSet } from '../models/metamodels/pure/packageableElements/externalFormat/schemaSet/DSLExternalFormat_SchemaSet';
-import { Binding } from '../models/metamodels/pure/packageableElements/externalFormat/store/DSLExternalFormat_Binding';
+import type { BasicModel } from '../graph/BasicModel.js';
+import type { PureModel } from '../graph/PureModel.js';
+import { SchemaSet } from '../models/metamodels/pure/packageableElements/externalFormat/schemaSet/DSLExternalFormat_SchemaSet.js';
+import { Binding } from '../models/metamodels/pure/packageableElements/externalFormat/store/DSLExternalFormat_Binding.js';
 
 export const getSchemaSet = (path: string, graph: PureModel): SchemaSet =>
   graph.getExtensionElement(path, SchemaSet, `Can't find schema set '${path}'`);

package/src/graphManager/DSLExternalFormat_PureGraphManagerPlugin.ts

@@ -15,27 +15,27 @@
  */
 
 import packageJson from '../../package.json';
-import type { Connection } from '../DSLMapping_Exports';
-import { ExternalFormatConnection } from '../models/metamodels/pure/packageableElements/externalFormat/connection/DSLExternalFormat_ExternalFormatConnection';
-import { SchemaSet } from '../models/metamodels/pure/packageableElements/externalFormat/schemaSet/DSLExternalFormat_SchemaSet';
-import { Binding } from '../models/metamodels/pure/packageableElements/externalFormat/store/DSLExternalFormat_Binding';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import type { ObserverContext } from './action/changeDetection/CoreObserverHelper';
+import type { Connection } from '../DSLMapping_Exports.js';
+import { ExternalFormatConnection } from '../models/metamodels/pure/packageableElements/externalFormat/connection/DSLExternalFormat_ExternalFormatConnection.js';
+import { SchemaSet } from '../models/metamodels/pure/packageableElements/externalFormat/schemaSet/DSLExternalFormat_SchemaSet.js';
+import { Binding } from '../models/metamodels/pure/packageableElements/externalFormat/store/DSLExternalFormat_Binding.js';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import type { ObserverContext } from './action/changeDetection/CoreObserverHelper.js';
 import {
   observe_Binding,
   observe_ExternalFormatConnection,
   observe_SchemaSet,
-} from './action/changeDetection/DSLExternalFormat_ObserverHelper';
+} from './action/changeDetection/DSLExternalFormat_ObserverHelper.js';
 import type {
   ConnectionObserver,
   DSLMapping_PureGraphManagerPlugin_Extension,
   PureGrammarConnectionLabeler,
-} from './DSLMapping_PureGraphManagerPlugin_Extension';
+} from './DSLMapping_PureGraphManagerPlugin_Extension.js';
 import {
   type PureGrammarElementLabeler,
   type ElementObserver,
   PureGraphManagerPlugin,
-} from './PureGraphManagerPlugin';
+} from './PureGraphManagerPlugin.js';
 
 export const PURE_GRAMMAR_EXTERNAL_FORMAT_PARSER_NAME = 'ExternalFormat';
 export const PURE_GRAMMAR_BINDING_ELEMENT_TYPE_LABEL = 'Binding';

package/src/graphManager/DSLGenerationSpecification_PureGraphManagerPlugin_Extension.ts

@@ -14,9 +14,9 @@
  * limitations under the License.
  */
 
-import type { ModelGenerationSpecification } from '../models/metamodels/pure/packageableElements/generationSpecification/ModelGenerationSpecification';
-import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin';
-import type { PureModel } from '../graph/PureModel';
+import type { ModelGenerationSpecification } from '../models/metamodels/pure/packageableElements/generationSpecification/ModelGenerationSpecification.js';
+import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin.js';
+import type { PureModel } from '../graph/PureModel.js';
 
 export type ModelGenerationElementGetter = (
   graph: PureModel,

package/src/graphManager/DSLMapping_PureGraphManagerPlugin_Extension.ts

@@ -14,12 +14,12 @@
  * limitations under the License.
  */
 
-import type { Connection } from '../models/metamodels/pure/packageableElements/connection/Connection';
-import type { InputData } from '../models/metamodels/pure/packageableElements/mapping/InputData';
-import type { PropertyMapping } from '../models/metamodels/pure/packageableElements/mapping/PropertyMapping';
-import type { SetImplementation } from '../models/metamodels/pure/packageableElements/mapping/SetImplementation';
-import type { ObserverContext } from './action/changeDetection/CoreObserverHelper';
-import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin';
+import type { Connection } from '../models/metamodels/pure/packageableElements/connection/Connection.js';
+import type { InputData } from '../models/metamodels/pure/packageableElements/mapping/InputData.js';
+import type { PropertyMapping } from '../models/metamodels/pure/packageableElements/mapping/PropertyMapping.js';
+import type { SetImplementation } from '../models/metamodels/pure/packageableElements/mapping/SetImplementation.js';
+import type { ObserverContext } from './action/changeDetection/CoreObserverHelper.js';
+import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin.js';
 
 export type PureGrammarConnectionLabeler = (
   connection: Connection,

package/src/graphManager/EmbeddedData_PureGraphManagerPlugin_Extension.ts

@@ -14,9 +14,9 @@
  * limitations under the License.
  */
 
-import type { EmbeddedData } from '../models/metamodels/pure/data/EmbeddedData';
-import type { ObserverContext } from './action/changeDetection/CoreObserverHelper';
-import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin';
+import type { EmbeddedData } from '../models/metamodels/pure/data/EmbeddedData.js';
+import type { ObserverContext } from './action/changeDetection/CoreObserverHelper.js';
+import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin.js';
 
 export type EmbeddedDataObserver = (
   embeddedData: EmbeddedData,

package/src/graphManager/GraphBuilderReport.ts

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { GRAPH_MANAGER_EVENT } from './GraphManagerEvent';
+import { GRAPH_MANAGER_EVENT } from './GraphManagerEvent.js';
 
 export class GraphBuilderReport {
   timings: {

package/src/graphManager/GraphManagerEvent.ts

@@ -15,31 +15,31 @@
  */
 
 export enum GRAPH_MANAGER_EVENT {
-  GRAPH_BUILDER_ELEMENTS_DESERIALIZED = 'GRAPH_BUILDER_ELEMENTS_DESERIALIZED',
-  GRAPH_BUILDER_ELEMENTS_INDEXED = 'GRAPH_BUILDER_ELEMENTS_INDEXED',
-  GRAPH_BUILDER_SECTION_INDICES_BUILT = 'GRAPH_BUILDER_SECTION_INDICES_BUILT',
-  GRAPH_BUILDER_DOMAIN_MODELS_BUILT = 'GRAPH_BUILDER_DOMAIN_MODELS_BUILT',
-  GRAPH_BUILDER_STORES_BUILT = 'GRAPH_BUILDER_STORES_BUILT',
-  GRAPH_BUILDER_MAPPINGS_BUILT = 'GRAPH_BUILDER_MAPPINGS_BUILT',
-  GRAPH_BUILDER_CONNECTIONS_AND_RUNTIMES_BUILT = 'GRAPH_BUILDER_CONNECTIONS_AND_RUNTIMES_BUILT',
-  GRAPH_BUILDER_SERVICES_BUILT = 'GRAPH_BUILDER_SERVICES_BUILT',
-  GRAPH_BUILDER_DATA_ELEMENTS_BUILT = 'GRAPH_BUILDER_DATA_ELEMENTS_BUILT',
-  GRAPH_BUILDER_OTHER_ELEMENTS_BUILT = 'GRAPH_BUILDER_OTHER_ELEMENTS_BUILT',
-  GRAPH_BUILDER_COMPLETED = 'GRAPH_BUILDER_COMPLETED',
+  GRAPH_BUILDER_ELEMENTS_DESERIALIZED = 'graph-manager.graph-builder.deserialization.success',
+  GRAPH_BUILDER_ELEMENTS_INDEXED = 'graph-manager.graph-builder.indexing.success',
+  GRAPH_BUILDER_SECTION_INDICES_BUILT = 'graph-manager.graph-builder.build-section-indices.success',
+  GRAPH_BUILDER_DOMAIN_MODELS_BUILT = 'graph-manager.graph-builder.build-domain-models.success',
+  GRAPH_BUILDER_STORES_BUILT = 'graph-manager.graph-builder.build-store.success',
+  GRAPH_BUILDER_MAPPINGS_BUILT = 'graph-manager.graph-builder.build-mappings.success',
+  GRAPH_BUILDER_CONNECTIONS_AND_RUNTIMES_BUILT = 'graph-manager.graph-builder.build-connections-and-runtimes.success',
+  GRAPH_BUILDER_SERVICES_BUILT = 'graph-manager.graph-builder.build-services.success',
+  GRAPH_BUILDER_DATA_ELEMENTS_BUILT = 'graph-manager.graph-builder.build-data.success',
+  GRAPH_BUILDER_OTHER_ELEMENTS_BUILT = 'graph-manager.graph-builder.build-other-elements.success',
+  GRAPH_BUILDER_COMPLETED = 'graph-manager.graph-builder.success',
 
-  GRAPH_INITIALIZED = 'GRAPH_INITIALIZED',
-  GRAPH_DEPENDENCIES_FETCHED = 'GRAPH_DEPENDENCIES_FETCHED',
-  GRAPH_ENTITIES_FETCHED = 'GRAPH_ENTITIES_FETCHED',
-  GRAPH_UPDATED_AND_REBUILT = 'GRAPH_REBUILT',
-  GRAPH_MODEL_TO_GRAMMAR_TRANSFORMED = 'GRAPH_MODEL_TO_GRAMMAR_TRANSFORMED',
-  GRAPH_GRAMMAR_TO_MODEL_TRANSFORMED = 'GRAPH_GRAMMAR_TO_MODEL_TRANSFORMED',
-  GRAPH_META_MODEL_TO_PROTOCOL_TRANSFORMED = 'GRAPH_META_MODEL_TO_PROTOCOL_TRANSFORMED',
-  GRAPH_COMPILE_CONTEXT_COLLECTED = 'GRAPH_COMPILE_CONTEXT_COLLECTED',
-  GRAPH_PROTOCOL_SERIALIZED = 'GRAPH_PROTOCOL_SERIALIZED',
+  GRAPH_INITIALIZED = 'graph-manager.graph-initialization.success',
+  GRAPH_DEPENDENCIES_FETCHED = 'graph-manager.fetch-dependencies.success',
+  GRAPH_ENTITIES_FETCHED = 'graph-manager.entities-dependencies.success',
+  GRAPH_UPDATED_AND_REBUILT = 'graph-manager.rebuild.success',
+  GRAPH_MODEL_TO_GRAMMAR_TRANSFORMED = 'graph-manager.grammar.composing.success',
+  GRAPH_GRAMMAR_TO_MODEL_TRANSFORMED = 'graph-manager.grammar.parsing.success',
+  GRAPH_META_MODEL_TO_PROTOCOL_TRANSFORMED = 'graph-manager.transformation.success',
+  GRAPH_COMPILE_CONTEXT_COLLECTED = 'graph-manager.build-graph-compile-context.success',
+  GRAPH_PROTOCOL_SERIALIZED = 'graph-manager.graph-serialization.success',
 
-  GRAPH_BUILDER_FAILURE = 'GRAPH_BUILD_FAILURE',
-  EXECUTION_FAILURE = 'EXECUTION_FAILURE',
-  COMPILATION_FAILURE = 'COMPILATION_FAILURE',
-  PARSING_FAILURE = 'PARSING_FAILURE',
-  GRAPH_MANAGER_FAILURE = 'GRAPH_MANAGER_FAILURE',
+  GRAPH_BUILDER_FAILURE = 'graph-manager.graph-builder.failure',
+  EXECUTION_FAILURE = 'graph-manager.execution.failure',
+  COMPILATION_FAILURE = 'graph-manager.compilation.failure',
+  PARSING_FAILURE = 'graph-manager.grammar.parsing.failure',
+  GRAPH_MANAGER_FAILURE = 'graph-manager.failure',
 }

package/src/graphManager/GraphManagerTelemetry.ts

@@ -15,8 +15,8 @@
  */
 
 import type { TelemetryService } from '@finos/legend-shared';
-import type { GraphBuilderReport } from './GraphBuilderReport';
-import { GRAPH_MANAGER_EVENT } from './GraphManagerEvent';
+import type { GraphBuilderReport } from './GraphBuilderReport.js';
+import { GRAPH_MANAGER_EVENT } from './GraphManagerEvent.js';
 
 type GraphBuilt_TelemetryData = {
   timings: Record<string, number>;

package/src/graphManager/PureGraphManagerPlugin.ts

@@ -15,9 +15,11 @@
  */
 
 import { AbstractPlugin } from '@finos/legend-shared';
-import type { GraphPluginManager } from '../GraphPluginManager';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import type { ObserverContext } from './action/changeDetection/CoreObserverHelper';
+import type { PureModel } from '../graph/PureModel.js';
+import type { GraphPluginManager } from '../GraphPluginManager.js';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import type { Testable } from '../models/metamodels/pure/test/Testable.js';
+import type { ObserverContext } from './action/changeDetection/CoreObserverHelper.js';
 
 /**
  * Unlike `PureGraphPlugin`, this is for plugins of graph manager, i.e. operations acting
@@ -27,11 +29,22 @@ import type { ObserverContext } from './action/changeDetection/CoreObserverHelpe
 export type PureGrammarElementLabeler = (
   metamodel: PackageableElement,
 ) => string | undefined;
+
 export type ElementObserver = (
   metamodel: PackageableElement,
   context: ObserverContext,
 ) => PackageableElement | undefined;
 
+export type TestableIDBuilder = (
+  testable: Testable,
+  graph: PureModel,
+) => string | undefined;
+
+export type TestableFinder = (
+  id: string,
+  graph: PureModel,
+) => Testable | undefined;
+
 export abstract class PureGraphManagerPlugin extends AbstractPlugin {
   /**
    * This helps to better type-checking for this empty abtract type
@@ -73,4 +86,14 @@ export abstract class PureGraphManagerPlugin extends AbstractPlugin {
    * Get the list of Pure grammar element labelers.
    */
   getExtraPureGrammarElementLabelers?(): PureGrammarElementLabeler[];
+
+  /**
+   * Get the list of Testable id builders.
+   */
+  getExtraTestableIDBuilders?(): TestableIDBuilder[];
+
+  /**
+   * Get the list of Testable finders.
+   */
+  getExtraTestableFinders?(): TestableFinder[];
 }

package/src/graphManager/StoreRelational_PureGraphManagerPlugin_Extension.ts

@@ -14,12 +14,12 @@
  * limitations under the License.
  */
 
-import type { AuthenticationStrategy } from '../models/metamodels/pure/packageableElements/store/relational/connection/AuthenticationStrategy';
-import type { DatasourceSpecification } from '../models/metamodels/pure/packageableElements/store/relational/connection/DatasourceSpecification';
-import type { PostProcessor } from '../models/metamodels/pure/packageableElements/store/relational/connection/postprocessor/PostProcessor';
-import type { Milestoning } from '../models/metamodels/pure/packageableElements/store/relational/model/milestoning/Milestoning';
-import type { ObserverContext } from './action/changeDetection/CoreObserverHelper';
-import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin';
+import type { AuthenticationStrategy } from '../models/metamodels/pure/packageableElements/store/relational/connection/AuthenticationStrategy.js';
+import type { DatasourceSpecification } from '../models/metamodels/pure/packageableElements/store/relational/connection/DatasourceSpecification.js';
+import type { PostProcessor } from '../models/metamodels/pure/packageableElements/store/relational/connection/postprocessor/PostProcessor.js';
+import type { Milestoning } from '../models/metamodels/pure/packageableElements/store/relational/model/milestoning/Milestoning.js';
+import type { ObserverContext } from './action/changeDetection/CoreObserverHelper.js';
+import type { PureGraphManagerPlugin } from './PureGraphManagerPlugin.js';
 
 export type MilestoningObserver = (
   milestoning: Milestoning,

package/src/graphManager/action/EngineError.ts

@@ -15,26 +15,10 @@
  */
 
 import { ApplicationError } from '@finos/legend-shared';
-import { observable, action, makeObservable } from 'mobx';
-import type { SourceInformation } from '../action/SourceInformation';
+import type { SourceInformation } from '../action/SourceInformation.js';
 
 export class EngineError extends ApplicationError {
   sourceInformation?: SourceInformation | undefined;
-
-  constructor(message: string | undefined) {
-    super(message);
-
-    // TODO: check if we need this to be observable or not?
-    makeObservable(this, {
-      message: observable,
-      sourceInformation: observable,
-      setSourceInformation: action,
-    });
-  }
-
-  setSourceInformation(sourceInformation: SourceInformation | undefined): void {
-    this.sourceInformation = sourceInformation;
-  }
 }
 
 export class ParserError extends EngineError {}

package/src/graphManager/action/ExecutionError.ts

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { EngineError } from './EngineError';
+import { EngineError } from './EngineError.js';
 
 export class ExecutionError extends EngineError {
   declare stack: string;

package/src/graphManager/action/SourceInformationHelper.ts

@@ -15,7 +15,7 @@
  */
 
 import { assertTrue } from '@finos/legend-shared';
-import type { SourceInformation } from './SourceInformation';
+import type { SourceInformation } from './SourceInformation.js';
 
 // NOTE: @ is chosen because it is not part of identifier token in Pure grammar
 // TODO: handle the case of quote identifier e.g. model::something::'I have an @ in me'

package/src/graphManager/action/TEMPORARY__AbstractEngineConfig.ts

@@ -14,8 +14,6 @@
  * limitations under the License.
  */
 
-import { action, makeObservable, observable } from 'mobx';
-
 const DEFAULT_TAB_SIZE = 2;
 
 /**
@@ -53,19 +51,4 @@ export class TEMPORARY__AbstractEngineConfig {
   setUseBase64ForAdhocConnectionDataUrls(val: boolean): void {
     this.useBase64ForAdhocConnectionDataUrls = val;
   }
-
-  constructor() {
-    makeObservable(this, {
-      env: observable,
-      currentUserId: observable,
-      baseUrl: observable,
-      useClientRequestPayloadCompression: observable,
-      useBase64ForAdhocConnectionDataUrls: observable,
-      setEnv: action,
-      setCurrentUserId: action,
-      setBaseUrl: action,
-      setUseClientRequestPayloadCompression: action,
-      setUseBase64ForAdhocConnectionDataUrls: action,
-    });
-  }
 }

package/src/graphManager/action/changeDetection/CoreObserverHelper.ts

@@ -21,14 +21,14 @@ import {
   makeObservable,
   observable,
 } from 'mobx';
-import type { PackageableElement } from '../../../models/metamodels/pure/packageableElements/PackageableElement';
+import type { PackageableElement } from '../../../models/metamodels/pure/packageableElements/PackageableElement.js';
 import type {
   OptionalPackageableElementReference,
   PackageableElementReference,
-} from '../../../models/metamodels/pure/packageableElements/PackageableElementReference';
-import type { Multiplicity } from '../../../models/metamodels/pure/packageableElements/domain/Multiplicity';
-import type { InferableValue } from '../../../models/metamodels/pure/InferableValue';
-import type { PureGraphManagerPlugin } from '../../PureGraphManagerPlugin';
+} from '../../../models/metamodels/pure/packageableElements/PackageableElementReference.js';
+import type { Multiplicity } from '../../../models/metamodels/pure/packageableElements/domain/Multiplicity.js';
+import type { InferableValue } from '../../../models/metamodels/pure/InferableValue.js';
+import type { PureGraphManagerPlugin } from '../../PureGraphManagerPlugin.js';
 
 export class ObserverContext {
   plugins: PureGraphManagerPlugin[] = [];
@@ -95,7 +95,6 @@ export const observe_Multiplicity = skipObserved(
     makeObservable(metamodel, {
       lowerBound: observable,
       upperBound: observable,
-      str: computed,
       hashCode: computed,
     }),
 );