@finos/legend-extension-dsl-persistence 3.0.15 → 4.0.1

package/src/graphManager/protocol/pure/v1/transformation/pureGraph/to/{V1_PersistenceBuilder.ts → V1_DSL_Persistence_BuilderHelper.ts}

@@ -18,14 +18,14 @@ import {
   type V1_Auditing,
   V1_DateTimeAuditing,
   V1_NoAuditing,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_Auditing.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Auditing.js';
 import {
   V1_AnyVersionDeduplicationStrategy,
   type V1_DeduplicationStrategy,
   V1_DuplicateCountDeduplicationStrategy,
   V1_MaxVersionDeduplicationStrategy,
   V1_NoDeduplicationStrategy,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_DeduplicationStrategy.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_DeduplicationStrategy.js';
 import {
   V1_AppendOnly,
   V1_BitemporalDelta,
@@ -35,12 +35,12 @@ import {
   V1_NontemporalSnapshot,
   V1_UnitemporalDelta,
   V1_UnitemporalSnapshot,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_IngestMode.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_IngestMode.js';
 import {
   V1_DeleteIndicatorMergeStrategy,
   type V1_MergeStrategy,
   V1_NoDeletesMergeStrategy,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_MergeStrategy.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_MergeStrategy.js';
 import {
   V1_BatchIdAndDateTimeTransactionMilestoning,
   V1_BatchIdTransactionMilestoning,
@@ -54,44 +54,44 @@ import {
   type V1_TransactionMilestoning,
   type V1_ValidityDerivation,
   type V1_ValidityMilestoning,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_Milestoning.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Milestoning.js';
 import {
   V1_EmailNotifyee,
   type V1_Notifier,
   type V1_Notifyee,
   V1_PagerDutyNotifyee,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_Notifier.js';
-import type { V1_Persistence } from '../../../model/packageableElements/persistence/V1_DSLPersistence_Persistence.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Notifier.js';
+import type { V1_Persistence } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Persistence.js';
 import {
   V1_BatchPersister,
   type V1_Persister,
   V1_StreamingPersister,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_Persister.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Persister.js';
 import {
   V1_ObjectStorageSink,
   V1_RelationalSink,
   type V1_Sink,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_Sink.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Sink.js';
 import {
   V1_FlatTarget,
   V1_MultiFlatTarget,
   type V1_MultiFlatTargetPart,
   type V1_TargetShape,
   V1_TransactionScope,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_TargetShape.js';
-import type { V1_Trigger } from '../../../model/packageableElements/persistence/V1_DSLPersistence_Trigger.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_TargetShape.js';
+import type { V1_Trigger } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Trigger.js';
 import {
   type Auditing,
   DateTimeAuditing,
   NoAuditing,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Auditing.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Auditing.js';
 import {
   AnyVersionDeduplicationStrategy,
   type DeduplicationStrategy,
   DuplicateCountDeduplicationStrategy,
   MaxVersionDeduplicationStrategy,
   NoDeduplicationStrategy,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_DeduplicationStrategy.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_DeduplicationStrategy.js';
 import {
   AppendOnly,
   BitemporalDelta,
@@ -101,12 +101,12 @@ import {
   NontemporalSnapshot,
   UnitemporalDelta,
   UnitemporalSnapshot,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_IngestMode.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_IngestMode.js';
 import {
   DeleteIndicatorMergeStrategy,
   type MergeStrategy,
   NoDeletesMergeStrategy,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_MergeStrategy.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_MergeStrategy.js';
 import {
   BatchIdAndDateTimeTransactionMilestoning,
   BatchIdTransactionMilestoning,
@@ -120,44 +120,59 @@ import {
   type TransactionMilestoning,
   type ValidityDerivation,
   type ValidityMilestoning,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Milestoning.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Milestoning.js';
 import {
   EmailNotifyee,
   Notifier,
   type Notifyee,
   PagerDutyNotifyee,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Notifier.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Notifier.js';
 import {
   BatchPersister,
   type Persister,
   StreamingPersister,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Persister.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Persister.js';
 import {
   ObjectStorageSink,
   RelationalSink,
   type Sink,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Sink.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Sink.js';
 import {
   FlatTarget,
   MultiFlatTarget,
   MultiFlatTargetPart,
   type TargetShape,
   TransactionScope,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_TargetShape.js';
-import type { Trigger } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Trigger.js';
-import { getOwnPersistence } from '../../../../../../DSLPersistence_GraphManagerHelper.js';
-import type { DSLPersistence_PureProtocolProcessorPlugin_Extension } from '../../../../DSLPersistence_PureProtocolProcessorPlugin_Extension.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_TargetShape.js';
+import type { Trigger } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Trigger.js';
+import { getOwnPersistence } from '../../../../../../DSL_Persistence_GraphManagerHelper.js';
+import type { DSL_Persistence_PureProtocolProcessorPlugin_Extension } from '../../../../DSL_Persistence_PureProtocolProcessorPlugin_Extension.js';
 import {
+  type AtomicTest,
   type Binding,
   type Database,
   type PackageableElementImplicitReference,
-  type V1_GraphBuilderContext,
+  type TestAssertion,
+  V1_buildAtomicTest,
+  V1_buildEmbeddedData,
+  V1_buildEqualToJson,
   V1_buildFullPath,
+  V1_EqualToJson,
+  type V1_GraphBuilderContext,
+  type V1_TestAssertion,
 } from '@finos/legend-graph';
 import {
   guaranteeNonEmptyString,
+  guaranteeType,
   UnsupportedOperationError,
 } from '@finos/legend-shared';
+import { PersistenceTest } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceTest.js';
+import type { V1_PersistenceTestBatch } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestBatch.js';
+import { PersistenceTestBatch } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceTestBatch.js';
+import type { V1_PersistenceTestData } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestData.js';
+import { PersistenceTestData } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceTestData.js';
+import type { V1_ConnectionTestData } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_ConnectionTestData.js';
+import { ConnectionTestData } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_ConnectionTestData.js';
 
 /**********
  * trigger
@@ -170,7 +185,7 @@ export const V1_buildTrigger = (
   const extraTriggerBuilders = context.extensions.plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraTriggerBuilders?.() ?? [],
   );
   for (const builder of extraTriggerBuilders) {
@@ -621,6 +636,62 @@ export const V1_buildNotifier = (
   return notifier;
 };
 
+/**********
+ * test
+ **********/
+
+export const V1_buildPersistenceTestAssertions = (
+  value: V1_TestAssertion[],
+  parentTest: AtomicTest | undefined,
+  context: V1_GraphBuilderContext,
+): TestAssertion[] =>
+  value.map((assertion) => {
+    if (assertion instanceof V1_EqualToJson) {
+      return V1_buildEqualToJson(assertion, parentTest, context);
+    } else {
+      throw new UnsupportedOperationError(`Can't build test assertion`, value);
+    }
+  });
+
+const V1_buildConnectionTestData = (
+  element: V1_ConnectionTestData,
+  context: V1_GraphBuilderContext,
+): ConnectionTestData => {
+  const connectionTestData = new ConnectionTestData();
+  connectionTestData.data = V1_buildEmbeddedData(element.data, context);
+  return connectionTestData;
+};
+
+const V1_buildTestData = (
+  element: V1_PersistenceTestData,
+  context: V1_GraphBuilderContext,
+): PersistenceTestData => {
+  const testData = new PersistenceTestData();
+  testData.connection = V1_buildConnectionTestData(element.connection, context);
+  return testData;
+};
+
+export const V1_buildTestBatch = (
+  protocol: V1_PersistenceTestBatch[],
+  parentTest: PersistenceTest | undefined,
+  context: V1_GraphBuilderContext,
+): PersistenceTestBatch[] =>
+  protocol.map((persistenceTestBatch) => {
+    const testBatch = new PersistenceTestBatch();
+    testBatch.id = persistenceTestBatch.id;
+    testBatch.batchId = persistenceTestBatch.batchId;
+    testBatch.assertions = V1_buildPersistenceTestAssertions(
+      persistenceTestBatch.assertions,
+      parentTest,
+      context,
+    );
+    testBatch.testData = V1_buildTestData(
+      persistenceTestBatch.testData,
+      context,
+    );
+    return testBatch;
+  });
+
 /**********
  * persistence
  **********/
@@ -639,4 +710,7 @@ export const V1_buildPersistence = (
   persistence.service = context.resolveService(protocol.service);
   persistence.persister = V1_buildPersister(protocol.persister, context);
   persistence.notifier = V1_buildNotifier(protocol.notifier, context);
+  persistence.tests = protocol.tests
+    .map((test) => V1_buildAtomicTest(test, context))
+    .map((e) => guaranteeType(e, PersistenceTest));
 };

package/src/graphManager/protocol/pure/v1/transformation/pureGraph/to/{V1_PersistenceContextBuilder.ts → V1_DSL_Persistence_ContextBuilder.ts}

@@ -14,30 +14,30 @@
  * limitations under the License.
  */
 
-import type { V1_PersistenceContext } from '../../../model/packageableElements/persistence/V1_DSLPersistence_PersistenceContext.js';
+import type { V1_PersistenceContext } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceContext.js';
 import {
   type V1_PersistencePlatform,
   V1_DefaultPersistencePlatform,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_PersistencePlatform.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistencePlatform.js';
 import {
   V1_ConnectionValue,
   V1_PrimitiveTypeValue,
   type V1_ServiceParameter,
   type V1_ServiceParameterValue,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_ServiceParameter.js';
-import type { Persistence } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Persistence.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_ServiceParameter.js';
+import type { Persistence } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Persistence.js';
 import {
   DefaultPersistencePlatform,
   type PersistencePlatform,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_PersistencePlatform.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistencePlatform.js';
 import {
   ConnectionValue,
   PrimitiveTypeValue,
   ServiceParameter,
   type ServiceParameterValue,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_ServiceParameter.js';
-import { getOwnPersistenceContext } from '../../../../../../DSLPersistence_GraphManagerHelper.js';
-import type { DSLPersistence_PureProtocolProcessorPlugin_Extension } from '../../../../DSLPersistence_PureProtocolProcessorPlugin_Extension.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_ServiceParameter.js';
+import { getOwnPersistenceContext } from '../../../../../../DSL_Persistence_GraphManagerHelper.js';
+import type { DSL_Persistence_PureProtocolProcessorPlugin_Extension } from '../../../../DSL_Persistence_PureProtocolProcessorPlugin_Extension.js';
 import {
   type PackageableElementImplicitReference,
   V1_buildFullPath,
@@ -61,7 +61,7 @@ export const V1_buildPersistencePlatform = (
   const extraPersistencePlatformBuilders = context.extensions.plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraPersistencePlatformBuilders?.() ?? [],
   );
   for (const builder of extraPersistencePlatformBuilders) {

package/src/graphManager/protocol/pure/v1/transformation/pureProtocol/{V1_DSLPersistenceContext_ProtocolHelper.ts → V1_DSL_PersistenceContext_ProtocolHelper.ts}

@@ -14,18 +14,18 @@
  * limitations under the License.
  */
 
-import { V1_PersistenceContext } from '../../model/packageableElements/persistence/V1_DSLPersistence_PersistenceContext.js';
+import { V1_PersistenceContext } from '../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceContext.js';
 import {
   V1_DefaultPersistencePlatform,
   type V1_PersistencePlatform,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_PersistencePlatform.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_PersistencePlatform.js';
 import {
   V1_ConnectionValue,
   V1_PrimitiveTypeValue,
   V1_ServiceParameter,
   type V1_ServiceParameterValue,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_ServiceParameter.js';
-import type { DSLPersistence_PureProtocolProcessorPlugin_Extension } from '../../../DSLPersistence_PureProtocolProcessorPlugin_Extension.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_ServiceParameter.js';
+import type { DSL_Persistence_PureProtocolProcessorPlugin_Extension } from '../../../DSL_Persistence_PureProtocolProcessorPlugin_Extension.js';
 import {
   type PureProtocolProcessorPlugin,
   V1_deserializeConnectionValue,
@@ -75,7 +75,7 @@ export const V1_serializePersistencePlatform = (
   const extraPersistencePlatformProtocolSerializers = plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraPersistencePlatformProtocolSerializers?.() ?? [],
   );
   for (const serializer of extraPersistencePlatformProtocolSerializers) {
@@ -101,7 +101,7 @@ export const V1_deserializePersistencePlatform = (
       const extraPersistencePlatformProtocolDeserializers = plugins.flatMap(
         (plugin) =>
           (
-            plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+            plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
           ).V1_getExtraPersistencePlatformProtocolDeserializers?.() ?? [],
       );
       for (const deserializer of extraPersistencePlatformProtocolDeserializers) {

package/src/graphManager/protocol/pure/v1/transformation/pureProtocol/{V1_DSLPersistence_ProtocolHelper.ts → V1_DSL_Persistence_ProtocolHelper.ts}

@@ -14,46 +14,46 @@
  * limitations under the License.
  */
 
-import { V1_Persistence } from '../../model/packageableElements/persistence/V1_DSLPersistence_Persistence.js';
+import { V1_Persistence } from '../../model/packageableElements/persistence/V1_DSL_Persistence_Persistence.js';
 import {
   V1_CronTrigger,
   V1_ManualTrigger,
   type V1_Trigger,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_Trigger.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_Trigger.js';
 import {
   V1_EmailNotifyee,
   V1_Notifier,
   type V1_Notifyee,
   V1_PagerDutyNotifyee,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_Notifier.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_Notifier.js';
 import {
   V1_ObjectStorageSink,
   V1_RelationalSink,
   type V1_Sink,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_Sink.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_Sink.js';
 import {
   V1_AnyVersionDeduplicationStrategy,
   type V1_DeduplicationStrategy,
   V1_DuplicateCountDeduplicationStrategy,
   V1_MaxVersionDeduplicationStrategy,
   V1_NoDeduplicationStrategy,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_DeduplicationStrategy.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_DeduplicationStrategy.js';
 import {
   type V1_Auditing,
   V1_DateTimeAuditing,
   V1_NoAuditing,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_Auditing.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_Auditing.js';
 import {
   V1_FlatTarget,
   V1_MultiFlatTarget,
   V1_MultiFlatTargetPart,
   type V1_TargetShape,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_TargetShape.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_TargetShape.js';
 import {
   V1_DeleteIndicatorMergeStrategy,
   type V1_MergeStrategy,
   V1_NoDeletesMergeStrategy,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_MergeStrategy.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_MergeStrategy.js';
 import {
   V1_BatchIdAndDateTimeTransactionMilestoning,
   V1_BatchIdTransactionMilestoning,
@@ -67,7 +67,7 @@ import {
   type V1_TransactionMilestoning,
   type V1_ValidityDerivation,
   type V1_ValidityMilestoning,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_Milestoning.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_Milestoning.js';
 import {
   V1_AppendOnly,
   V1_BitemporalDelta,
@@ -77,20 +77,32 @@ import {
   V1_NontemporalSnapshot,
   V1_UnitemporalDelta,
   V1_UnitemporalSnapshot,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_IngestMode.js';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_IngestMode.js';
 import {
   V1_BatchPersister,
   type V1_Persister,
   V1_StreamingPersister,
-} from '../../model/packageableElements/persistence/V1_DSLPersistence_Persister.js';
-import type { DSLPersistence_PureProtocolProcessorPlugin_Extension } from '../../../DSLPersistence_PureProtocolProcessorPlugin_Extension.js';
-import type { PureProtocolProcessorPlugin } from '@finos/legend-graph';
+} from '../../model/packageableElements/persistence/V1_DSL_Persistence_Persister.js';
+import type { DSL_Persistence_PureProtocolProcessorPlugin_Extension } from '../../../DSL_Persistence_PureProtocolProcessorPlugin_Extension.js';
+import {
+  type PureProtocolProcessorPlugin,
+  type V1_AtomicTest,
+  V1_serializeTestAssertion,
+  V1_deserializeTestAssertion,
+  V1_externalFormatDataModelSchema,
+  type V1_TestAssertion,
+  type V1_TestBatch,
+  V1_serializeAtomicTest,
+  V1_deserializeAtomicTest,
+} from '@finos/legend-graph';
 import {
   deserializeArray,
+  optionalCustom,
   type PlainObject,
   serializeArray,
   UnsupportedOperationError,
   usingConstantValueSchema,
+  usingModelSchema,
 } from '@finos/legend-shared';
 import {
   createModelSchema,
@@ -103,6 +115,10 @@ import {
   serialize,
   SKIP,
 } from 'serializr';
+import { V1_PersistenceTest } from '../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTest.js';
+import { V1_PersistenceTestBatch } from '../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestBatch.js';
+import { V1_ConnectionTestData } from '../../model/packageableElements/persistence/V1_DSL_Persistence_ConnectionTestData.js';
+import { V1_PersistenceTestData } from '../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestData.js';
 
 /**********
  * notifier
@@ -996,7 +1012,7 @@ export const V1_serializeTrigger = (
   const extraTriggerProtocolSerializers = plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraTriggerProtocolSerializers?.() ?? [],
   );
   for (const serializer of extraTriggerProtocolSerializers) {
@@ -1018,7 +1034,7 @@ export const V1_deserializeTrigger = (
   const extraTriggerProtocolDeserializers = plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraTriggerProtocolDeserializers?.() ?? [],
   );
   for (const deserializer of extraTriggerProtocolDeserializers) {
@@ -1032,6 +1048,92 @@ export const V1_deserializeTrigger = (
   );
 };
 
+/**********
+ * test
+ **********/
+
+export enum V1_AtomicTestType {
+  PERSISTENCE_TEST = 'test',
+}
+
+export const V1_persistenceConnectionTestDataModelSchema = createModelSchema(
+  V1_ConnectionTestData,
+  {
+    data: usingModelSchema(V1_externalFormatDataModelSchema),
+  },
+);
+
+export const V1_persistenceTestDataModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_PersistenceTestData> =>
+  createModelSchema(V1_PersistenceTestData, {
+    connection: usingModelSchema(V1_persistenceConnectionTestDataModelSchema),
+  });
+
+export const V1_persistenceTestBatchModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_PersistenceTestBatch> =>
+  createModelSchema(V1_PersistenceTestBatch, {
+    assertions: custom(
+      (values) =>
+        serializeArray(values, (value: V1_TestAssertion) =>
+          V1_serializeTestAssertion(value),
+        ),
+      (values) =>
+        deserializeArray(values, (v) => V1_deserializeTestAssertion(v), {
+          skipIfEmpty: false,
+        }),
+    ),
+    batchId: primitive(),
+    id: primitive(),
+    testData: usingModelSchema(V1_persistenceTestDataModelSchema(plugins)),
+  });
+
+export const V1_serializePersistenceTestBatch = (
+  protocol: V1_TestBatch,
+  plugins: PureProtocolProcessorPlugin[],
+): PlainObject<V1_TestBatch> => {
+  if (protocol instanceof V1_PersistenceTestBatch) {
+    return serialize(V1_persistenceTestBatchModelSchema(plugins), protocol);
+  }
+  throw new UnsupportedOperationError(`Can't serialize test batch `, protocol);
+};
+
+export const V1_deserializePersistenceTestBatch = (
+  json: PlainObject<V1_TestBatch>,
+  plugins: PureProtocolProcessorPlugin[],
+): V1_TestBatch =>
+  deserialize(V1_persistenceTestBatchModelSchema(plugins), json);
+
+export const V1_persistenceTestModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_PersistenceTest> =>
+  createModelSchema(V1_PersistenceTest, {
+    _type: usingConstantValueSchema(V1_AtomicTestType.PERSISTENCE_TEST),
+    id: primitive(),
+    isTestDataFromServiceOutput: primitive(),
+    testBatches: custom(
+      (values) =>
+        serializeArray(
+          values,
+          (value: V1_TestBatch) =>
+            V1_serializePersistenceTestBatch(value, plugins),
+          {
+            skipIfEmpty: true,
+            INTERNAL__forceReturnEmptyInTest: true,
+          },
+        ),
+      (values) =>
+        deserializeArray(
+          values,
+          (v) => V1_deserializePersistenceTestBatch(v, plugins),
+          {
+            skipIfEmpty: false,
+          },
+        ),
+    ),
+  });
+
 /**********
  * persistence
  **********/
@@ -1055,6 +1157,21 @@ export const V1_persistenceModelSchema = (
       (val) => V1_deserializePersister(val, plugins),
     ),
     service: primitive(),
+    tests: optionalCustom(
+      (values) =>
+        serializeArray(
+          values,
+          (value: V1_AtomicTest) => V1_serializeAtomicTest(value, plugins),
+          {
+            skipIfEmpty: true,
+            INTERNAL__forceReturnEmptyInTest: true,
+          },
+        ),
+      (values) =>
+        deserializeArray(values, (v) => V1_deserializeAtomicTest(v, plugins), {
+          skipIfEmpty: true,
+        }),
+    ),
     trigger: custom(
       (val) => V1_serializeTrigger(val, plugins),
       (val) => V1_deserializeTrigger(val, plugins),

package/src/index.ts

@@ -14,32 +14,33 @@
  * limitations under the License.
  */
 
-export * from './DSLPersistence_Extension.js';
-export { DSLPersistence_LegendStudioApplicationPlugin } from './components/studio/DSLPersistence_LegendStudioApplicationPlugin.js';
+export * from './DSL_Persistence_Extension.js';
+export { DSL_Persistence_LegendStudioApplicationPlugin } from './components/studio/DSL_Persistence_LegendStudioApplicationPlugin.js';
 
 // ---------- PACKAGEABLE ELEMENT ----------
 
 // metamodels
-export { Persistence } from './graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Persistence.js';
-export { PersistenceContext } from './graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_PersistenceContext.js';
-export { PersistencePlatform } from './graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_PersistencePlatform.js';
-export { Trigger } from './graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Trigger.js';
+export { Persistence } from './graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Persistence.js';
+export { PersistenceTest } from './graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceTest.js';
+export { PersistenceContext } from './graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceContext.js';
+export { PersistencePlatform } from './graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistencePlatform.js';
+export { Trigger } from './graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Trigger.js';
 
 // v1 protocols
-export { V1_Persistence } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSLPersistence_Persistence.js';
-export { V1_PersistenceContext } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSLPersistence_PersistenceContext.js';
-export { V1_PersistencePlatform } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSLPersistence_PersistencePlatform.js';
-export { V1_Trigger } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSLPersistence_Trigger.js';
+export { V1_Persistence } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSL_Persistence_Persistence.js';
+export { V1_PersistenceContext } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSL_Persistence_PersistenceContext.js';
+export { V1_PersistencePlatform } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSL_Persistence_PersistencePlatform.js';
+export { V1_Trigger } from './graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSL_Persistence_Trigger.js';
 
 // ---------- TRANSFORMATION ----------
 
 // extension
-export * from './graphManager/protocol/pure/DSLPersistence_PureProtocolProcessorPlugin_Extension.js';
+export * from './graphManager/protocol/pure/DSL_Persistence_PureProtocolProcessorPlugin_Extension.js';
 
 // builders
-export { V1_buildPersistence } from './graphManager/protocol/pure/v1/transformation/pureGraph/to/V1_PersistenceBuilder.js';
-export { V1_buildPersistenceContext } from './graphManager/protocol/pure/v1/transformation/pureGraph/to/V1_PersistenceContextBuilder.js';
+export { V1_buildPersistence } from './graphManager/protocol/pure/v1/transformation/pureGraph/to/V1_DSL_Persistence_BuilderHelper.js';
+export { V1_buildPersistenceContext } from './graphManager/protocol/pure/v1/transformation/pureGraph/to/V1_DSL_Persistence_ContextBuilder.js';
 
 // transformers
-export { V1_transformPersistence } from './graphManager/protocol/pure/v1/transformation/pureGraph/from/V1_PersistenceTransformer.js';
-export { V1_transformPersistenceContext } from './graphManager/protocol/pure/v1/transformation/pureGraph/from/V1_PersistenceContextTransformer.js';
+export { V1_transformPersistence } from './graphManager/protocol/pure/v1/transformation/pureGraph/from/V1_DSL_Persistence_TransformerHelper.js';
+export { V1_transformPersistenceContext } from './graphManager/protocol/pure/v1/transformation/pureGraph/from/V1_DSL_Persistence_ContextTransformer.js';