@finos/legend-graph 30.0.8 → 30.0.10

package/src/graph-manager/protocol/pure/v1/V1_PureGraphManager.ts

@@ -34,11 +34,7 @@ import {
   promisify,
   StopWatch,
   isNonNullable,
-  addUniqueEntry,
-  uuid,
-  deleteEntry,
   uniq,
-  IllegalStateError,
   filterByType,
   isString,
 } from '@finos/legend-shared';
@@ -235,23 +231,12 @@ import {
 import { V1_UniqueTestId } from './model/test/V1_UniqueTestId.js';
 import type { RunTestsTestableInput } from '../../../../graph/metamodel/pure/test/result/RunTestsTestableInput.js';
 import { V1_buildTestsResult } from './engine/test/V1_RunTestsResult.js';
-import {
-  type TestResult,
-  TestError,
-  TestExecuted,
-  TestExecutionStatus,
-} from '../../../../graph/metamodel/pure/test/result/TestResult.js';
+import { type TestResult } from '../../../../graph/metamodel/pure/test/result/TestResult.js';
 import type { Testable } from '../../../../graph/metamodel/pure/test/Testable.js';
 import {
   getNullableIDFromTestable,
   getNullableTestable,
 } from '../../../helpers/DSL_Data_GraphManagerHelper.js';
-import type { TestAssertion } from '../../../../graph/metamodel/pure/test/assertion/TestAssertion.js';
-import { AssertFail } from '../../../../graph/metamodel/pure/test/assertion/status/AssertFail.js';
-import {
-  type AtomicTest,
-  TestSuite,
-} from '../../../../graph/metamodel/pure/test/Test.js';
 import { pruneSourceInformation } from '../../../../graph/MetaModelUtils.js';
 import {
   V1_buildModelCoverageAnalysisResult,
@@ -273,7 +258,6 @@ import { V1_transformParameterValue } from './transformation/pureGraph/from/V1_S
 import { V1_transformModelUnit } from './transformation/pureGraph/from/V1_DSL_ExternalFormat_Transformer.js';
 import type { ModelUnit } from '../../../../graph/metamodel/pure/packageableElements/externalFormat/store/DSL_ExternalFormat_ModelUnit.js';
 import { V1_LambdaReturnTypeInput } from './engine/compilation/V1_LambdaReturnType.js';
-import { MultiExecutionServiceTestResult } from '../../../../graph/metamodel/pure/packageableElements/service/MultiExecutionServiceTestResult.js';
 import type { ParameterValue } from '../../../../graph/metamodel/pure/packageableElements/service/ParameterValue.js';
 import type { Service } from '../../../../graph/metamodel/pure/packageableElements/service/Service.js';
 import { V1_ExecutionEnvironmentInstance } from './model/packageableElements/service/V1_ExecutionEnvironmentInstance.js';
@@ -2133,89 +2117,6 @@ export class V1_PureGraphManager extends AbstractPureGraphManager {
     return result;
   }
 
-  async generateExpectedResult(
-    testable: Testable,
-    test: AtomicTest,
-    baseAssertion: TestAssertion,
-    graph: PureModel,
-  ): Promise<AssertFail> {
-    const id = uuid();
-    try {
-      baseAssertion.id = id;
-      addUniqueEntry(test.assertions, baseAssertion);
-      const runTestsInput = new V1_RunTestsInput();
-      runTestsInput.model = this.getFullGraphModelData(graph);
-      const runTestableInput = new V1_RunTestsTestableInput();
-      const unitAtomicTest = new V1_UniqueTestId();
-      runTestableInput.testable = guaranteeNonNullable(
-        getNullableIDFromTestable(
-          testable,
-          graph,
-          this.pluginManager.getPureGraphManagerPlugins(),
-        ),
-      );
-      runTestableInput.unitTestIds = [unitAtomicTest];
-      runTestsInput.testables = [runTestableInput];
-      const parent = test.__parent;
-      unitAtomicTest.testSuiteId =
-        parent instanceof TestSuite ? parent.id : undefined;
-      unitAtomicTest.atomicTestId = test.id;
-      const runTestsResult = await this.engine.runTests(runTestsInput);
-      const results = V1_buildTestsResult(
-        runTestsResult,
-        (_id: string): Testable | undefined =>
-          getNullableTestable(
-            _id,
-            graph,
-            this.pluginManager.getPureGraphManagerPlugins(),
-          ),
-        this.pluginManager.getPureProtocolProcessorPlugins(),
-      );
-      const result = results[0];
-      let status: AssertFail | undefined = undefined;
-      if (
-        result instanceof TestExecuted &&
-        result.testExecutionStatus === TestExecutionStatus.FAIL
-      ) {
-        status = result.assertStatuses.find(
-          (aStatus) =>
-            aStatus.assertion === baseAssertion &&
-            aStatus instanceof AssertFail,
-        );
-      } else if (result instanceof MultiExecutionServiceTestResult) {
-        status = Array.from(result.keyIndexedTestResults.values())
-          .map((testResult) => {
-            if (
-              testResult instanceof TestExecuted &&
-              testResult.testExecutionStatus === TestExecutionStatus.FAIL
-            ) {
-              return testResult.assertStatuses.find(
-                (aStatus) =>
-                  aStatus.assertion === baseAssertion &&
-                  aStatus instanceof AssertFail,
-              );
-            } else if (testResult instanceof TestError) {
-              throw new IllegalStateError(testResult.error);
-            }
-            return undefined;
-          })
-          .filter(isNonNullable)[0];
-      } else if (result instanceof TestError) {
-        throw new IllegalStateError(result.error);
-      } else {
-        throw new UnsupportedOperationError(
-          'Unable to derive expected result from test result',
-        );
-      }
-      return guaranteeNonNullable(status);
-    } catch (error) {
-      assertErrorThrown(error);
-      throw error;
-    } finally {
-      deleteEntry(test.assertions, baseAssertion);
-    }
-  }
-
   // ------------------------------------------- Value Specification -------------------------------------------
 
   buildValueSpecification(

package/src/graph-manager/protocol/pure/v1/engine/execution/V1_ExecutionHelper.ts

@@ -19,6 +19,8 @@ import {
   UnsupportedOperationError,
   type PlainObject,
   isString,
+  isNumber,
+  isBoolean,
 } from '@finos/legend-shared';
 import {
   type ExecutionResult,
@@ -167,12 +169,12 @@ export const V1_buildExecutionResult = (
 };
 
 export const V1_serializeExecutionResult = (
-  json: PlainObject<V1_ExecutionResult> | string,
+  json: PlainObject<V1_ExecutionResult> | string | null | boolean | number,
 ): V1_ExecutionResult => {
-  if (isString(json)) {
+  if (json === null || isString(json) || isNumber(json) || isBoolean(json)) {
     return new V1_RawExecutionResult(json);
   }
-  switch ((json.builder as PlainObject<V1_ResultBuilder>)._type) {
+  switch ((json.builder as PlainObject<V1_ResultBuilder> | undefined)?._type) {
     case BuilderType.CLASS_BUILDER:
       return V1_ClassExecutionResult.serialization.fromJson(json);
     case BuilderType.TDS_BUILDER:

package/src/graph-manager/protocol/pure/v1/engine/execution/V1_ExecutionResult.ts

@@ -163,9 +163,9 @@ export class V1_ClassExecutionResult extends V1_ExecutionResult {
 }
 
 export class V1_RawExecutionResult extends V1_ExecutionResult {
-  value: string;
+  value: string | number | boolean | null;
 
-  constructor(value: string) {
+  constructor(value: string | number | boolean | null) {
     super();
     this.value = value;
   }

package/src/graph-manager/protocol/pure/v1/model/packageableElements/mapping/V1_MappingTest.ts

@@ -16,39 +16,15 @@
 
 import { hashArray, type Hashable } from '@finos/legend-shared';
 import { CORE_HASH_STRUCTURE } from '../../../../../../../graph/Core_HashUtils.js';
-import type { V1_RawLambda } from '../../rawValueSpecification/V1_RawLambda.js';
 import { V1_AtomicTest } from '../../test/V1_AtomicTest.js';
 import type { V1_MappingStoreTestData } from './V1_MappingStoreTestData.js';
 
-export abstract class V1_MappingTest
-  extends V1_AtomicTest
-  implements Hashable {}
-
-export class V1_MappingQueryTest extends V1_MappingTest implements Hashable {
-  /**
-   * Studio does not process value specification, they are left in raw JSON form
-   *
-   * @discrepancy model
-   */
-  func!: V1_RawLambda;
-
-  get hashCode(): string {
-    return hashArray([
-      CORE_HASH_STRUCTURE.MAPPING_QUERY_TEST,
-      this.id,
-      this.doc ?? '',
-      this.func,
-      hashArray(this.assertions),
-    ]);
-  }
-}
-
-export class V1_MappingDataTest extends V1_MappingTest implements Hashable {
+export class V1_MappingTest extends V1_AtomicTest implements Hashable {
   storeTestData: V1_MappingStoreTestData[] = [];
 
   get hashCode(): string {
     return hashArray([
-      CORE_HASH_STRUCTURE.MAPPING_DATA_TEST,
+      CORE_HASH_STRUCTURE.MAPPING_TEST,
       this.id,
       this.doc ?? '',
       hashArray(this.storeTestData),

package/src/graph-manager/protocol/pure/v1/model/packageableElements/mapping/V1_MappingTestSuite.ts

@@ -17,34 +17,9 @@
 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_MappingStoreTestData } from './V1_MappingStoreTestData.js';
 import type { V1_RawLambda } from '../../rawValueSpecification/V1_RawLambda.js';
 
-export abstract class V1_MappingTestSuite
-  extends V1_TestSuite
-  implements Hashable {}
-
-export class V1_MappingDataTestSuite
-  extends V1_MappingTestSuite
-  implements Hashable
-{
-  storeTestData: V1_MappingStoreTestData[] = [];
-
-  get hashCode(): string {
-    return hashArray([
-      CORE_HASH_STRUCTURE.MAPPING_TEST_DATA_SUITE,
-      this.id,
-      this.doc ?? '',
-      hashArray(this.storeTestData),
-      hashArray(this.tests),
-    ]);
-  }
-}
-
-export class V1_MappingQueryTestSuite
-  extends V1_MappingTestSuite
-  implements Hashable
-{
+export class V1_MappingTestSuite extends V1_TestSuite implements Hashable {
   /**
    * Studio does not process value specification, they are left in raw JSON form
    *
@@ -54,7 +29,7 @@ export class V1_MappingQueryTestSuite
 
   get hashCode(): string {
     return hashArray([
-      CORE_HASH_STRUCTURE.MAPPING_TEST_QUERY_SUITE,
+      CORE_HASH_STRUCTURE.MAPPING_TEST_SUITE,
       this.id,
       this.doc ?? '',
       this.func,

package/src/graph-manager/protocol/pure/v1/model/valueSpecification/V1_ValueSpecification.ts

@@ -34,6 +34,7 @@ import type { V1_PackageableElementPtr } from './raw/V1_PackageableElementPtr.js
 import type { V1_INTERNAL__UnknownValueSpecification } from './V1_INTERNAL__UnknownValueSpecfication.js';
 import type { V1_GenericTypeInstance } from './raw/V1_GenericTypeInstance.js';
 import type { V1_ClassInstance } from './raw/V1_ClassInstance.js';
+import type { V1_CByteArray } from './raw/V1_CByteArray.js';
 
 export interface V1_ValueSpecificationVisitor<T> {
   visit_INTERNAL__UnknownValueSpecfication(
@@ -56,6 +57,7 @@ export interface V1_ValueSpecificationVisitor<T> {
   visit_CDecimal(valueSpecification: V1_CDecimal): T;
   visit_CString(valueSpecification: V1_CString): T;
   visit_CBoolean(valueSpecification: V1_CBoolean): T;
+  visit_CByteArray(valueSpecification: V1_CByteArray): T;
   visit_CFloat(valueSpecification: V1_CFloat): T;
   visit_CDateTime(valueSpecification: V1_CDateTime): T;
   visit_CStrictDate(valueSpecification: V1_CStrictDate): T;

package/src/graph-manager/protocol/pure/v1/model/valueSpecification/raw/V1_CByteArray.ts

@@ -0,0 +1,28 @@
+/**
+ * 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 type { V1_ValueSpecificationVisitor } from '../V1_ValueSpecification.js';
+import { V1_PrimitiveValueSpecification } from './V1_PrimitiveValueSpecification.js';
+
+export class V1_CByteArray extends V1_PrimitiveValueSpecification {
+  value!: string; // the Base64String transformed from byte[] by Jackson
+
+  accept_ValueSpecificationVisitor<T>(
+    visitor: V1_ValueSpecificationVisitor<T>,
+  ): T {
+    return visitor.visit_CByteArray(this);
+  }
+}

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

@@ -158,31 +158,15 @@ import { FlatDataAssociationImplementation } from '../../../../../../../graph/me
 import { V1_FlatDataAssociationMapping } from '../../../model/packageableElements/store/flatData/mapping/V1_FlatDataAssociationMapping.js';
 import type { FlatDataAssociationPropertyMapping } from '../../../../../../../graph/metamodel/pure/packageableElements/store/flatData/mapping/FlatDataAssociationPropertyMapping.js';
 import { V1_FlatDataAssociationPropertyMapping } from '../../../model/packageableElements/store/flatData/mapping/V1_FlatDataAssociationPropertyMapping.js';
-import {
-  V1_MappingDataTest,
-  V1_MappingQueryTest,
-  type V1_MappingTest,
-} from '../../../model/packageableElements/mapping/V1_MappingTest.js';
+import { V1_MappingTest } from '../../../model/packageableElements/mapping/V1_MappingTest.js';
 import {
   V1_transformAtomicTest,
   V1_transformTestAssertion,
   V1_transformTestSuite,
 } from './V1_TestTransformer.js';
-import {
-  type MappingTest,
-  MappingDataTest,
-  MappingQueryTest,
-} from '../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTest.js';
-import {
-  MappingDataTestSuite,
-  MappingQueryTestSuite,
-  type MappingTestSuite,
-} from '../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTestSuite.js';
-import {
-  type V1_MappingTestSuite,
-  V1_MappingDataTestSuite,
-  V1_MappingQueryTestSuite,
-} from '../../../model/packageableElements/mapping/V1_MappingTestSuite.js';
+import type { MappingTest } from '../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTest.js';
+import { type MappingTestSuite } from '../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTestSuite.js';
+import { V1_MappingTestSuite } from '../../../model/packageableElements/mapping/V1_MappingTestSuite.js';
 import type { StoreTestData } from '../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingStoreTestData.js';
 import { V1_MappingStoreTestData } from '../../../model/packageableElements/mapping/V1_MappingStoreTestData.js';
 import { V1_transformEmbeddedData } from './V1_DataElementTransformer.js';
@@ -378,27 +362,11 @@ const transformMappingStoreTestData = (
   return testData;
 };
 
-export const V1_transformMappingQueryTest = (
-  element: MappingQueryTest,
-  context: V1_GraphTransformerContext,
-): V1_MappingQueryTest => {
-  const mappingTest = new V1_MappingQueryTest();
-  mappingTest.id = element.id;
-  mappingTest.doc = element.doc;
-  mappingTest.assertions = element.assertions.map((assertion) =>
-    V1_transformTestAssertion(assertion),
-  );
-  mappingTest.func = element.func.accept_RawValueSpecificationVisitor(
-    new V1_RawValueSpecificationTransformer(context),
-  ) as V1_RawLambda;
-  return mappingTest;
-};
-
-export const V1_transformMappingDataTest = (
-  element: MappingDataTest,
+export const V1_transformMappingTest = (
+  element: MappingTest,
   context: V1_GraphTransformerContext,
-): V1_MappingDataTest => {
-  const mappingTest = new V1_MappingDataTest();
+): V1_MappingTest => {
+  const mappingTest = new V1_MappingTest();
   mappingTest.id = element.id;
   mappingTest.doc = element.doc;
   mappingTest.assertions = element.assertions.map((assertion) =>
@@ -410,39 +378,11 @@ export const V1_transformMappingDataTest = (
   return mappingTest;
 };
 
-export const V1_transformMappingTest = (
-  element: MappingTest,
-  context: V1_GraphTransformerContext,
-): V1_MappingTest => {
-  if (element instanceof MappingQueryTest) {
-    return V1_transformMappingQueryTest(element, context);
-  } else if (element instanceof MappingDataTest) {
-    return V1_transformMappingDataTest(element, context);
-  }
-  throw new UnsupportedOperationError('Unsupported mapping test');
-};
-
-export const V1_transformMappingDataTestSuite = (
-  element: MappingDataTestSuite,
-  context: V1_GraphTransformerContext,
-): V1_MappingDataTestSuite => {
-  const mappingTestSuite = new V1_MappingDataTestSuite();
-  mappingTestSuite.id = element.id;
-  mappingTestSuite.doc = element.doc;
-  mappingTestSuite.storeTestData = element.storeTestData.map((testData) =>
-    transformMappingStoreTestData(testData, context),
-  );
-  mappingTestSuite.tests = element.tests.map((test) =>
-    V1_transformAtomicTest(test, context),
-  );
-  return mappingTestSuite;
-};
-
-export const V1_transformMappingQueryTestSuite = (
-  element: MappingQueryTestSuite,
+export const V1_transformMappingTestSuite = (
+  element: MappingTestSuite,
   context: V1_GraphTransformerContext,
-): V1_MappingQueryTestSuite => {
-  const mappingTestSuite = new V1_MappingQueryTestSuite();
+): V1_MappingTestSuite => {
+  const mappingTestSuite = new V1_MappingTestSuite();
   mappingTestSuite.id = element.id;
   mappingTestSuite.doc = element.doc;
   mappingTestSuite.func = element.func.accept_RawValueSpecificationVisitor(
@@ -454,20 +394,7 @@ export const V1_transformMappingQueryTestSuite = (
   return mappingTestSuite;
 };
 
-export const V1_transformMappingTestSuite = (
-  element: MappingTestSuite,
-  context: V1_GraphTransformerContext,
-): V1_MappingTestSuite => {
-  if (element instanceof MappingDataTestSuite) {
-    return V1_transformMappingDataTestSuite(element, context);
-  } else if (element instanceof MappingQueryTestSuite) {
-    return V1_transformMappingQueryTestSuite(element, context);
-  }
-  throw new UnsupportedOperationError('Unsupported mapping test suite');
-};
-
 // Include Mapping
-
 const transformMappingInclude = (
   element: MappingInclude,
   context: V1_GraphTransformerContext,

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

@@ -85,6 +85,7 @@ import { V1_GenericTypeInstance } from '../../../model/valueSpecification/raw/V1
 import { V1_ClassInstance } from '../../../model/valueSpecification/raw/V1_ClassInstance.js';
 import { V1_ClassInstanceType } from '../../pureProtocol/serializationHelpers/V1_ValueSpecificationSerializer.js';
 import type { KeyExpressionInstanceValue } from '../../../../../../../graph/metamodel/pure/valueSpecification/KeyExpressionInstanceValue.js';
+import { V1_CByteArray } from '../../../model/valueSpecification/raw/V1_CByteArray.js';
 
 class V1_ValueSpecificationTransformer
   implements ValueSpecificationVisitor<V1_ValueSpecification>
@@ -292,6 +293,11 @@ class V1_ValueSpecificationTransformer
         cBoolean.value = guaranteeIsBoolean(valueSpecification.values[0]);
         return cBoolean;
       }
+      case PRIMITIVE_TYPE.BYTE: {
+        const cByteArray = new V1_CByteArray();
+        cByteArray.value = guaranteeIsString(valueSpecification.values[0]);
+        return cByteArray;
+      }
       // since we don't have a corresponding protocol for abstract type `Date`, we will default to use `DateTime`
       case PRIMITIVE_TYPE.DATE:
       case PRIMITIVE_TYPE.DATETIME: {

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

@@ -80,27 +80,11 @@ import {
 import { getRelationalInputType } from '../../../../../../../../graph/helpers/STO_Relational_Helper.js';
 import { getEnumValue } from '../../../../../../../../graph/helpers/DomainHelper.js';
 import { PrimitiveType } from '../../../../../../../../graph/metamodel/pure/packageableElements/domain/PrimitiveType.js';
-import {
-  V1_MappingDataTestSuite,
-  V1_MappingQueryTestSuite,
-  type V1_MappingTestSuite,
-} from '../../../../model/packageableElements/mapping/V1_MappingTestSuite.js';
-import {
-  MappingDataTestSuite,
-  MappingQueryTestSuite,
-  type MappingTestSuite,
-} from '../../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTestSuite.js';
-import {
-  V1_MappingDataTest,
-  V1_MappingQueryTest,
-  type V1_MappingTest,
-} from '../../../../model/packageableElements/mapping/V1_MappingTest.js';
+import { type V1_MappingTestSuite } from '../../../../model/packageableElements/mapping/V1_MappingTestSuite.js';
+import { MappingTestSuite } from '../../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTestSuite.js';
+import { V1_MappingTest } from '../../../../model/packageableElements/mapping/V1_MappingTest.js';
 import type { TestSuite } from '../../../../../../../../graph/metamodel/pure/test/Test.js';
-import {
-  MappingDataTest,
-  MappingQueryTest,
-  type MappingTest,
-} from '../../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTest.js';
+import { MappingTest } from '../../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingTest.js';
 import { V1_buildTestAssertion } from './V1_TestBuilderHelper.js';
 import type { V1_MappingStoreTestData } from '../../../../model/packageableElements/mapping/V1_MappingStoreTestData.js';
 import { StoreTestData } from '../../../../../../../../graph/metamodel/pure/packageableElements/mapping/MappingStoreTestData.js';
@@ -273,32 +257,12 @@ const buildMappingStoreTestData = (
   return mappingStoreTestData;
 };
 
-export const V1_buildMappingQueryTest = (
-  element: V1_MappingQueryTest,
-  parentSuite: TestSuite,
-  context: V1_GraphBuilderContext,
-): MappingQueryTest => {
-  const mappingTest = new MappingQueryTest();
-  mappingTest.id = element.id;
-  mappingTest.doc = element.doc;
-  mappingTest.__parent = parentSuite;
-  mappingTest.assertions = element.assertions.map((assertion) =>
-    V1_buildTestAssertion(assertion, mappingTest, context),
-  );
-  mappingTest.func = V1_buildRawLambdaWithResolvedPaths(
-    element.func.parameters,
-    element.func.body,
-    context,
-  );
-  return mappingTest;
-};
-
-export const V1_buildMappingDataTest = (
-  element: V1_MappingDataTest,
+export const V1_buildMappingTest = (
+  element: V1_MappingTest,
   parentSuite: TestSuite,
   context: V1_GraphBuilderContext,
-): MappingDataTest => {
-  const mappingTest = new MappingDataTest();
+): MappingTest => {
+  const mappingTest = new MappingTest();
   mappingTest.id = element.id;
   mappingTest.__parent = parentSuite;
   mappingTest.doc = element.doc;
@@ -311,65 +275,28 @@ export const V1_buildMappingDataTest = (
   return mappingTest;
 };
 
-export const V1_buildMappingTest = (
-  element: V1_MappingTest,
-  parentSuite: TestSuite,
-  context: V1_GraphBuilderContext,
-): MappingTest => {
-  if (element instanceof V1_MappingDataTest) {
-    return V1_buildMappingDataTest(element, parentSuite, context);
-  } else if (element instanceof V1_MappingQueryTest) {
-    return V1_buildMappingQueryTest(element, parentSuite, context);
-  }
-  throw new UnsupportedOperationError(
-    'Unable to build mapping test: Unsupported mapping test type',
-  );
-};
-
-const V1_buildMappingDataTestSuite = (
-  element: V1_MappingDataTestSuite,
-  context: V1_GraphBuilderContext,
-): MappingDataTestSuite => {
-  const mappingTestSuite = new MappingDataTestSuite();
-  mappingTestSuite.id = element.id;
-  mappingTestSuite.storeTestData = element.storeTestData.map((testData) =>
-    buildMappingStoreTestData(testData, context),
-  );
-  mappingTestSuite.tests = element.tests.map((test) =>
-    V1_buildMappingTest(test, mappingTestSuite, context),
-  );
-  return mappingTestSuite;
-};
-
-const V1_buildMappingQueryTestSuite = (
-  element: V1_MappingQueryTestSuite,
+export const V1_buildMappingTestSuite = (
+  element: V1_MappingTestSuite,
   context: V1_GraphBuilderContext,
-): MappingQueryTestSuite => {
-  const mappingTestSuite = new MappingQueryTestSuite();
+): MappingTestSuite => {
+  const mappingTestSuite = new MappingTestSuite();
   mappingTestSuite.id = element.id;
   mappingTestSuite.func = V1_buildRawLambdaWithResolvedPaths(
     element.func.parameters,
     element.func.body,
     context,
   );
-  mappingTestSuite.tests = element.tests.map((test) =>
-    V1_buildMappingTest(test, mappingTestSuite, context),
-  );
+  mappingTestSuite.tests = element.tests.map((test) => {
+    if (test instanceof V1_MappingTest) {
+      return V1_buildMappingTest(test, mappingTestSuite, context);
+    }
+    throw new UnsupportedOperationError(
+      'Unable to build mapping test: Unsupported mapping test type',
+    );
+  });
   return mappingTestSuite;
 };
 
-export const V1_buildMappingTestSuite = (
-  element: V1_MappingTestSuite,
-  context: V1_GraphBuilderContext,
-): MappingTestSuite => {
-  if (element instanceof V1_MappingDataTestSuite) {
-    return V1_buildMappingDataTestSuite(element, context);
-  } else if (element instanceof V1_MappingQueryTestSuite) {
-    return V1_buildMappingQueryTestSuite(element, context);
-  }
-  throw new UnsupportedOperationError('Unsupported mapping test suite type');
-};
-
 const V1_buildMappingTestInputData = (
   inputData: V1_DEPRECATED__InputData,
   context: V1_GraphBuilderContext,

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

@@ -72,6 +72,7 @@ import type { V1_GraphFetchTree } from '../../../../model/valueSpecification/raw
 import type { V1_AppliedFunction } from '../../../../model/valueSpecification/application/V1_AppliedFunction.js';
 import type { V1_AppliedProperty } from '../../../../model/valueSpecification/application/V1_AppliedProperty.js';
 import type { V1_CBoolean } from '../../../../model/valueSpecification/raw/V1_CBoolean.js';
+import type { V1_CByteArray } from '../../../../model/valueSpecification/raw/V1_CByteArray.js';
 import type { V1_CDateTime } from '../../../../model/valueSpecification/raw/V1_CDateTime.js';
 import type { V1_CStrictTime } from '../../../../model/valueSpecification/raw/V1_CStrictTime.js';
 import type { V1_CDecimal } from '../../../../model/valueSpecification/raw/V1_CDecimal.js';
@@ -364,6 +365,16 @@ export class V1_ValueSpecificationBuilder
     );
   }
 
+  visit_CByteArray(valueSpecification: V1_CByteArray): ValueSpecification {
+    const res = buildPrimtiveInstanceValue(
+      PRIMITIVE_TYPE.BYTE,
+      [valueSpecification.value], // Store a Base64String as the value for BYTE
+      this.context,
+    );
+
+    return res;
+  }
+
   visit_CFloat(valueSpecification: V1_CFloat): ValueSpecification {
     return buildPrimtiveInstanceValue(
       PRIMITIVE_TYPE.FLOAT,

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

@@ -73,6 +73,7 @@ import type { V1_GraphBuilderContext } from '../V1_GraphBuilderContext.js';
 import type { V1_GenericTypeInstance } from '../../../../model/valueSpecification/raw/V1_GenericTypeInstance.js';
 import type { V1_ClassInstance } from '../../../../model/valueSpecification/raw/V1_ClassInstance.js';
 import type { V1_GraphFetchTree } from '../../../../model/valueSpecification/raw/classInstance/graph/V1_GraphFetchTree.js';
+import type { V1_CByteArray } from '../../../../model/valueSpecification/raw/V1_CByteArray.js';
 
 class V1_ValueSpecificationPathResolver
   implements V1_ValueSpecificationVisitor<V1_ValueSpecification>
@@ -193,6 +194,10 @@ class V1_ValueSpecificationPathResolver
     return spec;
   }
 
+  visit_CByteArray(spec: V1_CByteArray): V1_ValueSpecification {
+    return spec;
+  }
+
   visit_CFloat(spec: V1_CFloat): V1_ValueSpecification {
     return spec;
   }
@@ -292,10 +297,10 @@ function V1_resolveClassInstanceValueElementPaths(
   } else if (spec instanceof V1_TDSColumnInformation) {
     spec.columnFn.accept_ValueSpecificationVisitor(resolver);
     return spec;
-  } else if (spec instanceof V1_TDSOlapAggregation) {
-    spec.function.accept_ValueSpecificationVisitor(resolver);
-    return spec;
-  } else if (spec instanceof V1_TDSOlapRank) {
+  } else if (
+    spec instanceof V1_TDSOlapAggregation ||
+    spec instanceof V1_TDSOlapRank
+  ) {
     spec.function.accept_ValueSpecificationVisitor(resolver);
     return spec;
   } else if (spec instanceof V1_TDSSortInformation) {