chore: added test cases

Author: sf-aastha-paruthi

messages/assess.json

@@ -224,5 +224,6 @@
   "omniStudioAllVersionsProcessingConsentNotGiven": "You've not consented to assess all versions of the Omnistudio components. We cannot proceed with the assessment.",
   "omniStudioAllVersionsProcessingConsentGiven": "You've consented to assess all versions of the Omnistudio components.",
   "preparingStorageForMetadataEnabledOrg": "Preparing storage for %s org on Standard Data Model with Metadata API enabled",
-  "processingNotRequired": "Assessment not required for Standard Data Model Orgs with Metadata API enabled"
+  "processingNotRequired": "Assessment not required for Standard Data Model Orgs with Metadata API enabled",
+  "skippingTruncation": "Skipping Trunction as standard data model org"
 }

messages/migrate.json

@@ -323,5 +323,6 @@
   "globalAutoNumberUnSupportedInOmnistudioPackage": "Feature not supported - Omni Global Auto Number is not supported in Omnistudio Foundation package orgs.",
   "orgUsecaseDetails": "The Org is on %s data model with %s and Omnistudio Metadata API %s",
   "preparingStorageForMetadataEnabledOrg": "Preparing storage for %s org on Standard Data Model with Metadata API enabled",
-  "processingNotRequired": "Migration not required for Standard Data Model Orgs with Metadata API enabled"
+  "processingNotRequired": "Migration not required for Standard Data Model Orgs with Metadata API enabled",
+  "skippingTruncation": "Skipping Trunction as standard data model org"
 }

src/commands/omnistudio/migration/migrate.ts

@@ -76,12 +76,7 @@ export default class Migrate extends OmniStudioBaseCommand {
   };
 
   // OmniStudio components that don't need migration logging when metadata API is enabled
-  private readonly OMNISTUDIO_COMPONENTS_FOR_LOGGING = [
-    Constants.Omniscript,
-    Constants.IntegrationProcedure,
-    Constants.Flexcard,
-    Constants.DataMapper,
-  ];
+  private readonly OMNISTUDIO_COMPONENTS_FOR_LOGGING = [Constants.CustomLabel, Constants.GlobalAutoNumber];
 
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   public async run(): Promise<any> {
@@ -370,10 +365,11 @@ export default class Migrate extends OmniStudioBaseCommand {
     // Truncate in reverse order (highest dependencies first) - this is correct for cleanup
     for (const cls of migrationObjects) {
       try {
-        Logger.log(messages.getMessage('cleaningComponent', [cls.getName()]));
-        debugTimer.lap('Cleaning: ' + cls.getName());
+        const componentName = cls.getName();
+
+        this.logTruncationStart(componentName, debugTimer);
         await cls.truncate();
-        Logger.log(messages.getMessage('cleaningDone', [cls.getName()]));
+        this.logTruncationComplete(componentName);
       } catch (ex: any) {
         objectMigrationResults.push({
           name: cls.getName(),
@@ -434,7 +430,7 @@ export default class Migrate extends OmniStudioBaseCommand {
    * Log the start of component migration and track timing
    */
   private logMigrationStart(componentName: string, debugTimer: DebugTimer): void {
-    const shouldSkipLogging = this.shouldSkipMigrationLogging(componentName);
+    const shouldSkipLogging = this.shouldSkipLogging(componentName);
 
     if (!shouldSkipLogging) {
       Logger.log(messages.getMessage('migratingComponent', [componentName]));
@@ -446,7 +442,7 @@ export default class Migrate extends OmniStudioBaseCommand {
    * Log the completion or failure of component migration
    */
   private logMigrationComplete(componentName: string, results: MigrationResult[]): void {
-    const shouldSkipLogging = this.shouldSkipMigrationLogging(componentName);
+    const shouldSkipLogging = this.shouldSkipLogging(componentName);
 
     if (!shouldSkipLogging) {
       if (results.some((result) => result?.errors?.length > 0) && !isStandardDataModelWithMetadataAPIEnabled()) {
@@ -461,17 +457,49 @@ export default class Migrate extends OmniStudioBaseCommand {
    * Check if migration logging should be skipped for a component
    * Returns true if Metadata API is enabled and the component doesn't need migration
    */
-  private shouldSkipMigrationLogging(componentName: string): boolean {
+  private shouldSkipTruncationLogging(componentName: string): boolean {
+    if (isStandardDataModel()) {
+      return true;
+    }
+
+    return this.shouldSkipLogging(componentName);
+  }
+
+  /**
+   * Check if migration logging should be skipped for a component
+   * Returns true if Metadata API is enabled and the component doesn't need migration
+   */
+  private shouldSkipLogging(componentName: string): boolean {
     if (!isStandardDataModelWithMetadataAPIEnabled()) {
       return false;
     }
 
     // Skip logging for OmniStudio components that don't need migration when metadata API is enabled
-    return this.OMNISTUDIO_COMPONENTS_FOR_LOGGING.some((component) =>
+    return !this.OMNISTUDIO_COMPONENTS_FOR_LOGGING.some((component) =>
       componentName.toLowerCase().includes(component.toLowerCase())
     );
   }
 
+  /**
+   * Log the start of component truncation and track timing
+   */
+  private logTruncationStart(componentName: string, debugTimer: DebugTimer): void {
+    const shouldSkipLogging = this.shouldSkipTruncationLogging(componentName);
+    if (!shouldSkipLogging) {
+      Logger.log(messages.getMessage('cleaningComponent', [componentName]));
+      debugTimer.lap('Cleaning: ' + componentName);
+    }
+  }
+  /**
+   * Log the completion of component truncation
+   */
+  private logTruncationComplete(componentName: string): void {
+    const shouldSkipLogging = this.shouldSkipTruncationLogging(componentName);
+
+    if (!shouldSkipLogging) {
+      Logger.log(messages.getMessage('cleaningDone', [componentName]));
+    }
+  }
   /**
    * Get migration objects in the correct dependency order:
    * 1. Data Mappers (lowest dependencies)

src/migration/dataraptor.ts

@@ -58,6 +58,7 @@ export class DataRaptorMigrationTool extends BaseMigrationTool implements Migrat
 
   async truncate(): Promise<void> {
     if (this.IS_STANDARD_DATA_MODEL) {
+      Logger.logVerbose(this.messages.getMessage('skippingTruncation'));
       return;
     }
     await super.truncate(DataRaptorMigrationTool.OMNIDATATRANSFORM_NAME);
@@ -295,6 +296,9 @@ export class DataRaptorMigrationTool extends BaseMigrationTool implements Migrat
   public async assess(): Promise<DataRaptorAssessmentInfo[]> {
     try {
       if (isStandardDataModelWithMetadataAPIEnabled()) {
+        Logger.log(
+          this.messages.getMessage('skippingAssessmentForStandardOrgWithMetadataAPIEnabled', [Constants.DataMapper])
+        );
         return [];
       }
       DebugTimer.getInstance().lap('Query data raptors');

src/migration/flexcard.ts

@@ -66,6 +66,7 @@ export class CardMigrationTool extends BaseMigrationTool implements MigrationToo
   // Perform Delete of OmniUiCard Records to start migration from scratch
   async truncate(): Promise<void> {
     if (this.IS_STANDARD_DATA_MODEL) {
+      Logger.logVerbose(this.messages.getMessage('skippingTruncation'));
       return;
     }
     const objectName = CardMigrationTool.OMNIUICARD_NAME;
@@ -450,15 +451,15 @@ export class CardMigrationTool extends BaseMigrationTool implements MigrationToo
   }
 
   private handleAssessmentForStdDataModelOrgsWithMetadataAPIEnabled(flexCards: AnyJson[]): FlexCardAssessmentInfo[] {
-    Logger.log(this.messages.getMessage('preparingStorageForMetadataEnabledOrg', [Constants.Flexcard]));
+    Logger.logVerbose(this.messages.getMessage('preparingStorageForMetadataEnabledOrg', [Constants.Flexcard]));
     let storage: MigrationStorage = StorageUtil.getOmnistudioAssessmentStorage();
     this.prepareStorageForRelatedObjectsWhenMetadataAPIEnabled(storage, flexCards);
     StorageUtil.printAssessmentStorage();
     return [];
   }
 
   private handleMigrationForStdDataModelOrgsWithMetadataAPIEnabled(flexcards: AnyJson[]): MigrationResult[] {
-    Logger.log(this.messages.getMessage('preparingStorageForMetadataEnabledOrg', [Constants.Flexcard]));
+    Logger.logVerbose(this.messages.getMessage('preparingStorageForMetadataEnabledOrg', [Constants.Flexcard]));
     let storage: MigrationStorage = StorageUtil.getOmnistudioMigrationStorage();
     this.prepareStorageForRelatedObjectsWhenMetadataAPIEnabled(storage, flexcards);
     StorageUtil.printAssessmentStorage();

src/migration/omniscript.ts

@@ -122,6 +122,7 @@ export class OmniScriptMigrationTool extends BaseMigrationTool implements Migrat
   async truncate(): Promise<void> {
     // Truncation is needed when we migrate from custom to standard data model, when on custom data model, no truncation is required
     if (this.IS_STANDARD_DATA_MODEL) {
+      Logger.logVerbose(this.messages.getMessage('skippingTruncation'));
       return;
     }
 
@@ -1415,7 +1416,7 @@ export class OmniScriptMigrationTool extends BaseMigrationTool implements Migrat
       return result;
     }
 
-    Logger.log(this.messages.getMessage('preparingStorageForMetadataEnabledOrg', [Constants.Omniscript]));
+    Logger.logVerbose(this.messages.getMessage('preparingStorageForMetadataEnabledOrg', [Constants.Omniscript]));
     let storage: MigrationStorage = StorageUtil.getOmnistudioMigrationStorage();
     Logger.logVerbose(this.messages.getMessage('updatingStorageForOmniscipt', ['Migration']));