@finos/legend-graph 32.6.1 → 32.6.3

package/src/graph-manager/protocol/pure/v1/model/packageableElements/dataProduct/test/V1_AccessPointTest.ts

@@ -0,0 +1,33 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { hashArray, type Hashable } from '@finos/legend-shared';
+import { V1_AtomicTest } from '../../../test/V1_AtomicTest.js';
+import { CORE_HASH_STRUCTURE } from '../../../../../../../../graph/Core_HashUtils.js';
+
+export class V1_AccessPointTest extends V1_AtomicTest implements Hashable {
+  accessPointId!: string;
+
+  get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.DATA_PRODUCT_ACCESS_POINT_TEST,
+      this.id,
+      this.doc ?? '',
+      this.accessPointId,
+      hashArray(this.assertions),
+    ]);
+  }
+}

package/src/graph-manager/protocol/pure/v1/model/packageableElements/dataProduct/test/V1_DataProductTestSuite.ts

@@ -0,0 +1,34 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { hashArray, type Hashable } from '@finos/legend-shared';
+import { CORE_HASH_STRUCTURE } from '../../../../../../../../graph/Core_HashUtils.js';
+import { V1_TestSuite } from '../../../test/V1_TestSuite.js';
+import type { V1_DataResolver } from '../../../data/V1_DataResolver.js';
+
+export class V1_DataProductTestSuite extends V1_TestSuite implements Hashable {
+  testData: V1_DataResolver[] | undefined;
+
+  get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.DATA_PRODUCT_TEST_SUITE,
+      this.id,
+      this.doc ?? '',
+      hashArray(this.testData ?? []),
+      hashArray(this.tests),
+    ]);
+  }
+}

package/src/graph-manager/protocol/pure/v1/model/packageableElements/ingest/V1_IngestDefinition.ts

@@ -40,12 +40,36 @@ export class V1_IngestDatasetSource {
   schema!: V1_IngestDatasetSchema;
 }
 
+export enum V1_WriteModeType {
+  APPEND_ONLY = 'append_only',
+  BATCH_MILESTONED = 'batch_milestoned',
+  BATCH_MILESTONED_BUSINESS_TEMPORAL = 'batch_milestoned_business_temporal',
+}
+
+export class V1_WriteMode {
+  _type!: V1_WriteModeType;
+}
+
+export class V1_AppendOnly extends V1_WriteMode {
+  override _type = V1_WriteModeType.APPEND_ONLY;
+}
+
+export class V1_BatchMilestoned extends V1_WriteMode {
+  override _type = V1_WriteModeType.BATCH_MILESTONED;
+}
+
+export class V1_BatchMilestonedBusinessTemporal extends V1_WriteMode {
+  override _type = V1_WriteModeType.BATCH_MILESTONED_BUSINESS_TEMPORAL;
+}
+
 export class V1_IngestDataset {
   name!: string;
   primaryKey: string[] = [];
   source!: V1_IngestDatasetSource;
+  writeMode?: V1_WriteMode;
 }
 
 export class V1_IngestDefinitionContent {
   datasets?: V1_IngestDataset[];
+  writeMode?: V1_WriteMode;
 }

package/src/graph-manager/protocol/pure/v1/model/test/assertion/V1_EqualToRelation.ts

@@ -0,0 +1,39 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { hashArray } from '@finos/legend-shared';
+import { CORE_HASH_STRUCTURE } from '../../../../../../../graph/Core_HashUtils.js';
+import type { V1_RelationElement } from '../../data/V1_EmbeddedData.js';
+import {
+  type V1_TestAssertionVisitor,
+  V1_TestAssertion,
+} from './V1_TestAssertion.js';
+
+export class V1_EqualToRelation extends V1_TestAssertion {
+  expected!: V1_RelationElement;
+
+  get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.EQUAL_TO_RELATION,
+      this.id,
+      this.expected,
+    ]);
+  }
+
+  accept_TestAssertionVisitor<T>(visitor: V1_TestAssertionVisitor<T>): T {
+    return visitor.visit_EqualToRelation(this);
+  }
+}

package/src/graph-manager/protocol/pure/v1/model/test/assertion/V1_TestAssertion.ts

@@ -17,6 +17,7 @@
 import type { Hashable } from '@finos/legend-shared';
 import type { V1_EqualTo } from './V1_EqualTo.js';
 import type { V1_EqualToJson } from './V1_EqualToJson.js';
+import type { V1_EqualToRelation } from './V1_EqualToRelation.js';
 import type { V1_EqualToTDS } from './V1_EqualToTDS.js';
 
 export interface V1_TestAssertionVisitor<T> {
@@ -24,6 +25,7 @@ export interface V1_TestAssertionVisitor<T> {
   visit_EqualToTDS(testAssertion: V1_EqualToTDS): T;
   visit_EqualTo(testAssertion: V1_EqualTo): T;
   visit_EqualToJSON(testAssertion: V1_EqualToJson): T;
+  visit_EqualToRelation(testAssertion: V1_EqualToRelation): T;
 }
 
 export abstract class V1_TestAssertion implements Hashable {

package/src/graph-manager/protocol/pure/v1/model/test/assertion/status/V1_EqualToRelationAssertFail.ts

@@ -0,0 +1,37 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { hashArray, type Hashable } from '@finos/legend-shared';
+import { CORE_HASH_STRUCTURE } from '../../../../../../../../graph/Core_HashUtils.js';
+import { V1_AssertFail } from './V1_AssertFail.js';
+
+export class V1_EqualToRelationAssertFail
+  extends V1_AssertFail
+  implements Hashable
+{
+  expected!: string;
+  actual!: string;
+
+  override get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.EQUAL_TO_RELATION_ASSERT_FAIL,
+      this.id,
+      this.message,
+      this.expected,
+      this.actual,
+    ]);
+  }
+}

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_DataProductTransformer.ts

@@ -72,6 +72,12 @@ import { V1_transformEmbeddedData } from './V1_DataElementTransformer.js';
 import { ConcreteFunctionDefinition } from '../../../../../../../graph/metamodel/pure/packageableElements/function/ConcreteFunctionDefinition.js';
 import { generateFunctionPrettyName } from '../../../../../../../graph/helpers/PureLanguageHelper.js';
 import { V1_AppDirNode } from '../../../lakehouse/entitlements/V1_CoreEntitlements.js';
+import { DataProductAccessPointTest } from '../../../../../../../graph/metamodel/pure/dataProduct/test/DataProductAccessPointTest.js';
+import type { DataProductTestSuite } from '../../../../../../../graph/metamodel/pure/dataProduct/test/DataProductTestSuite.js';
+import { V1_AccessPointTest } from '../../../model/packageableElements/dataProduct/test/V1_AccessPointTest.js';
+import { V1_DataProductTestSuite } from '../../../model/packageableElements/dataProduct/test/V1_DataProductTestSuite.js';
+import { V1_transformDataResolver } from './V1_DataResolverTransformer.js';
+import { V1_transformTestAssertion } from './V1_TestTransformer.js';
 
 const transformAccessPoint = (
   ap: AccessPoint,
@@ -131,6 +137,42 @@ const transformDataProductIcon = (
   );
 };
 
+export const V1_transformDataProductAccessPointTest = (
+  element: DataProductAccessPointTest,
+): V1_AccessPointTest => {
+  const accessPointTest = new V1_AccessPointTest();
+  accessPointTest.id = element.id;
+  accessPointTest.doc = element.doc;
+  accessPointTest.accessPointId = element.accessPointId;
+  accessPointTest.assertions = element.assertions.map(
+    V1_transformTestAssertion,
+  );
+  return accessPointTest;
+};
+
+export const V1_transformDataProductTestSuite = (
+  element: DataProductTestSuite,
+  context: V1_GraphTransformerContext,
+): V1_DataProductTestSuite => {
+  const testSuite = new V1_DataProductTestSuite();
+  testSuite.id = element.id;
+  testSuite.doc = element.doc;
+  if (element.testData?.length) {
+    testSuite.testData = element.testData.map((dataResolver) =>
+      V1_transformDataResolver(dataResolver, context),
+    );
+  }
+  testSuite.tests = element.tests.map((test) => {
+    if (test instanceof DataProductAccessPointTest) {
+      return V1_transformDataProductAccessPointTest(test);
+    }
+    throw new UnsupportedOperationError(
+      'Unsupported data product test to transform',
+    );
+  });
+  return testSuite;
+};
+
 export const V1_transformDataProduct = (
   element: DataProduct,
   context: V1_GraphTransformerContext,
@@ -345,6 +387,11 @@ export const V1_transformDataProduct = (
   product.taggedValues = element.taggedValues.map((taggedValue) =>
     V1_transformTaggedValue(taggedValue),
   );
+  if (element.tests.length) {
+    product.testSuites = element.tests.map((testSuite) =>
+      V1_transformDataProductTestSuite(testSuite, context),
+    );
+  }
 
   if (element.owner instanceof AppDirOwner) {
     const v1Owner = new V1_AppDirOwner();

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_DataResolverTransformer.ts

@@ -0,0 +1,55 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { UnsupportedOperationError } from '@finos/legend-shared';
+import {
+  type DataResolver,
+  BaseDataResolver,
+  ReferenceDataResolver,
+} from '../../../../../../../graph/metamodel/pure/data/DataResolver.js';
+import {
+  V1_BaseDataResolver,
+  V1_ReferenceDataResolver,
+  type V1_DataResolver,
+} from '../../../model/data/V1_DataResolver.js';
+import { V1_transformElementReferencePointer } from './V1_CoreTransformerHelper.js';
+import { V1_transformEmbeddedData } from './V1_DataElementTransformer.js';
+import type { V1_GraphTransformerContext } from './V1_GraphTransformerContext.js';
+
+export const V1_transformDataResolver = (
+  dataResolver: DataResolver,
+  context: V1_GraphTransformerContext,
+): V1_DataResolver => {
+  if (dataResolver instanceof BaseDataResolver) {
+    const protocol = new V1_BaseDataResolver();
+    protocol.elementPointer = V1_transformElementReferencePointer(
+      undefined,
+      dataResolver.element,
+    );
+    protocol.data = V1_transformEmbeddedData(dataResolver.data, context);
+    return protocol;
+  } else if (dataResolver instanceof ReferenceDataResolver) {
+    const protocol = new V1_ReferenceDataResolver();
+    protocol.elementPointer = V1_transformElementReferencePointer(
+      undefined,
+      dataResolver.element,
+    );
+    return protocol;
+  }
+  throw new UnsupportedOperationError(
+    `Unable to transform data resolver of type '${dataResolver.constructor.name}'`,
+  );
+};

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_TestTransformer.ts

@@ -17,8 +17,10 @@
 import { UnsupportedOperationError } from '@finos/legend-shared';
 import { V1_EqualTo } from '../../../model/test/assertion/V1_EqualTo.js';
 import { V1_EqualToJson } from '../../../model/test/assertion/V1_EqualToJson.js';
+import { V1_EqualToRelation } from '../../../model/test/assertion/V1_EqualToRelation.js';
 import { EqualTo } from '../../../../../../../graph/metamodel/pure/test/assertion/EqualTo.js';
 import { EqualToJson } from '../../../../../../../graph/metamodel/pure/test/assertion/EqualToJson.js';
+import { EqualToRelation } from '../../../../../../../graph/metamodel/pure/test/assertion/EqualToRelation.js';
 import { V1_transformExternalFormatData } from './V1_DataElementTransformer.js';
 import type { V1_AtomicTest } from '../../../model/test/V1_AtomicTest.js';
 import { ServiceTest } from '../../../../../../../graph/metamodel/pure/packageableElements/service/ServiceTest.js';
@@ -37,6 +39,10 @@ import type {
 } from '../../../../../../../graph/metamodel/pure/test/Test.js';
 import { EqualToTDS } from '../../../../../../../graph/metamodel/pure/test/assertion/EqualToTDS.js';
 import { V1_EqualToTDS } from '../../../model/test/assertion/V1_EqualToTDS.js';
+import {
+  V1_RelationElement,
+  V1_RelationRowTestData,
+} from '../../../model/data/V1_EmbeddedData.js';
 import type { Testable_PureProtocolProcessorPlugin_Extension } from '../../../../extensions/Testable_PureProtocolProcessorPlugin_Extension.js';
 import {
   V1_transformMappingTest,
@@ -66,6 +72,23 @@ const transformEqualToTDS = (element: EqualToTDS): V1_EqualToTDS => {
   return equalToTDS;
 };
 
+const transformEqualToRelation = (
+  element: EqualToRelation,
+): V1_EqualToRelation => {
+  const v1 = new V1_EqualToRelation();
+  v1.id = element.id;
+  const rel = new V1_RelationElement();
+  rel.columns = element.expected.columns;
+  rel.paths = element.expected.paths;
+  rel.rows = element.expected.rows.map((row) => {
+    const r = new V1_RelationRowTestData();
+    r.values = row.values;
+    return r;
+  });
+  v1.expected = rel;
+  return v1;
+};
+
 export const V1_transformAtomicTest = (
   value: AtomicTest,
   context: V1_GraphTransformerContext,
@@ -103,6 +126,8 @@ export const V1_transformTestAssertion = (
     return transformEqualToJson(value);
   } else if (value instanceof EqualToTDS) {
     return transformEqualToTDS(value);
+  } else if (value instanceof EqualToRelation) {
+    return transformEqualToRelation(value);
   }
   throw new UnsupportedOperationError(`Can't transform test assertion`, value);
 };

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/V1_ElementSecondPassBuilder.ts

@@ -143,6 +143,7 @@ import {
   V1_buildDataProductIcon,
   V1_buildDataProductLink,
   V1_buildDataProductOperationalMetadata,
+  V1_buildDataProductTestSuite,
   V1_buildNativeModelAccess,
   V1_buildDataProductOwner,
 } from './helpers/V1_DataProductBuilder.js';
@@ -798,6 +799,11 @@ export class V1_ElementSecondPassBuilder
     dataProduct.sampleValues = element.sampleValues?.map((data) =>
       V1_buildEmbeddedData(data, this.context),
     );
+    if (element.testSuites?.length) {
+      dataProduct.tests = element.testSuites.map((testSuite) =>
+        V1_buildDataProductTestSuite(testSuite, this.context),
+      );
+    }
     if (element.operationalMetadata) {
       dataProduct.operationalMetadata = V1_buildDataProductOperationalMetadata(
         element.operationalMetadata,

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_DataProductBuilder.ts

@@ -80,6 +80,13 @@ import { Class } from '../../../../../../../../graph/metamodel/pure/packageableE
 import { Enumeration } from '../../../../../../../../graph/metamodel/pure/packageableElements/domain/Enumeration.js';
 import { Association } from '../../../../../../../../graph/metamodel/pure/packageableElements/domain/Association.js';
 import { generateFunctionPrettyName } from '../../../../../../../../graph/helpers/PureLanguageHelper.js';
+import { DataProductAccessPointTest } from '../../../../../../../../graph/metamodel/pure/dataProduct/test/DataProductAccessPointTest.js';
+import { DataProductTestSuite } from '../../../../../../../../graph/metamodel/pure/dataProduct/test/DataProductTestSuite.js';
+import { V1_AccessPointTest } from '../../../../model/packageableElements/dataProduct/test/V1_AccessPointTest.js';
+import type { V1_DataProductTestSuite } from '../../../../model/packageableElements/dataProduct/test/V1_DataProductTestSuite.js';
+import { V1_buildDataResolver } from './V1_DataResolverBuilderHelper.js';
+import { V1_buildTestAssertion } from './V1_TestBuilderHelper.js';
+import type { TestSuite } from '../../../../../../../../graph/metamodel/pure/test/Test.js';
 
 export const V1_buildDataProductLink = (
   link: V1_DataProductLink,
@@ -420,3 +427,42 @@ export const V1_buildDataProductOwner = (
     v1Owner,
   );
 };
+
+const V1_buildDataProductAccessPointTest = (
+  element: V1_AccessPointTest,
+  parentSuite: TestSuite,
+  context: V1_GraphBuilderContext,
+): DataProductAccessPointTest => {
+  const accessPointTest = new DataProductAccessPointTest();
+  accessPointTest.id = element.id;
+  accessPointTest.__parent = parentSuite;
+  accessPointTest.doc = element.doc;
+  accessPointTest.accessPointId = element.accessPointId;
+  accessPointTest.assertions = element.assertions.map((assertion) =>
+    V1_buildTestAssertion(assertion, accessPointTest, context),
+  );
+  return accessPointTest;
+};
+
+export const V1_buildDataProductTestSuite = (
+  element: V1_DataProductTestSuite,
+  context: V1_GraphBuilderContext,
+): DataProductTestSuite => {
+  const testSuite = new DataProductTestSuite();
+  testSuite.id = element.id;
+  testSuite.doc = element.doc;
+  if (element.testData?.length) {
+    testSuite.testData = element.testData.map((dataResolver) =>
+      V1_buildDataResolver(dataResolver, context),
+    );
+  }
+  testSuite.tests = element.tests.map((test) => {
+    if (test instanceof V1_AccessPointTest) {
+      return V1_buildDataProductAccessPointTest(test, testSuite, context);
+    }
+    throw new UnsupportedOperationError(
+      'Unable to build data product test: Unsupported data product test type',
+    );
+  });
+  return testSuite;
+};

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_DataResolverBuilderHelper.ts

@@ -0,0 +1,54 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { UnsupportedOperationError } from '@finos/legend-shared';
+import {
+  BaseDataResolver,
+  ReferenceDataResolver,
+  type DataResolver,
+} from '../../../../../../../../graph/metamodel/pure/data/DataResolver.js';
+import {
+  V1_BaseDataResolver,
+  V1_ReferenceDataResolver,
+  type V1_DataResolver,
+} from '../../../../model/data/V1_DataResolver.js';
+import type { V1_GraphBuilderContext } from '../V1_GraphBuilderContext.js';
+import { V1_buildEmbeddedData } from './V1_DataElementBuilderHelper.js';
+
+export const V1_buildDataResolver = (
+  protocol: V1_DataResolver,
+  context: V1_GraphBuilderContext,
+): DataResolver => {
+  if (protocol instanceof V1_BaseDataResolver) {
+    const resolver = new BaseDataResolver();
+    resolver.element = context.resolveElement(
+      protocol.elementPointer.path,
+      false,
+    );
+    resolver.data = V1_buildEmbeddedData(protocol.data, context);
+    return resolver;
+  } else if (protocol instanceof V1_ReferenceDataResolver) {
+    const resolver = new ReferenceDataResolver();
+    resolver.element = context.resolveElement(
+      protocol.elementPointer.path,
+      false,
+    );
+    return resolver;
+  }
+  throw new UnsupportedOperationError(
+    `Unable to build data resolver of type '${protocol.constructor.name}'`,
+  );
+};

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_TestBuilderHelper.ts

@@ -15,9 +15,14 @@
  */
 
 import { guaranteeType, UnsupportedOperationError } from '@finos/legend-shared';
-import { ExternalFormatData } from '../../../../../../../../graph/metamodel/pure/data/EmbeddedData.js';
+import {
+  ExternalFormatData,
+  RelationElement,
+  RelationRowTestData,
+} from '../../../../../../../../graph/metamodel/pure/data/EmbeddedData.js';
 import { EqualTo } from '../../../../../../../../graph/metamodel/pure/test/assertion/EqualTo.js';
 import { EqualToJson } from '../../../../../../../../graph/metamodel/pure/test/assertion/EqualToJson.js';
+import { EqualToRelation } from '../../../../../../../../graph/metamodel/pure/test/assertion/EqualToRelation.js';
 import { EqualToTDS } from '../../../../../../../../graph/metamodel/pure/test/assertion/EqualToTDS.js';
 import type { TestAssertion } from '../../../../../../../../graph/metamodel/pure/test/assertion/TestAssertion.js';
 import type {
@@ -29,6 +34,7 @@ import { V1_MappingTestSuite } from '../../../../model/packageableElements/mappi
 import { V1_ServiceTestSuite } from '../../../../model/packageableElements/service/V1_ServiceTestSuite.js';
 import { V1_EqualTo } from '../../../../model/test/assertion/V1_EqualTo.js';
 import { V1_EqualToJson } from '../../../../model/test/assertion/V1_EqualToJson.js';
+import { V1_EqualToRelation } from '../../../../model/test/assertion/V1_EqualToRelation.js';
 import { V1_EqualToTDS } from '../../../../model/test/assertion/V1_EqualToTDS.js';
 import type { V1_TestAssertion } from '../../../../model/test/assertion/V1_TestAssertion.js';
 import type { V1_TestSuite } from '../../../../model/test/V1_TestSuite.js';
@@ -80,6 +86,25 @@ const buildEqualToTDS = (
   return equalToTDS;
 };
 
+const buildEqualToRelation = (
+  element: V1_EqualToRelation,
+  parentTest: AtomicTest | undefined,
+): EqualToRelation => {
+  const equalToRelation = new EqualToRelation();
+  equalToRelation.id = element.id;
+  equalToRelation.parentTest = parentTest;
+  const rel = new RelationElement();
+  rel.columns = element.expected.columns;
+  rel.paths = element.expected.paths;
+  rel.rows = element.expected.rows.map((row) => {
+    const r = new RelationRowTestData();
+    r.values = row.values;
+    return r;
+  });
+  equalToRelation.expected = rel;
+  return equalToRelation;
+};
+
 export const V1_buildTestAssertion = (
   value: V1_TestAssertion,
   parentTest: AtomicTest | undefined,
@@ -91,6 +116,8 @@ export const V1_buildTestAssertion = (
     return V1_buildEqualToJson(value, parentTest, context);
   } else if (value instanceof V1_EqualToTDS) {
     return buildEqualToTDS(value, parentTest, context);
+  } else if (value instanceof V1_EqualToRelation) {
+    return buildEqualToRelation(value, parentTest);
   }
   throw new UnsupportedOperationError(`Can't build test assertion`, value);
 };

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_TestResultBuilderHelper.ts

@@ -25,6 +25,7 @@ import { AssertFail } from '../../../../../../../../graph/metamodel/pure/test/as
 import type { AssertionStatus } from '../../../../../../../../graph/metamodel/pure/test/assertion/status/AssertionStatus.js';
 import { AssertPass } from '../../../../../../../../graph/metamodel/pure/test/assertion/status/AssertPass.js';
 import { EqualToJsonAssertFail } from '../../../../../../../../graph/metamodel/pure/test/assertion/status/EqualToJsonAssertFail.js';
+import { EqualToRelationAssertFail } from '../../../../../../../../graph/metamodel/pure/test/assertion/status/EqualToRelationAssertFail.js';
 import {
   type TestResult,
   TestError,
@@ -40,6 +41,7 @@ import { V1_AssertFail } from '../../../../model/test/assertion/status/V1_Assert
 import type { V1_AssertionStatus } from '../../../../model/test/assertion/status/V1_AssertionStatus.js';
 import { V1_AssertPass } from '../../../../model/test/assertion/status/V1_AssertPass.js';
 import { V1_EqualToJsonAssertFail } from '../../../../model/test/assertion/status/V1_EqualToJsonAssertFail.js';
+import { V1_EqualToRelationAssertFail } from '../../../../model/test/assertion/status/V1_EqualToRelationAssertFail.js';
 import {
   type V1_TestResult,
   V1_TestExecuted,
@@ -176,12 +178,49 @@ const buildEqualToJsonAssertFail = (
   );
 };
 
+const buildEqualToRelationAssertFail = (
+  element: V1_EqualToRelationAssertFail,
+  atomicTest: AtomicTest,
+  plugins: PureProtocolProcessorPlugin[],
+): EqualToRelationAssertFail => {
+  let assertion = atomicTest.assertions.find((a) => a.id === element.id);
+  const extraAssertionBuilder = plugins.flatMap(
+    (plugin) =>
+      (
+        plugin as Testable_PureProtocolProcessorPlugin_Extension
+      ).V1_getExtraTestableAssertionBuilders?.() ?? [],
+  );
+
+  for (const builder of extraAssertionBuilder) {
+    const assertionBuilder = builder(atomicTest, element);
+    if (assertionBuilder) {
+      assertion = assertionBuilder;
+    }
+  }
+
+  if (assertion) {
+    const equalToRelationAssertFail = new EqualToRelationAssertFail(
+      assertion,
+      element.message,
+    );
+    equalToRelationAssertFail.expected = element.expected;
+    equalToRelationAssertFail.actual = element.actual;
+    return equalToRelationAssertFail;
+  }
+  throw new UnsupportedOperationError(
+    `Can't build EqualToRelationAssertFail: no compatible builder available from plugins`,
+    element,
+  );
+};
+
 const buildAssertionStatus = (
   value: V1_AssertionStatus,
   atomicTest: AtomicTest,
   plugins: PureProtocolProcessorPlugin[],
 ): AssertionStatus => {
-  if (value instanceof V1_EqualToJsonAssertFail) {
+  if (value instanceof V1_EqualToRelationAssertFail) {
+    return buildEqualToRelationAssertFail(value, atomicTest, plugins);
+  } else if (value instanceof V1_EqualToJsonAssertFail) {
     return buildEqualToJsonAssertFail(value, atomicTest, plugins);
   } else if (value instanceof V1_AssertFail) {
     return buildAssertFail(value, atomicTest, plugins);