@finos/legend-graph 31.4.1 → 31.5.0

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

@@ -54,6 +54,17 @@ import { V1_DerivedProperty } from '../../../model/packageableElements/domain/V1
 import type { V1_RawVariable } from '../../../model/rawValueSpecification/V1_RawVariable.js';
 import type { V1_GraphTransformerContext } from './V1_GraphTransformerContext.js';
 import { isStubbed_RawLambda } from '../../../../../../../graph/helpers/creator/RawValueSpecificationCreatorHelper.js';
+import type { FunctionTestSuite } from '../../../../../../../graph/metamodel/pure/packageableElements/function/test/FunctionTestSuite.js';
+import { V1_FunctionTestSuite } from '../../../model/packageableElements/function/test/V1_FunctionTestSuite.js';
+import { FunctionTest } from '../../../../../../../graph/metamodel/pure/packageableElements/function/test/FunctionTest.js';
+import {
+  V1_FunctionParameterValue,
+  V1_FunctionTest,
+} from '../../../model/packageableElements/function/test/V1_FunctionTest.js';
+import { UnsupportedOperationError } from '@finos/legend-shared';
+import { V1_FunctionStoreTestData } from '../../../model/packageableElements/function/test/V1_FunctionStoreTestData.js';
+import { V1_transformEmbeddedData } from './V1_DataElementTransformer.js';
+import { V1_transformTestAssertion } from './V1_TestTransformer.js';
 
 export const V1_transformProfile = (element: Profile): V1_Profile => {
   const profile = new V1_Profile();
@@ -224,6 +235,49 @@ export const V1_transformAssociation = (
   return association;
 };
 
+const V1_transformFunctionTest = (element: FunctionTest): V1_FunctionTest => {
+  const functionTest = new V1_FunctionTest();
+  functionTest.id = element.id;
+  functionTest.doc = element.doc;
+  functionTest.assertions = element.assertions.map((assertion) =>
+    V1_transformTestAssertion(assertion),
+  );
+  functionTest.parameters = element.parameters?.map((p) => {
+    const parameterValue = new V1_FunctionParameterValue();
+    parameterValue.name = p.name;
+    parameterValue.value = p.value;
+    return parameterValue;
+  });
+  return functionTest;
+};
+
+const V1_transformFunctionSuite = (
+  element: FunctionTestSuite,
+  context: V1_GraphTransformerContext,
+): V1_FunctionTestSuite => {
+  const testSuite = new V1_FunctionTestSuite();
+  testSuite.id = element.id;
+  testSuite.doc = element.doc;
+  if (element.testData?.length) {
+    testSuite.testData = element.testData.map((elementData) => {
+      const pTestData = new V1_FunctionStoreTestData();
+      pTestData.doc = elementData.doc;
+      pTestData.store = elementData.store.valueForSerialization ?? '';
+      pTestData.data = V1_transformEmbeddedData(elementData.data, context);
+      return pTestData;
+    });
+  }
+  testSuite.tests = element.tests.map((el) => {
+    if (el instanceof FunctionTest) {
+      return V1_transformFunctionTest(el);
+    }
+    throw new UnsupportedOperationError(
+      'Unsupported function test to transform',
+    );
+  });
+  return testSuite;
+};
+
 export const V1_transformFunction = (
   element: ConcreteFunctionDefinition,
   context: V1_GraphTransformerContext,
@@ -243,5 +297,10 @@ export const V1_transformFunction = (
   _function.returnMultiplicity = V1_transformMultiplicity(
     element.returnMultiplicity,
   );
+  if (element.tests.length) {
+    _function.tests = element.tests.map((suite) =>
+      V1_transformFunctionSuite(suite, context),
+    );
+  }
   return _function;
 };

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

@@ -14,9 +14,12 @@
  * limitations under the License.
  */
 
+import { UnsupportedOperationError } from '@finos/legend-shared';
 import type { SnowflakeAppDeploymentConfiguration } from '../../../../../../../graph/metamodel/pure/functionActivator/SnowflakeAppDeploymentConfiguration.js';
 import { V1_SnowflakeAppDeploymentConfiguration } from '../../../engine/functionActivator/V1_SnowflakeAppDeploymentConfiguration.js';
 import { V1_transformConnectionPointer } from './V1_ConnectionTransformer.js';
+import { V1_SnowflakeAppType } from '../../../engine/functionActivator/V1_SnowflakeAppType.js';
+import { SnowflakeAppType } from '../../../../../../../graph/metamodel/pure/functionActivator/SnowflakeAppType.js';
 
 export const V1_transformSnowflakeAppDeploymentConfiguration = (
   element: SnowflakeAppDeploymentConfiguration,
@@ -31,3 +34,19 @@ export const V1_transformSnowflakeAppDeploymentConfiguration = (
 
   return protocol;
 };
+
+export const V1_transformSnowflakeAppType = (
+  value: SnowflakeAppType,
+): V1_SnowflakeAppType => {
+  switch (value) {
+    case SnowflakeAppType.FULL:
+      return V1_SnowflakeAppType.FULL;
+    case SnowflakeAppType.STAGE:
+      return V1_SnowflakeAppType.STAGE;
+    default:
+      throw new UnsupportedOperationError(
+        `Can't transform SnowflakeApp type`,
+        value,
+      );
+  }
+};

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

@@ -78,7 +78,10 @@ import { V1_INTERNAL__UnknownStore } from '../../../model/packageableElements/st
 import { generateFunctionPrettyName } from '../../../../../../../graph/helpers/PureLanguageHelper.js';
 import { V1_SnowflakeApp } from '../../../model/packageableElements/function/V1_SnowflakeApp.js';
 import type { SnowflakeApp } from '../../../../../../../graph/metamodel/pure/packageableElements/function/SnowflakeApp.js';
-import { V1_transformSnowflakeAppDeploymentConfiguration } from './V1_FunctionActivatorTransformer.js';
+import {
+  V1_transformSnowflakeAppDeploymentConfiguration,
+  V1_transformSnowflakeAppType,
+} from './V1_FunctionActivatorTransformer.js';
 
 class V1_PackageableElementTransformer
   implements PackageableElementVisitor<V1_PackageableElement>
@@ -137,6 +140,7 @@ class V1_PackageableElementTransformer
     protocol.function = generateFunctionPrettyName(element.function.value, {
       fullPath: true,
       spacing: false,
+      notIncludeParamName: true,
     });
     protocol.applicationName = element.applicationName;
     protocol.description = element.description;
@@ -145,6 +149,9 @@ class V1_PackageableElementTransformer
       V1_transformSnowflakeAppDeploymentConfiguration(
         element.activationConfiguration,
       );
+    if (element.type) {
+      protocol.type = V1_transformSnowflakeAppType(element.type);
+    }
     return protocol;
   }
 

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

@@ -76,6 +76,7 @@ import type { V1_INTERNAL__UnknownStore } from '../../../model/packageableElemen
 import { INTERNAL__UnknownStore } from '../../../../../../../graph/metamodel/pure/packageableElements/store/INTERNAL__UnknownStore.js';
 import type { V1_SnowflakeApp } from '../../../model/packageableElements/function/V1_SnowflakeApp.js';
 import { SnowflakeApp } from '../../../../../../../graph/metamodel/pure/packageableElements/function/SnowflakeApp.js';
+import { V1_buildSnowflakeAppType } from './helpers/V1_FunctionActivatorBuilderHelper.js';
 
 export class V1_ElementFirstPassBuilder
   implements V1_PackageableElementVisitor<PackageableElement>
@@ -159,6 +160,9 @@ export class V1_ElementFirstPassBuilder
     metamodel.applicationName = element.applicationName;
     metamodel.description = element.description;
     metamodel.owner = element.owner;
+    if (element.type) {
+      metamodel.type = V1_buildSnowflakeAppType(element.type);
+    }
     return metamodel;
   }
 

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

@@ -50,6 +50,7 @@ import {
   V1_buildVariable,
   V1_buildUnit,
   V1_buildTaggedValue,
+  V1_buildFunctionSuite,
 } from './helpers/V1_DomainBuilderHelper.js';
 import {
   V1_buildServiceExecution,
@@ -156,6 +157,7 @@ export class V1_ElementSecondPassBuilder
             generateFunctionPrettyName(fn, {
               fullPath: true,
               spacing: false,
+              notIncludeParamName: true,
             }) === element.function.replaceAll(/\s*/gu, ''),
         ),
       ),
@@ -321,6 +323,11 @@ export class V1_ElementSecondPassBuilder
     );
     func.expressionSequence = protocol.body;
     func.functionName = getFunctionName(func, func.name);
+    if (protocol.tests?.length) {
+      func.tests = protocol.tests.map((l) =>
+        V1_buildFunctionSuite(l, this.context),
+      );
+    }
   }
 
   visit_FlatData(element: V1_FlatData): void {

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

@@ -58,6 +58,18 @@ import {
 } from '../../../../../../../../graph/helpers/DomainHelper.js';
 import { AggregationKind } from '../../../../../../../../graph/metamodel/pure/packageableElements/domain/AggregationKind.js';
 import { GraphBuilderError } from '../../../../../../../GraphManagerUtils.js';
+import type { V1_FunctionTestSuite } from '../../../../model/packageableElements/function/test/V1_FunctionTestSuite.js';
+import { FunctionTestSuite } from '../../../../../../../../graph/metamodel/pure/packageableElements/function/test/FunctionTestSuite.js';
+import type { V1_FunctionStoreTestData } from '../../../../model/packageableElements/function/test/V1_FunctionStoreTestData.js';
+import { FunctionStoreTestData } from '../../../../../../../../graph/metamodel/pure/packageableElements/function/test/FunctionStoreTestData.js';
+import { V1_buildEmbeddedData } from './V1_DataElementBuilderHelper.js';
+import { V1_FunctionTest } from '../../../../model/packageableElements/function/test/V1_FunctionTest.js';
+import {
+  FunctionParameterValue,
+  FunctionTest,
+} from '../../../../../../../../graph/metamodel/pure/packageableElements/function/test/FunctionTest.js';
+import { V1_buildTestAssertion } from './V1_TestBuilderHelper.js';
+import type { TestSuite } from '../../../../../../../../graph/metamodel/pure/test/Test.js';
 
 export const V1_buildTaggedValue = (
   taggedValue: V1_TaggedValue,
@@ -335,3 +347,59 @@ export const V1_getAppliedProperty = (
     `Property '${name}' not found in class '${parentClass.path}'`,
   );
 };
+
+// Function Suite
+const V1_buildFunctionTest = (
+  element: V1_FunctionTest,
+  parentSuite: TestSuite,
+  context: V1_GraphBuilderContext,
+): FunctionTest => {
+  const functionTest = new FunctionTest();
+  functionTest.id = element.id;
+  functionTest.__parent = parentSuite;
+  functionTest.doc = element.doc;
+  functionTest.assertions = element.assertions.map((assertion) =>
+    V1_buildTestAssertion(assertion, functionTest, context),
+  );
+  functionTest.parameters = element.parameters?.map((param) => {
+    const parameterValue = new FunctionParameterValue();
+    parameterValue.name = param.name;
+    parameterValue.value = param.value;
+    return parameterValue;
+  });
+  return functionTest;
+};
+
+const V1_buildFunctionStoreTestData = (
+  element: V1_FunctionStoreTestData,
+  context: V1_GraphBuilderContext,
+): FunctionStoreTestData => {
+  const storeTestData = new FunctionStoreTestData();
+  storeTestData.doc = element.doc;
+  storeTestData.store = context.resolveStore(element.store);
+  storeTestData.data = V1_buildEmbeddedData(element.data, context);
+  return storeTestData;
+};
+
+export const V1_buildFunctionSuite = (
+  element: V1_FunctionTestSuite,
+  context: V1_GraphBuilderContext,
+): FunctionTestSuite => {
+  const functionSuite = new FunctionTestSuite();
+  functionSuite.id = element.id;
+  functionSuite.doc = element.doc;
+  if (element.testData?.length) {
+    functionSuite.testData = element.testData.map((e) =>
+      V1_buildFunctionStoreTestData(e, context),
+    );
+  }
+  functionSuite.tests = element.tests.map((test) => {
+    if (test instanceof V1_FunctionTest) {
+      return V1_buildFunctionTest(test, functionSuite, context);
+    }
+    throw new UnsupportedOperationError(
+      'Unable to build function test: Unsupported function test type',
+    );
+  });
+  return functionSuite;
+};

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

@@ -19,7 +19,9 @@ import { SnowflakeAppDeploymentConfiguration } from '../../../../../../../../gra
 import type { V1_SnowflakeAppDeploymentConfiguration } from '../../../../engine/functionActivator/V1_SnowflakeAppDeploymentConfiguration.js';
 import type { V1_GraphBuilderContext } from '../V1_GraphBuilderContext.js';
 import { V1_buildConnection } from './V1_ConnectionBuilderHelper.js';
-import { guaranteeType } from '@finos/legend-shared';
+import { UnsupportedOperationError, guaranteeType } from '@finos/legend-shared';
+import { V1_SnowflakeAppType } from '../../../../engine/functionActivator/V1_SnowflakeAppType.js';
+import { SnowflakeAppType } from '../../../../../../../../graph/metamodel/pure/functionActivator/SnowflakeAppType.js';
 
 export const V1_buildSnowflakeAppDeploymentConfiguration = (
   element: V1_SnowflakeAppDeploymentConfiguration,
@@ -37,3 +39,19 @@ export const V1_buildSnowflakeAppDeploymentConfiguration = (
   }
   return metamodel;
 };
+
+export const V1_buildSnowflakeAppType = (
+  element: V1_SnowflakeAppType,
+): SnowflakeAppType => {
+  switch (element) {
+    case V1_SnowflakeAppType.FULL:
+      return SnowflakeAppType.FULL;
+    case V1_SnowflakeAppType.STAGE:
+      return SnowflakeAppType.STAGE;
+    default:
+      throw new UnsupportedOperationError(
+        `Can't build V1_SnowflakeApp type`,
+        element,
+      );
+  }
+};

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

@@ -74,6 +74,7 @@ import {
 } from '../../../../model/packageableElements/store/relational/connection/V1_AuthenticationStrategy.js';
 import type { STO_Relational_PureProtocolProcessorPlugin_Extension } from '../../../../../extensions/STO_Relational_PureProtocolProcessorPlugin_Extension.js';
 import { INTERNAL__UnknownDatasourceSpecification } from '../../../../../../../../graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownDatasourceSpecification.js';
+import { INTERNAL__UnknownAuthenticationStrategy } from '../../../../../../../../graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownAuthenticationStrategy.js';
 import { V1_INTERNAL__UnknownDatasourceSpecification } from '../../../../model/packageableElements/store/relational/connection/V1_INTERNAL__UnknownDatasourceSpecification.js';
 import { V1_INTERNAL__UnknownAuthenticationStrategy } from '../../../../model/packageableElements/store/relational/connection/V1_INTERNAL__UnknownAuthenticationStrategy.js';
 
@@ -306,7 +307,7 @@ export const V1_buildAuthenticationStrategy = (
   context: V1_GraphBuilderContext,
 ): AuthenticationStrategy => {
   if (protocol instanceof V1_INTERNAL__UnknownAuthenticationStrategy) {
-    const metamodel = new V1_INTERNAL__UnknownAuthenticationStrategy();
+    const metamodel = new INTERNAL__UnknownAuthenticationStrategy();
     metamodel.content = protocol.content;
     return metamodel;
   } else if (protocol instanceof V1_DefaultH2AuthenticationStrategy) {

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/V1_PackageableElementSerialization.ts

@@ -193,7 +193,7 @@ class V1_PackageableElementSerializer
   visit_ConcreteFunctionDefinition(
     element: V1_ConcreteFunctionDefinition,
   ): PlainObject<V1_PackageableElement> {
-    return serialize(V1_functionModelSchema, element);
+    return serialize(V1_functionModelSchema(this.plugins), element);
   }
 
   visit_FlatData(element: V1_FlatData): PlainObject<V1_PackageableElement> {
@@ -304,7 +304,7 @@ export const V1_deserializePackageableElement = (
       case V1_ASSOCIATION_ELEMENT_PROTOCOL_TYPE:
         return deserialize(V1_associationModelSchema, json);
       case V1_FUNCTION_ELEMENT_PROTOCOL_TYPE:
-        return deserialize(V1_functionModelSchema, json);
+        return deserialize(V1_functionModelSchema(plugins), json);
       case V1_FLAT_DATA_ELEMENT_PROTOCOL_TYPE:
         return deserialize(V1_flatDataModelSchema, json);
       case V1_DATABASE_ELEMENT_PROTOCOL_TYPE:

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_DomainSerializationHelper.ts

@@ -22,11 +22,13 @@ import {
   alias,
   optional,
   SKIP,
+  type ModelSchema,
 } from 'serializr';
 import {
   customEquivalentList,
   customList,
   customListWithSchema,
+  optionalCustomList,
   usingConstantValueSchema,
   usingModelSchema,
 } from '@finos/legend-shared';
@@ -55,6 +57,12 @@ import {
 import { V1_INTERNAL__UnknownFunctionActivator } from '../../../model/packageableElements/function/V1_INTERNAL__UnknownFunctionActivator.js';
 import { V1_SnowflakeApp } from '../../../model/packageableElements/function/V1_SnowflakeApp.js';
 import { V1_SnowflakeAppDeploymentConfigurationAppModelSchema } from './V1_FunctionActivatorSerializationHelper.js';
+import {
+  V1_deserializeTestSuite,
+  V1_serializeTestSuite,
+} from './V1_TestSerializationHelper.js';
+import type { PureProtocolProcessorPlugin } from '../../../../PureProtocolProcessorPlugin.js';
+import type { V1_TestSuite } from '../../../model/test/V1_TestSuite.js';
 
 export const V1_CLASS_ELEMENT_PROTOCOL_TYPE = 'class';
 export const V1_PROFILE_ELEMENT_PROTOCOL_TYPE = 'profile';
@@ -158,6 +166,7 @@ export const V1_snowflakeAppModelSchema = createModelSchema(V1_SnowflakeApp, {
   activationConfiguration: usingModelSchema(
     V1_SnowflakeAppDeploymentConfigurationAppModelSchema,
   ),
+  type: optional(primitive()),
 });
 
 // ------------------------------------- Class -------------------------------------
@@ -280,9 +289,10 @@ export const V1_associationModelSchema = createModelSchema(V1_Association, {
 
 // ------------------------------------- Function -------------------------------------
 
-export const V1_functionModelSchema = createModelSchema(
-  V1_ConcreteFunctionDefinition,
-  {
+export const V1_functionModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_ConcreteFunctionDefinition> =>
+  createModelSchema(V1_ConcreteFunctionDefinition, {
     _type: usingConstantValueSchema(V1_FUNCTION_ELEMENT_PROTOCOL_TYPE),
     body: list(raw()),
     name: primitive(),
@@ -298,8 +308,11 @@ export const V1_functionModelSchema = createModelSchema(
     taggedValues: customListWithSchema(V1_taggedValueModelSchema, {
       INTERNAL__forceReturnEmptyInTest: true,
     }),
-  },
-);
+    tests: optionalCustomList(
+      (value: V1_TestSuite) => V1_serializeTestSuite(value, plugins),
+      (value) => V1_deserializeTestSuite(value, plugins),
+    ),
+  });
 
 export const V1_INTERNAL__UnknownFunctionActivatorModelSchema =
   createModelSchema(V1_INTERNAL__UnknownFunctionActivator, {

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_FunctionSeriaizationHelper.ts

@@ -0,0 +1,100 @@
+/**
+ * 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 {
+  createModelSchema,
+  type ModelSchema,
+  optional,
+  primitive,
+  list,
+  custom,
+  raw,
+} from 'serializr';
+import type { PureProtocolProcessorPlugin } from '../../../../PureProtocolProcessorPlugin.js';
+import { V1_FunctionTestSuite } from '../../../model/packageableElements/function/test/V1_FunctionTestSuite.js';
+import {
+  V1_TestSuiteType,
+  V1_deserializeAtomicTest,
+  V1_deserializeTestAssertion,
+  V1_serializeAtomicTest,
+  V1_serializeTestAssertion,
+} from './V1_TestSerializationHelper.js';
+import {
+  customListWithSchema,
+  usingConstantValueSchema,
+  usingModelSchema,
+} from '@finos/legend-shared';
+import { V1_FunctionStoreTestData } from '../../../model/packageableElements/function/test/V1_FunctionStoreTestData.js';
+import {
+  V1_deserializeEmbeddedDataType,
+  V1_serializeEmbeddedDataType,
+} from './V1_DataElementSerializationHelper.js';
+import {
+  V1_FunctionParameterValue,
+  V1_FunctionTest,
+} from '../../../model/packageableElements/function/test/V1_FunctionTest.js';
+import { ATOMIC_TEST_TYPE } from '../../../../../../../graph/MetaModelConst.js';
+
+export const V1_parameterValueModelSchema = createModelSchema(
+  V1_FunctionParameterValue,
+  {
+    name: primitive(),
+    value: raw(),
+  },
+);
+
+export const V1_functionTestModelSchema = createModelSchema(V1_FunctionTest, {
+  _type: usingConstantValueSchema(ATOMIC_TEST_TYPE.Function_Test),
+  assertions: list(
+    custom(
+      (val) => V1_serializeTestAssertion(val),
+      (val) => V1_deserializeTestAssertion(val),
+    ),
+  ),
+  id: primitive(),
+  doc: optional(primitive()),
+  parameters: customListWithSchema(V1_parameterValueModelSchema),
+});
+
+const V1_FunctionStoreTestDataModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_FunctionStoreTestData> =>
+  createModelSchema(V1_FunctionStoreTestData, {
+    data: custom(
+      (val) => V1_serializeEmbeddedDataType(val, plugins),
+      (val) => V1_deserializeEmbeddedDataType(val, plugins),
+    ),
+    doc: optional(primitive()),
+    store: primitive(),
+  });
+
+export const V1_functionTestSuiteModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_FunctionTestSuite> =>
+  createModelSchema(V1_FunctionTestSuite, {
+    _type: usingConstantValueSchema(V1_TestSuiteType.FUNCTION_TEST_SUITE),
+    doc: optional(primitive()),
+    id: primitive(),
+    testData: optional(
+      list(usingModelSchema(V1_FunctionStoreTestDataModelSchema(plugins))),
+    ),
+    tests: list(
+      custom(
+        (val) => V1_serializeAtomicTest(val, plugins),
+        (val) => V1_deserializeAtomicTest(val, plugins),
+      ),
+    ),
+  });

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_TestSerializationHelper.ts

@@ -65,6 +65,12 @@ import {
   V1_serviceTestModelSchema,
   V1_serviceTestSuiteModelSchema,
 } from './V1_ServiceSerializationHelper.js';
+import { V1_FunctionTestSuite } from '../../../model/packageableElements/function/test/V1_FunctionTestSuite.js';
+import {
+  V1_functionTestModelSchema,
+  V1_functionTestSuiteModelSchema,
+} from './V1_FunctionSeriaizationHelper.js';
+import { V1_FunctionTest } from '../../../model/packageableElements/function/test/V1_FunctionTest.js';
 
 enum V1_AssertionStatusType {
   ASSERT_FAIL = 'assertFail',
@@ -89,6 +95,7 @@ enum V1_TestResultType {
 export enum V1_TestSuiteType {
   SERVICE_TEST_SUITE = 'serviceTestSuite',
   MAPPING_TEST_SUITE = 'mappingTestSuite',
+  FUNCTION_TEST_SUITE = 'functionTestSuite',
 }
 
 export const V1_uniqueTestIdModelSchema = createModelSchema(V1_UniqueTestId, {
@@ -233,6 +240,8 @@ export const V1_serializeAtomicTest = (
     return serialize(V1_serviceTestModelSchema, protocol);
   } else if (protocol instanceof V1_MappingTest) {
     return serialize(V1_mappingTestModelSchema(plugins), protocol);
+  } else if (protocol instanceof V1_FunctionTest) {
+    return serialize(V1_functionTestModelSchema, protocol);
   }
   const extraAtomicTestSerializers = plugins.flatMap(
     (plugin) =>
@@ -260,6 +269,8 @@ export const V1_deserializeAtomicTest = (
   switch (json._type) {
     case ATOMIC_TEST_TYPE.Service_Test:
       return deserialize(V1_serviceTestModelSchema, json);
+    case ATOMIC_TEST_TYPE.Function_Test:
+      return deserialize(V1_functionTestModelSchema, json);
     case ATOMIC_TEST_TYPE.Mapping_Test:
       return deserialize(V1_mappingTestModelSchema(plugins), json);
     default: {
@@ -323,6 +334,8 @@ export const V1_serializeTestSuite = (
     return serialize(V1_serviceTestSuiteModelSchema(plugins), protocol);
   } else if (protocol instanceof V1_MappingTestSuite) {
     return serialize(V1_mappingTestSuiteModelSchema(plugins), protocol);
+  } else if (protocol instanceof V1_FunctionTestSuite) {
+    return serialize(V1_functionTestSuiteModelSchema(plugins), protocol);
   }
   throw new UnsupportedOperationError(`Can't serialize test suite`, protocol);
 };
@@ -336,6 +349,8 @@ export const V1_deserializeTestSuite = (
       return deserialize(V1_serviceTestSuiteModelSchema(plugins), json);
     case V1_TestSuiteType.MAPPING_TEST_SUITE:
       return deserialize(V1_mappingTestSuiteModelSchema(plugins), json);
+    case V1_TestSuiteType.FUNCTION_TEST_SUITE:
+      return deserialize(V1_functionTestSuiteModelSchema(plugins), json);
     default:
       throw new UnsupportedOperationError(
         `Can't deserialize test suite of type '${json._type}'`,

package/src/index.ts

@@ -77,6 +77,8 @@ export { FunctionActivator } from './graph/metamodel/pure/packageableElements/fu
 export { INTERNAL__UnknownFunctionActivator } from './graph/metamodel/pure/packageableElements/function/INTERNAL__UnknownFunctionActivator.js';
 export { SnowflakeApp } from './graph/metamodel/pure/packageableElements/function/SnowflakeApp.js';
 export { SnowflakeAppDeploymentConfiguration } from './graph/metamodel/pure/functionActivator/SnowflakeAppDeploymentConfiguration.js';
+export { SnowflakeAppType } from './graph/metamodel/pure/functionActivator/SnowflakeAppType.js';
+
 // --------------------------------------------- VALUE SPECIFICATION --------------------------------------------------
 
 // metamodels
@@ -510,7 +512,9 @@ export * from './graph/metamodel/pure/packageableElements/mapping/MappingTest.js
 export * from './graph/metamodel/pure/packageableElements/mapping/MappingTestSuite.js';
 export * from './graph/metamodel/pure/packageableElements/mapping/MappingStoreTestData.js';
 export { type EmbeddedSetImplementation } from './graph/metamodel/pure/packageableElements/mapping/EmbeddedSetImplementation.js';
-
+export * from './graph/metamodel/pure/packageableElements/function/test/FunctionTestSuite.js';
+export * from './graph/metamodel/pure/packageableElements/function/test/FunctionStoreTestData.js';
+export * from './graph/metamodel/pure/packageableElements/function/test/FunctionTest.js';
 export {
   type RawMappingModelCoverageAnalysisResult,
   MappingModelCoverageAnalysisResult,

package/tsconfig.json

@@ -114,6 +114,7 @@
     "./src/graph/metamodel/pure/executionPlan/result/VoidResultType.ts",
     "./src/graph/metamodel/pure/functionActivator/DeploymentConfiguration.ts",
     "./src/graph/metamodel/pure/functionActivator/SnowflakeAppDeploymentConfiguration.ts",
+    "./src/graph/metamodel/pure/functionActivator/SnowflakeAppType.ts",
     "./src/graph/metamodel/pure/packageableElements/INTERNAL__UnknownPackageableElement.ts",
     "./src/graph/metamodel/pure/packageableElements/PackageableElement.ts",
     "./src/graph/metamodel/pure/packageableElements/PackageableElementReference.ts",
@@ -162,6 +163,9 @@
     "./src/graph/metamodel/pure/packageableElements/function/FunctionActivator.ts",
     "./src/graph/metamodel/pure/packageableElements/function/INTERNAL__UnknownFunctionActivator.ts",
     "./src/graph/metamodel/pure/packageableElements/function/SnowflakeApp.ts",
+    "./src/graph/metamodel/pure/packageableElements/function/test/FunctionStoreTestData.ts",
+    "./src/graph/metamodel/pure/packageableElements/function/test/FunctionTest.ts",
+    "./src/graph/metamodel/pure/packageableElements/function/test/FunctionTestSuite.ts",
     "./src/graph/metamodel/pure/packageableElements/generationSpecification/AbstractGenerationSpecification.ts",
     "./src/graph/metamodel/pure/packageableElements/generationSpecification/GenerationSpecification.ts",
     "./src/graph/metamodel/pure/packageableElements/generationSpecification/ModelGenerationSpecification.ts",
@@ -412,6 +416,7 @@
     "./src/graph-manager/protocol/pure/v1/engine/functionActivator/V1_FunctionActivatorInfo.ts",
     "./src/graph-manager/protocol/pure/v1/engine/functionActivator/V1_FunctionActivatorInput.ts",
     "./src/graph-manager/protocol/pure/v1/engine/functionActivator/V1_SnowflakeAppDeploymentConfiguration.ts",
+    "./src/graph-manager/protocol/pure/v1/engine/functionActivator/V1_SnowflakeAppType.ts",
     "./src/graph-manager/protocol/pure/v1/engine/generation/V1_ArtifactGenerationExtensionApi.ts",
     "./src/graph-manager/protocol/pure/v1/engine/generation/V1_DatabaseBuilderInput.ts",
     "./src/graph-manager/protocol/pure/v1/engine/generation/V1_FileGenerationInput.ts",
@@ -527,6 +532,9 @@
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/function/V1_FunctionActivator.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/function/V1_INTERNAL__UnknownFunctionActivator.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/function/V1_SnowflakeApp.ts",
+    "./src/graph-manager/protocol/pure/v1/model/packageableElements/function/test/V1_FunctionStoreTestData.ts",
+    "./src/graph-manager/protocol/pure/v1/model/packageableElements/function/test/V1_FunctionTest.ts",
+    "./src/graph-manager/protocol/pure/v1/model/packageableElements/function/test/V1_FunctionTestSuite.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/generationSpecification/V1_AbstractGenerationSpecification.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/generationSpecification/V1_GenerationSpecification.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/generationSpecification/V1_ModelGenerationSpecification.ts",
@@ -767,6 +775,7 @@
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_ExecutionEnvironmentSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_FileGenerationSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_FunctionActivatorSerializationHelper.ts",
+    "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_FunctionSeriaizationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_GenerationSpecificationSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_MappingSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_MilestoningSerializationHelper.ts",