Fix for custom_int's in feeder gateway

starknet_devnet/blueprints/feeder_gateway.py

@@ -40,6 +40,18 @@ def validate_request(data: bytes, cls):
  ) from err
 
 
+def validate_int(request_args: MultiDict, attribute: str):
+ """Validate if attribute is int."""
+ try:
+ return int(request_args.get(attribute))
+ except (ValueError) as err:
+ raise StarknetDevnetException(
+ code=StarkErrorCode.MALFORMED_REQUEST,
+ message=str(err),
+ status_code=400,
+ ) from err
+
+
 def _check_block_hash(request_args: MultiDict):
  block_hash = request_args.get("blockHash", type=custom_int)
  if block_hash is not None:
@@ -186,7 +198,7 @@ async def get_storage_at():
  _check_block_hash(request.args)
 
  contract_address = request.args.get("contractAddress", type=custom_int)
- key = request.args.get("key", type=custom_int)
+ key = validate_int(request.args, "key")
 
  storage = await state.starknet_wrapper.get_storage_at(contract_address, key)
  return jsonify(storage)

test/rpc/test_rpc_transactions.py

@@ -382,7 +382,7 @@ def test_add_invoke_transaction():
  storage = gateway_call(
  "get_storage_at",
  contractAddress=contract_address,
- key=hex(get_storage_var_address("balance")),
+ key=get_storage_var_address("balance"),
  )
 
  assert set(receipt.keys()) == {"transaction_hash"}
@@ -419,7 +419,7 @@ def test_add_invoke_transaction_v0():
  storage = gateway_call(
  "get_storage_at",
  contractAddress=contract_address,
- key=hex(get_storage_var_address("balance")),
+ key=get_storage_var_address("balance"),
  )
 
  assert set(receipt.keys()) == {"transaction_hash"}