Add account reward summary

Author: cffls

.gitignore

@@ -11,3 +11,4 @@ coverage.xml
 .code
 /integration-test/.env
 /integration-test/tmp_configs/*
+/integration-test/.coverage*

integration-test/test/base.py

@@ -8,7 +8,7 @@
 
 from pycardano import *
 
-TEST_RETRIES = 8
+TEST_RETRIES = 1
 
 
 @retry(tries=10, delay=4)

integration-test/test/test_certificate.py

@@ -73,7 +73,11 @@ def test_stake_delegation(self):
             network=self.NETWORK,
         )
 
-        builder.withdrawals = Withdrawals({bytes(stake_address): 0})
+        rewards = self.chain_context.query_account_reward_summaries(keys=[stake_address.encode()])
+
+        stake_address_reward = rewards[stake_address.staking_part.payload.hex()]["rewards"]["ada"]["lovelace"]
+
+        builder.withdrawals = Withdrawals({bytes(stake_address): stake_address_reward})
 
         builder.add_output(TransactionOutput(address, 1000000))
 

pycardano/backend/ogmios_v6.py

@@ -307,6 +307,11 @@ def utxo_by_tx_id(self, tx_id: str, index: int) -> Optional[UTxO]:
             return self._utxo_from_ogmios_result(utxos[0])
         return None
 
+    def query_account_reward_summaries(self, scripts: List[str]=None, keys: List[str]=None) -> List[dict]:
+        with OgmiosClient(self.host, self.port, self.path, self.secure) as client:
+            summaries, _ = client.query_reward_account_summaries.execute(scripts=scripts, keys=keys)
+            return summaries
+
     def submit_tx_cbor(self, cbor: Union[bytes, str]):
         if isinstance(cbor, bytes):
             cbor = cbor.hex()