diff --git a/tests/pallet_did_test.py b/tests/pallet_did_test.py index 5f1c09a..f8e2c2a 100644 --- a/tests/pallet_did_test.py +++ b/tests/pallet_did_test.py @@ -7,6 +7,17 @@ from tools.constants import WS_URL from peaq.utils import ExtrinsicBatch from peaq.did import did_add_payload, did_update_payload, did_remove_payload, did_rpc_read +from tools.runtime_upgrade import wait_until_block_height +from tools.constants import PARACHAIN_WS_URL, RELAYCHAIN_WS_URL +from peaq.utils import get_chain +from tools.utils import get_modified_chain_spec + + +DID_MIN_DEPOSIT = { + 'peaq-dev': 0.1 * 10 ** 18, + 'krest-network': 0.1 * 10 ** 18, + 'peaq-network': 0.005 * 10 ** 18, +} @pytest.mark.substrate @@ -14,6 +25,10 @@ class TestPalletDid(unittest.TestCase): def setUp(self): self.substrate = SubstrateInterface(url=WS_URL) self.kp_src = Keypair.create_from_uri('//Alice') + wait_until_block_height(SubstrateInterface(url=RELAYCHAIN_WS_URL), 1) + wait_until_block_height(SubstrateInterface(url=PARACHAIN_WS_URL), 1) + self.chain_spec = get_chain(self.substrate) + self.chain_spec = get_modified_chain_spec(self.chain_spec) def test_did_add(self): reserved_before = get_balance_reserve_value(self.substrate, self.kp_src.ss58_address, 'peaq_did') @@ -32,6 +47,7 @@ def test_did_add(self): self.assertEqual(data['value'], value) reserved_after = get_balance_reserve_value(self.substrate, self.kp_src.ss58_address, 'peaq_did') self.assertGreater(reserved_after, reserved_before) + self.assertGreaterEqual(reserved_after - reserved_before, DID_MIN_DEPOSIT[self.chain_spec]) def test_did_update(self): name = int(time.time()) diff --git a/tests/pallet_rbac_test.py b/tests/pallet_rbac_test.py index 30417b9..b31d4cc 100644 --- a/tests/pallet_rbac_test.py +++ b/tests/pallet_rbac_test.py @@ -19,9 +19,20 @@ from peaq.rbac import rbac_rpc_fetch_groups from tools.utils import get_balance_reserve_value from tools.constants import KP_GLOBAL_SUDO +from tools.runtime_upgrade import wait_until_block_height +from tools.constants import PARACHAIN_WS_URL, RELAYCHAIN_WS_URL +from peaq.utils import get_chain +from tools.utils import get_modified_chain_spec import unittest +RBAC_MIN_DEPOSIT = { + 'peaq-dev': 0.1 * 10 ** 18, + 'krest-network': 0.1 * 10 ** 18, + 'peaq-network': 0.005 * 10 ** 18, +} + + KP_TEST = Keypair.create_from_mnemonic(Keypair.generate_mnemonic()) RANDOM_PREFIX = KP_TEST.public_key.hex()[2:26] @@ -437,6 +448,10 @@ def verify_rpc_fail_disabled_id(self, kp_src): def setUp(self): self.substrate = SubstrateInterface(url=WS_URL) + wait_until_block_height(SubstrateInterface(url=RELAYCHAIN_WS_URL), 1) + wait_until_block_height(SubstrateInterface(url=PARACHAIN_WS_URL), 1) + self.chain_spec = get_chain(self.substrate) + self.chain_spec = get_modified_chain_spec(self.chain_spec) def test_pallet_rbac(self): print('---- pallet_rbac_test!! ----') @@ -475,6 +490,21 @@ def test_pallet_rbac(self): print(f'🔥 Test/{func}, Failed') raise + def test_reserve_check(self): + ROLE_TEST = '{0}03456789abcdef0123456111abcde00123456111'.format(RANDOM_PREFIX) + + kp_src = Keypair.create_from_mnemonic(Keypair.generate_mnemonic()) + fund(self.substrate, KP_GLOBAL_SUDO, kp_src, 1000 * 10 ** 18) + + reserved_before = get_balance_reserve_value(self.substrate, kp_src.ss58_address, 'peaqrbac') + batch = ExtrinsicBatch(self.substrate, kp_src) + rbac_add_role_payload(batch, f'0x{ROLE_TEST}', ROLE_NM1) + receipt = batch.execute() + self.assertTrue(receipt.is_success, f'Extrinsic-call-stack failed: {receipt.error_message}') + reserved_after = get_balance_reserve_value(self.substrate, kp_src.ss58_address, 'peaqrbac') + self.assertGreater(reserved_after, reserved_before) + self.assertGreaterEqual(reserved_after - reserved_before, RBAC_MIN_DEPOSIT[self.chain_spec]) + def test_delete_types(self): ROLE_TEST = '{0}03456789abcdef0123456111abcdef0123456111'.format(RANDOM_PREFIX) GROUP_TEST = '{0}0defabcdabcdefabc111abcdabcdefabcdefa111'.format(RANDOM_PREFIX) diff --git a/tests/pallet_storage_test.py b/tests/pallet_storage_test.py index fa497f8..27b34e1 100644 --- a/tests/pallet_storage_test.py +++ b/tests/pallet_storage_test.py @@ -5,10 +5,21 @@ from substrateinterface import SubstrateInterface, Keypair from peaq.utils import ExtrinsicBatch from peaq.storage import storage_add_payload, storage_update_payload, storage_rpc_read +from tools.utils import get_modified_chain_spec +from tools.runtime_upgrade import wait_until_block_height +from tools.constants import PARACHAIN_WS_URL, RELAYCHAIN_WS_URL +from peaq.utils import get_chain import unittest +STORAGE_MIN_DEPOSIT = { + 'peaq-dev': 0.1 * 10 ** 18, + 'krest-network': 0.1 * 10 ** 18, + 'peaq-network': 0.005 * 10 ** 18, +} + + def storage_remove_payload(batch, item_type): batch.compose_call( 'PeaqStorage', @@ -23,6 +34,10 @@ class TestPalletStorage(unittest.TestCase): def setUp(self): self._substrate = SubstrateInterface(url=WS_URL) self._kp_src = Keypair.create_from_uri('//Alice') + wait_until_block_height(SubstrateInterface(url=RELAYCHAIN_WS_URL), 1) + wait_until_block_height(SubstrateInterface(url=PARACHAIN_WS_URL), 1) + self._chain_spec = get_chain(self._substrate) + self._chain_spec = get_modified_chain_spec(self._chain_spec) def test_storage(self): reserved_before = get_balance_reserve_value(self._substrate, self._kp_src.ss58_address, 'peaqstor') @@ -40,6 +55,7 @@ def test_storage(self): item) reserved_after = get_balance_reserve_value(self._substrate, self._kp_src.ss58_address, 'peaqstor') self.assertGreater(reserved_after, reserved_before) + self.assertGreaterEqual(reserved_after - reserved_before, STORAGE_MIN_DEPOSIT[self._chain_spec]) def test_storage_update(self): batch = ExtrinsicBatch(self._substrate, self._kp_src)