Fix a few certificates (#423)

* Add account reward summary

* Fix certificates

* Increase epoch length

* cert test bugs fix

---------

Co-authored-by: tacmota <[email protected]>

Author: cffls

.gitignore

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

integration-test/configs/local-chang/config.json

@@ -99,8 +99,8 @@
   "TestAllegraHardForkAtEpoch": 0,
   "TestMaryHardForkAtEpoch": 0,
   "TestAlonzoHardForkAtEpoch": 0,
-  "TestBabbageHardForkAtEpoch": 10,
-  "TestConwayHardForkAtEpoch": 11,
+  "TestBabbageHardForkAtEpoch": 4,
+  "TestConwayHardForkAtEpoch": 5,
   "EnableDevelopmentHardForkEras": true,
   "ExperimentalProtocolsEnabled": true
 }

integration-test/configs/local-chang/shelley-genesis.json

@@ -46,5 +46,5 @@
         "pools": {},
         "stake": {}
     },
-    "epochLength": 4
+    "epochLength": 10
 }

integration-test/docker-compose-chang.yml

@@ -56,7 +56,7 @@ services:
         max-file: "10"
 
   ogmios:
-    image: cardanosolutions/ogmios:v6.11.0
+    image: cardanosolutions/ogmios:v6.11.2
     platform: linux/amd64
     environment:
       NETWORK: "${NETWORK:-local-alonzo}"

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],
@@ -62,7 +84,7 @@ def test_stake_delegation(self):
             print("############### Submitting transaction ###############")
             self.chain_context.submit_tx(signed_tx)
 
-        time.sleep(8)
+        time.sleep(120)
 
         builder = TransactionBuilder(self.chain_context)
 
@@ -73,7 +95,15 @@ 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,15 @@ 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: 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
+            )
+            return summaries
+
     def submit_tx_cbor(self, cbor: Union[bytes, str]):
         if isinstance(cbor, bytes):
             cbor = cbor.hex()