Fix certificates

Author: cffls

integration-test/test/base.py

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

integration-test/test/test_certificate.py

@@ -42,14 +42,36 @@ def test_stake_delegation(self):
             )
             stake_registration = StakeRegistration(stake_credential)
             pool_hash = PoolKeyHash(bytes.fromhex(os.environ.get("POOL_ID").strip()))
-            stake_delegation = StakeDelegation(stake_credential, pool_keyhash=pool_hash)
+            # stake_delegation = StakeDelegation(stake_credential, pool_keyhash=pool_hash)
+
+            drep = DRep(
+                DRepKind.VERIFICATION_KEY_HASH,
+                self.stake_key_pair.verification_key.hash(),
+            )
+
+            drep_credential = DRepCredential(
+                self.stake_key_pair.verification_key.hash()
+            )
+
+            anchor = Anchor(
+                url="https://drep.com",
+                data_hash=AnchorDataHash(bytes.fromhex("0" * 64)),
+            )
+
+            drep_registration = RegDRepCert(
+                drep_credential=drep_credential, coin=500000000, anchor=anchor
+            )
+
+            all_in_one_cert = StakeRegistrationAndDelegationAndVoteDelegation(
+                stake_credential, pool_hash, drep, 1000000
+            )
 
             builder = TransactionBuilder(self.chain_context)
 
             builder.add_input_address(address)
             builder.add_output(TransactionOutput(address, 35000000))
 
-            builder.certificates = [stake_registration, stake_delegation]
+            builder.certificates = [drep_registration, all_in_one_cert]
 
             signed_tx = builder.build_and_sign(
                 [self.stake_key_pair.signing_key, self.payment_key_pair.signing_key],
@@ -73,9 +95,13 @@ def test_stake_delegation(self):
             network=self.NETWORK,
         )
 
-        rewards = self.chain_context.query_account_reward_summaries(keys=[stake_address.encode()])
+        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"]
+        stake_address_reward = rewards[stake_address.staking_part.payload.hex()][
+            "rewards"
+        ]["ada"]["lovelace"]
 
         builder.withdrawals = Withdrawals({bytes(stake_address): stake_address_reward})
 

pycardano/backend/ogmios_v6.py

@@ -307,9 +307,13 @@ 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]:
+    def query_account_reward_summaries(
+        self, scripts: Optional[List[str]] = None, keys: Optional[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)
+            summaries, _ = client.query_reward_account_summaries.execute(
+                scripts=scripts, keys=keys
+            )
             return summaries
 
     def submit_tx_cbor(self, cbor: Union[bytes, str]):