test: add integration tests (#87)

* feat: add integartion tests

Signed-off-by: dosi <[email protected]>

* test: move unit tests to separate folder

Signed-off-by: dosi <[email protected]>

* test: add pytest.ini with markers for unit and integration tests

Signed-off-by: dosi <[email protected]>

* ci: update GitHub Actions workflow and pyproject.toml for testing and configuration

Signed-off-by: dosi <[email protected]>

* test: remove transaction.sign() where the call is unnecessary

Signed-off-by: dosi <[email protected]>

* test: remove except Exception lines in integration tests

Signed-off-by: dosi <[email protected]>

* test: remove token deletion and dissociation where unnecessary

Signed-off-by: dosi <[email protected]>

---------

Signed-off-by: dosi <[email protected]>

Author: Dosik13

.github/workflows/test.yml

@@ -43,12 +43,16 @@ jobs:
       - name: Install your package
         run: pip install -e .
 
-      - name: Run tests
+      - name: Run integration tests
         env:
           OPERATOR_ID: ${{ steps.solo.outputs.accountId }}
           OPERATOR_KEY: ${{ steps.solo.outputs.privateKey }}
           ADMIN_KEY: ${{ steps.solo.outputs.privateKey }}
           PUBLIC_KEY: ${{ steps.solo.outputs.publicKey }}
           NETWORK: solo
         run: |
-          python test.py; uv run pytest
+          uv run pytest -m integration
+
+      - name: Run unit tests
+        run: |
+          uv run pytest -m unit

pyproject.toml

@@ -53,7 +53,8 @@ package-dir = "src"
 excludes = ["**/.pytest_cache"]
 
 [tool.pytest.ini_options]
-pythonpath = ["src"]
+pythonpath = ["src", "."]
+testpaths = ["tests"]
 
 [tool.ruff]
 line-length = 120

pytest.ini

@@ -0,0 +1,4 @@
+[pytest]
+markers =
+    integration: mark a test as an integration test.
+    unit: mark a test as a unit test. 

tests/integration/__init__.py

@@ -0,0 +1,14 @@
+import pytest
+
+from hiero_sdk_python.query.account_balance_query import CryptoGetAccountBalanceQuery
+from tests.integration.utils_for_test import IntegrationTestEnv
+
+
+@pytest.mark.integration
+def test_integration_account_balance_query_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        CryptoGetAccountBalanceQuery(account_id=env.operator_id).execute(env.client)
+    finally:
+        env.close()

tests/integration/account_balance_query_e2e_test.py

@@ -0,0 +1,33 @@
+import pytest
+from hiero_sdk_python import AccountCreateTransaction
+from hiero_sdk_python.crypto.private_key import PrivateKey
+from hiero_sdk_python.hbar import Hbar
+from tests.integration.utils_for_test import IntegrationTestEnv
+
+
+@pytest.mark.integration
+def test_integration_account_create_transaction_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        new_account_private_key = PrivateKey.generate()
+        new_account_public_key = new_account_private_key.public_key()
+        initial_balance = Hbar(2)
+        
+        assert initial_balance.to_tinybars() == 200000000
+        
+        transaction = AccountCreateTransaction(
+            key=new_account_public_key,
+            initial_balance=initial_balance,
+            memo="Recipient Account"
+        )
+        
+        transaction.freeze_with(env.client)
+        receipt = transaction.execute(env.client)
+        
+        assert receipt.accountId is not None, "AccountID not found in receipt. Account may not have been created."
+    finally:
+        env.close()
+    
+    
+    

tests/integration/account_create_transaction_e2e_test.py

@@ -0,0 +1,58 @@
+import pytest
+
+from hiero_sdk_python.crypto.private_key import PrivateKey
+from hiero_sdk_python.hbar import Hbar
+from hiero_sdk_python.tokens.token_associate_transaction import TokenAssociateTransaction
+from hiero_sdk_python.tokens.token_dissociate_transaction import TokenDissociateTransaction
+from hiero_sdk_python.account.account_create_transaction import AccountCreateTransaction
+from hiero_sdk_python.response_code import ResponseCode
+from tests.integration.utils_for_test import IntegrationTestEnv, create_fungible_token
+
+
+@pytest.mark.integration
+def test_integration_token_associate_transaction_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        new_account_private_key = PrivateKey.generate()
+        new_account_public_key = new_account_private_key.public_key()
+        
+        initial_balance = Hbar(2)
+        assert initial_balance.to_tinybars() == 200000000
+        
+        account_transaction = AccountCreateTransaction(
+            key=new_account_public_key,
+            initial_balance=initial_balance,
+            memo="Recipient Account"
+        )
+        
+        account_transaction.freeze_with(env.client)
+        account_receipt = account_transaction.execute(env.client)
+        new_account_id = account_receipt.accountId
+        
+        token_id = create_fungible_token(env)
+        assert token_id is not None, "TokenID not found in receipt. Token may not have been created."
+        
+        associate_transaction = TokenAssociateTransaction(
+            account_id=new_account_id,
+            token_ids=[token_id]
+        )
+        
+        associate_transaction.freeze_with(env.client)
+        associate_transaction.sign(new_account_private_key)
+        associate_receipt = associate_transaction.execute(env.client)
+        
+        assert associate_receipt.status == ResponseCode.SUCCESS, f"Token association failed with status: {ResponseCode.get_name(associate_receipt.status)}"
+    
+        dissociate_transaction = TokenDissociateTransaction(
+            account_id=new_account_id,
+            token_ids=[token_id]
+        )
+        
+        dissociate_transaction.freeze_with(env.client)
+        dissociate_transaction.sign(new_account_private_key)
+        dissociate_receipt = dissociate_transaction.execute(env.client)
+        
+        assert dissociate_receipt.status == ResponseCode.SUCCESS, f"Token dissociation failed with status: {ResponseCode.get_name(dissociate_receipt.status)}"
+    finally:
+        env.close() 

tests/integration/token_associate_transaction_e2e_test.py

@@ -0,0 +1,27 @@
+import pytest
+
+from tests.integration.utils_for_test import IntegrationTestEnv, create_fungible_token, create_nft_token
+
+
+@pytest.mark.integration
+def test_integration_fungible_token_create_transaction_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        token_id = create_fungible_token(env)
+        
+        assert token_id is not None, "TokenID not found in receipt. Token may not have been created."
+    finally:
+        env.close()
+
+
+@pytest.mark.integration
+def test_integration_nft_token_create_transaction_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        token_id = create_nft_token(env)
+        
+        assert token_id is not None, "TokenID not found in receipt. Token may not have been created."
+    finally:
+        env.close()

tests/integration/token_create_transaction_e2e_test.py

@@ -0,0 +1,22 @@
+import pytest
+
+from tests.integration.utils_for_test import IntegrationTestEnv, create_fungible_token
+from hiero_sdk_python.tokens.token_delete_transaction import TokenDeleteTransaction
+from hiero_sdk_python.response_code import ResponseCode
+
+@pytest.mark.integration
+def test_integration_token_delete_transaction_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        token_id = create_fungible_token(env)
+        
+        assert token_id is not None, "TokenID not found in receipt. Token may not have been created."
+        
+        transaction = TokenDeleteTransaction(token_id=token_id)
+        transaction.freeze_with(env.client)
+        receipt = transaction.execute(env.client)
+        
+        assert receipt.status == ResponseCode.SUCCESS, f"Token deletion failed with status: {ResponseCode.get_name(receipt.status)}"
+    finally:
+        env.close() 

tests/integration/token_delete_transaction_e2e_test.py

@@ -0,0 +1,58 @@
+import pytest
+
+from hiero_sdk_python.account.account_create_transaction import AccountCreateTransaction
+from hiero_sdk_python.crypto.private_key import PrivateKey
+from hiero_sdk_python.hbar import Hbar
+from hiero_sdk_python.tokens.token_associate_transaction import TokenAssociateTransaction
+from hiero_sdk_python.tokens.token_dissociate_transaction import TokenDissociateTransaction
+from hiero_sdk_python.response_code import ResponseCode
+from tests.integration.utils_for_test import IntegrationTestEnv, create_fungible_token
+
+
+@pytest.mark.integration
+def test_integration_token_dissociate_transaction_can_execute():
+    env = IntegrationTestEnv()
+    
+    try:
+        new_account_private_key = PrivateKey.generate()
+        new_account_public_key = new_account_private_key.public_key()
+        
+        initial_balance = Hbar(2)
+        
+        assert initial_balance.to_tinybars() == 200000000
+        
+        account_transaction = AccountCreateTransaction(
+            key=new_account_public_key,
+            initial_balance=initial_balance,
+            memo="Recipient Account"
+        )
+        
+        account_transaction.freeze_with(env.client)
+        account_receipt = account_transaction.execute(env.client)
+        new_account_id = account_receipt.accountId
+        
+        token_id = create_fungible_token(env)
+        
+        associate_transaction = TokenAssociateTransaction(
+            account_id=new_account_id,
+            token_ids=[token_id]
+        )
+        associate_transaction.freeze_with(env.client)
+        associate_transaction.sign(new_account_private_key)
+        
+        receipt = associate_transaction.execute(env.client)
+        
+        assert receipt.status == ResponseCode.SUCCESS, f"Token association failed with status: {ResponseCode.get_name(receipt.status)}"
+        
+        dissociate_transaction = TokenDissociateTransaction(
+            account_id=new_account_id,
+            token_ids=[token_id]
+        )
+        dissociate_transaction.freeze_with(env.client)
+        dissociate_transaction.sign(new_account_private_key)
+        
+        receipt = dissociate_transaction.execute(env.client)
+        
+        assert receipt.status == ResponseCode.SUCCESS, f"Token dissociation failed with status: {ResponseCode.get_name(receipt.status)}"
+    finally:
+        env.close()