@finos/legend-application-studio 28.21.3 → 28.21.4

package/src/stores/editor/editor-state/element-editor-state/testable/TestAssertionState.ts

@@ -25,10 +25,13 @@ import {
   EqualToJson,
   ExternalFormatData,
   EqualToJsonAssertFail,
+  EqualToRelationAssertFail,
   MultiExecutionServiceTestResult,
   AssertPass,
   TestExecutionStatus,
   EqualTo,
+  EqualToRelation,
+  RelationElement,
   observe_ValueSpecification,
 } from '@finos/legend-graph';
 import {
@@ -53,6 +56,7 @@ import {
   TESTABLE_RESULT,
 } from '../../../sidebar-state/testable/GlobalTestRunnerState.js';
 import type { TestableTestEditorState } from './TestableEditorState.js';
+import { RelationElementState } from '../data/EmbeddedDataState.js';
 import { isTestPassing } from '../../../utils/TestableUtils.js';
 import { equalTo_setExpected } from '../../../../graph-modifier/Testable_GraphModifierHelper.js';
 
@@ -120,6 +124,27 @@ export class EqualToJsonAssertFailState extends AssertFailState {
   }
 }
 
+export class EqualToRelationAssertFailState extends AssertFailState {
+  declare status: EqualToRelationAssertFail;
+  diffModal = false;
+
+  constructor(
+    resultState: TestAssertionResultState,
+    status: EqualToRelationAssertFail,
+  ) {
+    super(resultState, status);
+    this.status = status;
+    makeObservable(this, {
+      diffModal: observable,
+      setDiffModal: action,
+    });
+  }
+
+  setDiffModal(val: boolean): void {
+    this.diffModal = val;
+  }
+}
+
 export class UnsupportedAssertionStatusState extends TestAssertionStatusState {}
 
 export class TestAssertionResultState {
@@ -172,6 +197,9 @@ export class TestAssertionResultState {
       if (val instanceof EqualToJsonAssertFail) {
         return new EqualToJsonAssertFailState(this, val);
       }
+      if (val instanceof EqualToRelationAssertFail) {
+        return new EqualToRelationAssertFailState(this, val);
+      }
       if (val instanceof AssertFail) {
         if (this.assertionState.assertion instanceof EqualTo) {
           const message = val.message ?? '';
@@ -298,6 +326,42 @@ export class EqualToJsonAssertionState extends TestAssertionState {
   }
 }
 
+export class EqualToRelationAssertionState extends TestAssertionState {
+  declare assertion: EqualToRelation;
+  expectedRelationElementState: RelationElementState;
+
+  constructor(
+    editorStore: EditorStore,
+    assertionState: TestAssertionEditorState,
+  ) {
+    super(editorStore, assertionState);
+    this.expectedRelationElementState = new RelationElementState(
+      this.assertion.expected,
+    );
+  }
+
+  override get supportsGeneratingAssertion(): boolean {
+    return false;
+  }
+
+  generateExpected(_status: AssertFail): boolean {
+    return false;
+  }
+
+  generateBare(): TestAssertion {
+    const bareAssertion = new EqualToRelation();
+    bareAssertion.expected = new RelationElement();
+    bareAssertion.expected.columns = [];
+    bareAssertion.expected.paths = [];
+    bareAssertion.expected.rows = [];
+    return bareAssertion;
+  }
+
+  label(): string {
+    return 'EqualToRelation';
+  }
+}
+
 export class EqualToAssertionState extends TestAssertionState {
   declare assertion: EqualTo;
   valueSpec: ValueSpecification;
@@ -466,6 +530,8 @@ export class TestAssertionEditorState {
   buildAssertionState(assertion: TestAssertion): TestAssertionState {
     if (assertion instanceof EqualToJson) {
       return new EqualToJsonAssertionState(this.editorStore, this);
+    } else if (assertion instanceof EqualToRelation) {
+      return new EqualToRelationAssertionState(this.editorStore, this);
     } else if (assertion instanceof EqualTo) {
       const val = returnUndefOnError(() =>
         this.editorStore.graphManagerState.graphManager.buildValueSpecification(

package/tsconfig.json

@@ -133,6 +133,7 @@
     "./src/stores/editor/editor-state/element-editor-state/data/DataEditorState.ts",
     "./src/stores/editor/editor-state/element-editor-state/data/EmbeddedDataState.ts",
     "./src/stores/editor/editor-state/element-editor-state/dataProduct/DataProductEditorState.ts",
+    "./src/stores/editor/editor-state/element-editor-state/dataProduct/testable/DataProductTestableState.ts",
     "./src/stores/editor/editor-state/element-editor-state/external-format/DSL_ExternalFormat_BindingEditorState.ts",
     "./src/stores/editor/editor-state/element-editor-state/external-format/DSL_ExternalFormat_SchemaSetEditorState.ts",
     "./src/stores/editor/editor-state/element-editor-state/function-activator/HostedServiceFunctionActivatorEditorState.ts",
@@ -264,6 +265,7 @@
     "./src/components/editor/editor-group/data-editor/RelationElementsDataEditor.tsx",
     "./src/components/editor/editor-group/data-editor/RelationalCSVDataEditor.tsx",
     "./src/components/editor/editor-group/dataProduct/DataProductEditor.tsx",
+    "./src/components/editor/editor-group/dataProduct/testable/DataProductTestableEditor.tsx",
     "./src/components/editor/editor-group/database-editor/DatabaseAnnotationDisplay.tsx",
     "./src/components/editor/editor-group/database-editor/DatabaseDiagramCanvas.tsx",
     "./src/components/editor/editor-group/database-editor/DatabaseEditor.tsx",