chore: again

Signed-off-by: mukundkumarjha <mukundiiitg@gmail.com>

Author: mukundkumarjha

CHANGELOG.md

@@ -167,6 +167,7 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - Added missing type hints to sign method in Transaction class (#1630)
 - Refactored `examples/consensus/topic_create_transaction.py` to use `Client.from_env()` (#1611)
 - Updated GitHub Actions setup-node action to v6.2.0.
+- Refactored AccountInfo class to use the staking_info(#1366)
 - chore: format tests/unit/mock_server.py with black (#1542)
 - Refactored AccountInfo class to use the staking_info
 - Updated actions/checkout to v6.0.1 and actions/github-script v8.0.0 in bot-next-issue-recommendation workflow (#1586)

src/hiero_sdk_python/account/account_info.py

@@ -9,7 +9,7 @@
 from hiero_sdk_python.account.account_id import AccountId
 from hiero_sdk_python.crypto.public_key import PublicKey
 from hiero_sdk_python.Duration import Duration
-from hiero_sdk_python.hapi.services.basic_types_pb2 import StakingInfo
+from hiero_sdk_python.staking_info import StakingInfo
 from hiero_sdk_python.hapi.services.crypto_get_info_pb2 import CryptoGetInfoResponse
 from hiero_sdk_python.hbar import Hbar
 from hiero_sdk_python.timestamp import Timestamp
@@ -99,13 +99,14 @@ def _from_proto(cls, proto: CryptoGetInfoResponse.AccountInfo) -> "AccountInfo":
             account_memo=proto.memo,
             owned_nfts=proto.ownedNfts,
             max_automatic_token_associations=proto.max_automatic_token_associations,
+            staking_info=(
+                StakingInfo._from_proto(proto.staking_info)
+                if proto.HasField("staking_info")
+                else None
+         )
         )
 
-        staking_info=(
-           StakingInfo.from_proto(proto.staking_info)
-           if proto.HasField("staking_info")
-           else None
-         )
+        
 
         return account_info
 
@@ -140,9 +141,9 @@ def _to_proto(self) -> CryptoGetInfoResponse.AccountInfo:
             ownedNfts=self.owned_nfts,
             max_automatic_token_associations=self.max_automatic_token_associations,
             staking_info=(
-            self.staking_info.to_proto()
-            if self.staking_info is not None
-            else None
+                self.staking_info._to_proto()
+                if self.staking_info is not None
+                else None
           ),
         )
 
@@ -158,8 +159,7 @@ def __str__(self) -> str:
             (self.account_memo, "Memo"),
             (self.owned_nfts, "Owned NFTs"),
             (self.max_automatic_token_associations, "Max Automatic Token Associations"),
-            (self.staking_info, "Staked Account ID"),
-            (self.staking_info, "Staked Node ID"),
+            (self.staking_info, "Staked Info"),
             (self.proxy_received, "Proxy Received"),
             (self.expiration_time, "Expiration Time"),
             (self.auto_renew_period, "Auto Renew Period"),
@@ -174,9 +174,6 @@ def __str__(self) -> str:
 
         if self.receiver_signature_required is not None:
             lines.append(f"Receiver Signature Required: {self.receiver_signature_required}")
-            
-        if self.staking_info is not None:
-            lines.append(f"Decline Staking Reward: {self.staking_info}")
 
         if self.token_relationships:
             lines.append(f"Token Relationships: {len(self.token_relationships)}")
@@ -194,7 +191,6 @@ def __repr__(self) -> str:
             f"receiver_signature_required={self.receiver_signature_required!r}, "
             f"owned_nfts={self.owned_nfts!r}, "
             f"account_memo={self.account_memo!r}, "
-            f"staked_node_id={self.staking_info!r}, "
-            f"staked_account_id={self.staking_info!r}"
+            f"staked_info={self.staking_info!r}, "
             f")"
         )

tests/unit/account_info_test.py

@@ -9,6 +9,7 @@
 from hiero_sdk_python.tokens.token_relationship import TokenRelationship
 from hiero_sdk_python.tokens.token_id import TokenId
 from hiero_sdk_python.hapi.services.crypto_get_info_pb2 import CryptoGetInfoResponse
+from hiero_sdk_python.staking_info import StakingInfo
 
 pytestmark = pytest.mark.unit
 
@@ -28,6 +29,8 @@ def account_info():
         token_relationships=[],
         account_memo="Test account memo",
         owned_nfts=5,
+        max_automatic_token_associations=10,
+        staking_info=None
     )
 
 
@@ -47,6 +50,8 @@ def proto_account_info():
         tokenRelationships=[],
         memo="Test account memo",
         ownedNfts=5,
+        max_automatic_token_associations=10,
+        staking_info=None
     )
     return proto
 
@@ -65,7 +70,31 @@ def test_account_info_initialization(account_info):
     assert account_info.token_relationships == []
     assert account_info.account_memo == "Test account memo"
     assert account_info.owned_nfts == 5
+    assert account_info.max_automatic_token_associations == 10
+    assert account_info.staking_info is None
 
+def test_proto_conversion_with_staking_info():
+    """Test converting AccountInfo with StakingInfo to proto and back preserves data"""
+    public_key = PrivateKey.generate_ed25519().public_key()
+    
+    staking_info = StakingInfo(
+        decline_reward=True,
+        staked_node_id=3,
+    )
+
+    account_info = AccountInfo(
+        account_id=AccountId(0, 0, 100),
+        key=public_key,
+        staking_info=staking_info,
+    )
+
+    proto = account_info._to_proto()
+    converted = AccountInfo._from_proto(proto)
+    
+    # Now these assertions should work
+    assert converted.staking_info is not None
+    assert converted.staking_info.decline_reward is True
+    assert converted.staking_info.staked_node_id == 3
 
 def test_account_info_default_initialization():
     """Test the default initialization of the AccountInfo class"""
@@ -82,7 +111,25 @@ def test_account_info_default_initialization():
     assert account_info.token_relationships == []
     assert account_info.account_memo is None
     assert account_info.owned_nfts is None
+    assert account_info.max_automatic_token_associations is None
+    assert account_info.staking_info is None
 
+def test_staking_info_persistence(account_info):
+    """Ensure staking info is preserved through proto conversion"""
+    
+    account_info.staking_info = StakingInfo(
+        decline_reward=True,
+        staked_node_id=5,
+        staked_account_id=None
+    )
+    
+    proto = account_info._to_proto()
+    converted_info = AccountInfo._from_proto(proto)
+
+    assert converted_info.staking_info is not None
+    assert converted_info.staking_info.decline_reward is True
+    assert converted_info.staking_info.staked_node_id == 5
+    assert converted_info.staking_info.staked_account_id is None
 
 def test_from_proto(proto_account_info):
     """Test the from_proto method of the AccountInfo class"""
@@ -100,6 +147,8 @@ def test_from_proto(proto_account_info):
     assert account_info.token_relationships == []
     assert account_info.account_memo == "Test account memo"
     assert account_info.owned_nfts == 5
+    assert account_info.max_automatic_token_associations == 10
+    assert account_info.staking_info == None
 
 
 def test_from_proto_with_token_relationships():
@@ -141,6 +190,11 @@ def test_to_proto(account_info):
     assert proto.tokenRelationships == []
     assert proto.memo == "Test account memo"
     assert proto.ownedNfts == 5
+    assert proto.max_automatic_token_associations == 10
+    assert not proto.HasField("staking_info")
+
+
+
 
 
 def test_to_proto_with_none_values():
@@ -192,6 +246,7 @@ def test_proto_conversion(account_info):
     )
     assert converted_account_info.account_memo == account_info.account_memo
     assert converted_account_info.owned_nfts == account_info.owned_nfts
+    assert converted_account_info.staking_info == account_info.staking_info
 
 
 def test_str_and_repr(account_info):