feat: update container types

packages/transaction-controller/CHANGELOG.md

@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- Support `containerTypes` property in `updateEditableParams` method ([#6014](https://github.com/MetaMask/core/pull/6014))
 - Add specific transaction types to outgoing transactions retrieved from accounts API ([#5987](https://github.com/MetaMask/core/pull/5987))
   - Add optional `amount` property to `transferInformation` object in `TransactionMeta` type.
 

packages/transaction-controller/src/TransactionController.test.ts

@@ -79,6 +79,7 @@ import {
   GasFeeEstimateType,
   SimulationErrorCode,
   SimulationTokenStandard,
+  TransactionContainerType,
   TransactionEnvelopeType,
   TransactionStatus,
   TransactionType,
@@ -7157,6 +7158,29 @@ describe('TransactionController', () => {
       );
     });
 
+    it('updates container types', async () => {
+      const { controller } = setupController({
+        options: {
+          state: {
+            transactions: [transactionMeta],
+          },
+        },
+        updateToInitialState: true,
+      });
+
+      const updatedTransaction = await controller.updateEditableParams(
+        transactionId,
+        {
+          ...params,
+          containerTypes: [TransactionContainerType.EnforcedSimulations],
+        },
+      );
+
+      expect(updatedTransaction?.containerTypes).toStrictEqual([
+        TransactionContainerType.EnforcedSimulations,
+      ]);
+    });
+
     it('throws an error if no transaction metadata is found', async () => {
       const { controller } = setupController();
       await expect(

packages/transaction-controller/src/TransactionController.ts

@@ -119,6 +119,7 @@ import type {
   TransactionBatchMeta,
   AfterSimulateHook,
   BeforeSignHook,
+  TransactionContainerType,
 } from './types';
 import {
   GasFeeEstimateLevel,
@@ -2027,6 +2028,7 @@ export class TransactionController extends BaseController<
    *
    * @param txId - The ID of the transaction to update.
    * @param params - The editable parameters to update.
+   * @param params.containerTypes - Container types applied to the parameters.
    * @param params.data - Data to pass with the transaction.
    * @param params.from - Address to send the transaction from.
    * @param params.gas - Maximum number of units of gas to use for the transaction.
@@ -2040,6 +2042,7 @@ export class TransactionController extends BaseController<
   async updateEditableParams(
     txId: string,
     {
+      containerTypes,
       data,
       from,
       gas,
@@ -2049,6 +2052,7 @@ export class TransactionController extends BaseController<
       to,
       value,
     }: {
+      containerTypes?: TransactionContainerType[];
       data?: string;
       from?: string;
       gas?: string;
@@ -2099,6 +2103,10 @@ export class TransactionController extends BaseController<
 
     updatedTransaction.type = type;
 
+    if (containerTypes) {
+      updatedTransaction.containerTypes = containerTypes;
+    }
+
     await updateTransactionLayer1GasFee({
       layer1GasFeeFlows: this.#layer1GasFeeFlows,
       messenger: this.messagingSystem,