Missed a couple

Author: Quexington

chia/_tests/cmds/wallet/test_wallet.py

@@ -51,6 +51,8 @@
     CATSpend,
     CATSpendResponse,
     ClawbackPuzzleDecoratorOverride,
+    CreateNewWallet,
+    CreateNewWalletType,
     CreateOfferForIDs,
     CreateOfferForIDsResponse,
     DeleteUnconfirmedTransactions,
@@ -83,6 +85,7 @@
     TakeOffer,
     TakeOfferResponse,
     TransactionRecordWithMetadata,
+    WalletCreationMode,
     WalletInfoResponse,
 )
 from chia.wallet.wallet_spend_bundle import WalletSpendBundle
@@ -707,10 +710,6 @@ def test_add_token(capsys: object, get_test_cli_clients: tuple[TestRpcClients, P
 
     # set RPC Client
     class AddTokenRpcClient(TestWalletRpcClient):
-        async def create_wallet_for_existing_cat(self, asset_id: bytes) -> dict[str, int]:
-            self.add_to_log("create_wallet_for_existing_cat", (asset_id,))
-            return {"wallet_id": 3}
-
         async def set_cat_name(self, request: CATSetName) -> CATSetNameResponse:
             self.add_to_log("set_cat_name", (request.wallet_id, request.name))
             return CATSetNameResponse(wallet_id=request.wallet_id)
@@ -728,7 +727,19 @@ async def set_cat_name(self, request: CATSetName) -> CATSetNameResponse:
 
     expected_calls: logType = {
         "cat_asset_id_to_name": [(get_bytes32(1),), (bytes32([1, 2] * 16),)],
-        "create_wallet_for_existing_cat": [(bytes32([1, 2] * 16),)],
+        "create_new_wallet": [
+            (
+                CreateNewWallet(
+                    wallet_type=CreateNewWalletType.CAT_WALLET,
+                    mode=WalletCreationMode.EXISTING,
+                    asset_id=bytes32([1, 2] * 16),
+                    push=True,
+                ),
+                DEFAULT_TX_CONFIG,
+                tuple(),
+                ConditionValidTimes(),
+            )
+        ],
         "set_cat_name": [(2, "examplecat"), (3, "examplecat")],
     }
     test_rpc_clients.wallet_rpc_client.check_log(expected_calls)

chia/_tests/pools/test_pool_cmdline.py

@@ -992,9 +992,7 @@ async def test_plotnft_cli_misc(mocker: MockerFixture, consensus_mode: Consensus
         )
 
     # Test fall-through raise in create
-    mocker.patch.object(
-        test_rpc_client, "create_new_pool_wallet", create=True, side_effect=ValueError("Injected error")
-    )
+    mocker.patch.object(test_rpc_client, "create_new_wallet", create=True, side_effect=ValueError("Injected error"))
     with pytest.raises(CliRpcConnectionError, match="Error creating plot NFT: Injected error"):
         await create(
             wallet_info=WalletClientInfo(client=cast(WalletRpcClient, test_rpc_client), fingerprint=0, config=dict()),