Fixed policy_assets

Added more tests

Author: mathiasfrohlich

blockfrost/api/__init__.py

@@ -67,7 +67,7 @@ def root(self):
         asset_history, \
         asset_transactions, \
         asset_addresses, \
-        policy_assets
+        assets_policy
     from .cardano.blocks import \
         block_latest, \
         block_latest_transactions, \

blockfrost/api/cardano/addresses.py

@@ -14,12 +14,14 @@ class Amount:
     amount: [Amount]
     stake_address: str
     type: str
+    script: bool
 
-    def __init__(self, address: str, amount: [Amount], stake_address: str, type: str) -> None:
+    def __init__(self, address: str, amount: [Amount], stake_address: str, type: str, script: bool) -> None:
         self.address = address
         self.amount = [self.Amount(**o) for o in amount]
         self.stake_address = stake_address
         self.type = type
+        self.script = script
 
 
 @object_request_wrapper(AddressResponse)
@@ -92,12 +94,14 @@ class Amount:
     output_index: int
     amount: [Amount]
     block: str
+    data_hash: str
 
-    def __init__(self, tx_hash: str, output_index: int, amount: [Amount], block: str) -> None:
+    def __init__(self, tx_hash: str, output_index: int, amount: [Amount], block: str, data_hash: str) -> None:
         self.tx_hash = tx_hash
         self.output_index = output_index
         self.amount = [self.Amount(**o) for o in amount]
         self.block = block
+        self.data_hash = data_hash
 
 
 @object_list_request_wrapper(AddressesUTXOSResponse)

blockfrost/api/cardano/assets.py

@@ -210,13 +210,13 @@ def asset_addresses(self, asset: str, **kwargs):
 
 
 @dataclass
-class PolicyAssetResponse:
-    address: str
+class AssetPolicyResponse:
+    asset: str
     quantity: str
 
 
-@object_request_wrapper(PolicyAssetResponse)
-def policy_assets(self, policy_id: str, **kwargs):
+@object_list_request_wrapper(AssetPolicyResponse)
+def assets_policy(self, policy_id: str, **kwargs):
     """
     List of asset minted under a specific policy
 
@@ -238,7 +238,7 @@ def policy_assets(self, policy_id: str, **kwargs):
     :raises Exception: If the API response is somehow malformed.
     """
     return requests.get(
-        url=f"{self.url}/assets/{policy_id}",
+        url=f"{self.url}/assets/policy/{policy_id}",
         params=self.query_parameters(kwargs),
         headers=self.default_headers
     )

setup.py

@@ -16,7 +16,7 @@
 
 setup(
     name='blockfrost-python',
-    version='0.1.1',
+    version='0.1.2',
     description='The official Python SDK for Blockfrost API v0.1.26',
     long_description=long_description,
     long_description_content_type='text/markdown',

tests/test_cardano_accounts.py

@@ -1,9 +1,15 @@
 from blockfrost import BlockFrostApi, ApiError
 from blockfrost.api.cardano.accounts import \
     AccountResponse, \
-    AccountRewardResponse
+    AccountRewardResponse, \
+    AccountHistoryResponse, \
+    AccountDelegationResponse, \
+    AccountRegistrationResponse, \
+    AccountWithdrawalResponse, \
+    AccountMIRSResponse, \
+    AccountAddressResponse, \
+    AccountAddressesAssetResponse
 
-address = 'addr1q9ucyt3s5f4naz3n7fwpnt3a0t75kl3rxdvpe63e2gdes7dzs4s26v4800nwg8jygvrdqh6xhsphct0d4zqsnd3sagxq6kwrts'
 stake_address = 'stake1ux3g2c9dx2nhhehyrezyxpkstartcqmu9hk63qgfkccw5rqttygt7'
 
 
@@ -23,7 +29,7 @@ def test_accounts(requests_mock):
     }
     requests_mock.get(f"{api.url}/accounts/{stake_address}", json=mock_data)
     mock_object = AccountResponse(**mock_data)
-    assert api.accounts(stake_address=stake_address).stake_address == mock_object.stake_address
+    assert api.accounts(stake_address=stake_address) == mock_object
 
 
 def test_account_rewards(requests_mock):
@@ -52,68 +58,133 @@ def test_account_rewards(requests_mock):
     ]
     requests_mock.get(f"{api.url}/accounts/{stake_address}/rewards", json=mock_data)
     mock_object = [AccountRewardResponse(**data) for data in mock_data]
-    assert api.account_rewards(stake_address=stake_address)[0].pool_id == mock_object[0].pool_id
+    assert api.account_rewards(stake_address=stake_address) == mock_object
 
 
-def test_account_history():
-    assert True
-
-
-def test_account_delegations():
-    assert True
-
-
-def test_account_registrations():
-    assert True
-
-
-def test_account_withdrawals():
-    assert True
-
-
-def test_account_mirs():
-    assert True
-
-
-def test_account_addresses():
-    assert True
-
-
-def test_account_addresses_assets():
-    assert True
-
-
-def test_account():
-    assert True
-
-
-def test_account_reward():
-    assert True
-
-
-def test_account_history():
-    assert True
+def test_account_history(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "active_epoch": 210,
+            "amount": "12695385",
+            "pool_id": "pool1pu5jlj4q9w9jlxeu370a3c9myx47md5j5m2str0naunn2q3lkdy"
+        },
+        {
+            "active_epoch": 211,
+            "amount": "22695385",
+            "pool_id": "pool1pu5jlj4q9w9jlxeu370a3c9myx47md5j5m2str0naunn2q3lkdy"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/history", json=mock_data)
+    mock_object = [AccountHistoryResponse(**data) for data in mock_data]
+    assert api.account_history(stake_address=stake_address) == mock_object
 
 
-def test_account_delegation():
-    assert True
+def test_account_delegations(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "active_epoch": 210,
+            "tx_hash": "2dd15e0ef6e6a17841cb9541c27724072ce4d4b79b91e58432fbaa32d9572531",
+            "amount": "12695385",
+            "pool_id": "pool1pu5jlj4q9w9jlxeu370a3c9myx47md5j5m2str0naunn2q3lkdy"
+        },
+        {
+            "active_epoch": 242,
+            "tx_hash": "1a0570af966fb355a7160e4f82d5a80b8681b7955f5d44bec0dde628516157f0",
+            "amount": "12691385",
+            "pool_id": "pool1kchver88u3kygsak8wgll7htr8uxn5v35lfrsyy842nkscrzyvj"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/delegations", json=mock_data)
+    mock_object = [AccountDelegationResponse(**data) for data in mock_data]
+    assert api.account_delegations(stake_address=stake_address) == mock_object
 
 
-def test_account_registration():
-    assert True
+def test_account_registrations(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "tx_hash": "2dd15e0ef6e6a17841cb9541c27724072ce4d4b79b91e58432fbaa32d9572531",
+            "action": "registered"
+        },
+        {
+            "tx_hash": "1a0570af966fb355a7160e4f82d5a80b8681b7955f5d44bec0dde628516157f0",
+            "action": "deregistered"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/registrations", json=mock_data)
+    mock_object = [AccountRegistrationResponse(**data) for data in mock_data]
+    assert api.account_registrations(stake_address=stake_address) == mock_object
 
 
-def test_account_withdrawals():
-    assert True
+def test_account_withdrawals(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "tx_hash": "48a9625c841eea0dd2bb6cf551eabe6523b7290c9ce34be74eedef2dd8f7ecc5",
+            "amount": "454541212442"
+        },
+        {
+            "tx_hash": "4230b0cbccf6f449f0847d8ad1d634a7a49df60d8c142bb8cc2dbc8ca03d9e34",
+            "amount": "97846969"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/withdrawals", json=mock_data)
+    mock_object = [AccountWithdrawalResponse(**data) for data in mock_data]
+    assert api.account_withdrawals(stake_address=stake_address) == mock_object
 
 
-def test_account_mirs():
-    assert True
+def test_account_mirs(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "tx_hash": "69705bba1d687a816ff5a04ec0c358a1f1ef075ab7f9c6cc2763e792581cec6d",
+            "amount": "2193707473"
+        },
+        {
+            "tx_hash": "baaa77b63d4d7d2bb3ab02c9b85978c2092c336dede7f59e31ad65452d510c13",
+            "amount": "14520198574"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/mirs", json=mock_data)
+    mock_object = [AccountMIRSResponse(**data) for data in mock_data]
+    assert api.account_mirs(stake_address=stake_address) == mock_object
 
 
-def test_account_address():
-    assert True
+def test_account_addresses(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "address": "addr1qx2kd28nq8ac5prwg32hhvudlwggpgfp8utlyqxu6wqgz62f79qsdmm5dsknt9ecr5w468r9ey0fxwkdrwh08ly3tu9sy0f4qd"
+        },
+        {
+            "address": "addr1qys3czp8s9thc6u2fqed9yq3h24nyw28uk0m6mkgn9dkckjf79qsdmm5dsknt9ecr5w468r9ey0fxwkdrwh08ly3tu9suth4w4"
+        },
+        {
+            "address": "addr1q8j55h253zcvl326sk5qdt2n8z7eghzspe0ekxgncr796s2f79qsdmm5dsknt9ecr5w468r9ey0fxwkdrwh08ly3tu9sjmd35m"
+        },
+        {
+            "address": "addr1q8f7gxrprank3drhx8k5grlux7ene0nlwun8y9thu8mc3yjf79qsdmm5dsknt9ecr5w468r9ey0fxwkdrwh08ly3tu9sls6vnt"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/addresses", json=mock_data)
+    mock_object = [AccountAddressResponse(**data) for data in mock_data]
+    assert api.account_addresses(stake_address=stake_address) == mock_object
 
 
-def test_account_addresses_asset():
-    assert True
+def test_account_addresses_assets(requests_mock):
+    api = BlockFrostApi()
+    mock_data = [
+        {
+            "unit": "d5e6bf0500378d4f0da4e8dde6becec7621cd8cbf5cbb9b87013d4cc537061636542756433343132",
+            "quantity": "1"
+        },
+        {
+            "unit": "b0d07d45fe9514f80213f4020e5a61241458be626841cde717cb38a76e7574636f696e",
+            "quantity": "125"
+        }
+    ]
+    requests_mock.get(f"{api.url}/accounts/{stake_address}/addresses/assets", json=mock_data)
+    mock_object = [AccountAddressesAssetResponse(**data) for data in mock_data]
+    assert api.account_addresses_assets(stake_address=stake_address) == mock_object