@finos/legend-extension-dsl-persistence 3.0.14 → 4.0.0

package/src/graphManager/protocol/pure/v1/model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestData.ts

@@ -0,0 +1,31 @@
+/**
+ * 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 {} from '@finos/legend-graph';
+import { type Hashable, hashArray } from '@finos/legend-shared';
+import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSL_Persistence_HashUtils.js';
+import type { V1_ConnectionTestData } from './V1_DSL_Persistence_ConnectionTestData.js';
+
+export class V1_PersistenceTestData implements Hashable {
+  connection!: V1_ConnectionTestData;
+
+  get hashCode(): string {
+    return hashArray([
+      PERSISTENCE_HASH_STRUCTURE.PERSISTENCE_TEST_DATA,
+      this.connection,
+    ]);
+  }
+}

package/src/graphManager/protocol/pure/v1/model/packageableElements/persistence/{V1_DSLPersistence_Persister.ts → V1_DSL_Persistence_Persister.ts}

@@ -14,10 +14,10 @@
  * limitations under the License.
  */
 
-import type { V1_IngestMode } from './V1_DSLPersistence_IngestMode.js';
-import type { V1_Sink } from './V1_DSLPersistence_Sink.js';
-import type { V1_TargetShape } from './V1_DSLPersistence_TargetShape.js';
-import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSLPersistence_HashUtils.js';
+import type { V1_IngestMode } from './V1_DSL_Persistence_IngestMode.js';
+import type { V1_Sink } from './V1_DSL_Persistence_Sink.js';
+import type { V1_TargetShape } from './V1_DSL_Persistence_TargetShape.js';
+import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSL_Persistence_HashUtils.js';
 import { type Hashable, hashArray } from '@finos/legend-shared';
 
 export abstract class V1_Persister implements Hashable {

package/src/graphManager/protocol/pure/v1/model/packageableElements/persistence/{V1_DSLPersistence_ServiceParameter.ts → V1_DSL_Persistence_ServiceParameter.ts}

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSLPersistence_HashUtils.js';
+import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSL_Persistence_HashUtils.js';
 import type { V1_Connection, V1_ValueSpecification } from '@finos/legend-graph';
 import { type Hashable, hashArray } from '@finos/legend-shared';
 

package/src/graphManager/protocol/pure/v1/model/packageableElements/persistence/{V1_DSLPersistence_Sink.ts → V1_DSL_Persistence_Sink.ts}

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSLPersistence_HashUtils.js';
+import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSL_Persistence_HashUtils.js';
 import { type Hashable, hashArray } from '@finos/legend-shared';
 
 export abstract class V1_Sink implements Hashable {

package/src/graphManager/protocol/pure/v1/model/packageableElements/persistence/{V1_DSLPersistence_TargetShape.ts → V1_DSL_Persistence_TargetShape.ts}

@@ -14,8 +14,8 @@
  * limitations under the License.
  */
 
-import type { V1_DeduplicationStrategy } from './V1_DSLPersistence_DeduplicationStrategy.js';
-import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSLPersistence_HashUtils.js';
+import type { V1_DeduplicationStrategy } from './V1_DSL_Persistence_DeduplicationStrategy.js';
+import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSL_Persistence_HashUtils.js';
 import { type Hashable, hashArray } from '@finos/legend-shared';
 
 export abstract class V1_TargetShape implements Hashable {

package/src/graphManager/protocol/pure/v1/model/packageableElements/persistence/{V1_DSLPersistence_Trigger.ts → V1_DSL_Persistence_Trigger.ts}

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSLPersistence_HashUtils.js';
+import { PERSISTENCE_HASH_STRUCTURE } from '../../../../../../../graph/DSL_Persistence_HashUtils.js';
 import { type Hashable, hashArray } from '@finos/legend-shared';
 
 export abstract class V1_Trigger implements Hashable {

package/src/graphManager/protocol/pure/v1/transformation/pureGraph/{V1_DSLPersistence_GraphBuilderHelper.ts → V1_DSL_Persistence_GraphBuilderHelper.ts}

@@ -14,12 +14,12 @@
  * limitations under the License.
  */
 
-import type { Persistence } from '../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Persistence.js';
-import type { PersistenceContext } from '../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_PersistenceContext.js';
+import type { Persistence } from '../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Persistence.js';
+import type { PersistenceContext } from '../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceContext.js';
 import {
   getPersistence,
   getPersistenceContext,
-} from '../../../../../../graphManager/DSLPersistence_GraphManagerHelper.js';
+} from '../../../../../DSL_Persistence_GraphManagerHelper.js';
 import type {
   PackageableElementImplicitReference,
   V1_GraphBuilderContext,

package/src/graphManager/protocol/pure/v1/transformation/pureGraph/from/{V1_PersistenceContextTransformer.ts → V1_DSL_Persistence_ContextTransformer.ts}

@@ -14,29 +14,29 @@
  * limitations under the License.
  */
 
-import type { PersistenceContext } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_PersistenceContext.js';
+import type { PersistenceContext } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceContext.js';
 import {
   type PersistencePlatform,
   DefaultPersistencePlatform,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_PersistencePlatform.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistencePlatform.js';
 import {
   ConnectionValue,
   PrimitiveTypeValue,
   type ServiceParameter,
   type ServiceParameterValue,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_ServiceParameter.js';
-import { V1_PersistenceContext } from '../../../model/packageableElements/persistence/V1_DSLPersistence_PersistenceContext.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_ServiceParameter.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 V1_GraphTransformerContext,
   V1_initPackageableElement,
@@ -59,7 +59,7 @@ export const V1_transformPersistencePlatform = (
   const extraPersistencePlatformTransformers = context.plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraPersistencePlatformTransformers?.() ?? [],
   );
   for (const transformer of extraPersistencePlatformTransformers) {

package/src/graphManager/protocol/pure/v1/transformation/pureGraph/from/{V1_PersistenceTransformer.ts → V1_DSL_Persistence_TransformerHelper.ts}

@@ -18,14 +18,14 @@ 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,
@@ -35,12 +35,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,
@@ -54,44 +54,44 @@ 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,
   type Notifier,
   type Notifyee,
   PagerDutyNotifyee,
-} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Notifier.js';
-import type { Persistence } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSLPersistence_Persistence.js';
+} from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Notifier.js';
+import type { Persistence } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_Persistence.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,
   type 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';
+} 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 {
   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,
@@ -101,12 +101,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,
@@ -120,38 +120,49 @@ 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,
   V1_Notifier,
   type V1_Notifyee,
   V1_PagerDutyNotifyee,
-} from '../../../model/packageableElements/persistence/V1_DSLPersistence_Notifier.js';
-import { V1_Persistence } from '../../../model/packageableElements/persistence/V1_DSLPersistence_Persistence.js';
+} from '../../../model/packageableElements/persistence/V1_DSL_Persistence_Notifier.js';
+import { 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,
   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';
-import type { DSLPersistence_PureProtocolProcessorPlugin_Extension } from '../../../../DSLPersistence_PureProtocolProcessorPlugin_Extension.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 { DSL_Persistence_PureProtocolProcessorPlugin_Extension } from '../../../../DSL_Persistence_PureProtocolProcessorPlugin_Extension.js';
 import {
+  type TestAssertion,
   type V1_GraphTransformerContext,
   V1_initPackageableElement,
+  V1_transformEmbeddedData,
+  V1_transformTestAssertion,
+  V1_transformAtomicTest,
 } from '@finos/legend-graph';
-import { UnsupportedOperationError } from '@finos/legend-shared';
+import { guaranteeType, UnsupportedOperationError } from '@finos/legend-shared';
+import type { PersistenceTestBatch } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceTestBatch.js';
+import { V1_PersistenceTestBatch } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestBatch.js';
+import type { PersistenceTestData } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_PersistenceTestData.js';
+import { V1_PersistenceTestData } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTestData.js';
+import type { ConnectionTestData } from '../../../../../../../graph/metamodel/pure/model/packageableElements/persistence/DSL_Persistence_ConnectionTestData.js';
+import { V1_ConnectionTestData } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_ConnectionTestData.js';
+import { V1_PersistenceTest } from '../../../model/packageableElements/persistence/V1_DSL_Persistence_PersistenceTest.js';
 
 /**********
  * trigger
@@ -164,7 +175,7 @@ export const V1_transformTrigger = (
   const extraTriggerTransformers = context.plugins.flatMap(
     (plugin) =>
       (
-        plugin as DSLPersistence_PureProtocolProcessorPlugin_Extension
+        plugin as DSL_Persistence_PureProtocolProcessorPlugin_Extension
       ).V1_getExtraTriggerTransformers?.() ?? [],
   );
   for (const transformer of extraTriggerTransformers) {
@@ -608,6 +619,47 @@ export const V1_transformNotifier = (
   return notifier;
 };
 
+/**********
+ * test
+ **********/
+
+const V1_transformConnectionTestData = (
+  element: ConnectionTestData,
+  context: V1_GraphTransformerContext,
+): V1_ConnectionTestData => {
+  const connectionTestData = new V1_ConnectionTestData();
+  connectionTestData.data = V1_transformEmbeddedData(element.data, context);
+  return connectionTestData;
+};
+
+const V1_transformTestData = (
+  element: PersistenceTestData,
+  context: V1_GraphTransformerContext,
+): V1_PersistenceTestData => {
+  const testData = new V1_PersistenceTestData();
+  testData.connection = V1_transformConnectionTestData(
+    element.connection,
+    context,
+  );
+  return testData;
+};
+
+export const V1_transformPersistenceTestBatch = (
+  element: PersistenceTestBatch,
+  context: V1_GraphTransformerContext,
+): V1_PersistenceTestBatch => {
+  const persistenceTestBatch = new V1_PersistenceTestBatch();
+  persistenceTestBatch.id = element.id;
+  persistenceTestBatch.testData = V1_transformTestData(
+    element.testData,
+    context,
+  );
+  persistenceTestBatch.assertions = element.assertions.map(
+    (assertion: TestAssertion) => V1_transformTestAssertion(assertion),
+  );
+  return persistenceTestBatch;
+};
+
 /**********
  * persistence
  **********/
@@ -623,5 +675,8 @@ export const V1_transformPersistence = (
   protocol.service = element.service.valueForSerialization ?? '';
   protocol.persister = V1_transformPersister(element.persister, context);
   protocol.notifier = V1_transformNotifier(element.notifier, context);
+  protocol.tests = element.tests.map((test) =>
+    guaranteeType(V1_transformAtomicTest(test, context), V1_PersistenceTest),
+  );
   return protocol;
 };

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));
 };