Skip to content

Commit f515c01

Browse files
committed
Really disliking this framework FYI
1 parent 66a2ec7 commit f515c01

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

chia/_tests/cmds/wallet/test_wallet.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
CreateOfferForIDsResponse,
4848
DeleteUnconfirmedTransactions,
4949
FungibleAsset,
50+
GetCurrentDerivationIndexResponse,
5051
GetHeightInfoResponse,
5152
GetNextAddress,
5253
GetNextAddressResponse,
@@ -597,9 +598,9 @@ def test_get_derivation_index(capsys: object, get_test_cli_clients: tuple[TestRp
597598

598599
# set RPC Client
599600
class GetDerivationIndexRpcClient(TestWalletRpcClient):
600-
async def get_current_derivation_index(self) -> str:
601+
async def get_current_derivation_index(self) -> GetCurrentDerivationIndexResponse:
601602
self.add_to_log("get_current_derivation_index", ())
602-
return str(520)
603+
return GetCurrentDerivationIndexResponse(uint32(520))
603604

604605
inst_rpc_client = GetDerivationIndexRpcClient()
605606
test_rpc_clients.wallet_rpc_client = inst_rpc_client

0 commit comments

Comments
 (0)