Fix comment and add TODO back

Author: lowhung

common/src/types.rs

@@ -19,7 +19,7 @@ use std::fmt::{Display, Formatter};
 use std::ops::{AddAssign, Neg};
 use std::{cmp::Ordering, fmt};
 
-/// Address network identifier
+/// Network identifier
 #[derive(
     Debug,
     Clone,

modules/accounts_state/src/state.rs

@@ -991,8 +991,8 @@ impl State {
 mod tests {
     use super::*;
     use acropolis_common::{
-        protocol_params::ConwayParams, rational_number::RationalNumber, NetworkId, Anchor,
-        Committee, Constitution, CostModel, DRepVotingThresholds, PoolVotingThresholds, Pot,
+        protocol_params::ConwayParams, rational_number::RationalNumber, Anchor, Committee,
+        Constitution, CostModel, DRepVotingThresholds, NetworkId, PoolVotingThresholds, Pot,
         PotDelta, Ratio, Registration, StakeAddress, StakeAddressDelta, StakeAddressPayload,
         StakeAndVoteDelegation, StakeRegistrationAndStakeAndVoteDelegation,
         StakeRegistrationAndVoteDelegation, VoteDelegation, Withdrawal,

modules/drep_state/src/state.rs

@@ -490,6 +490,7 @@ impl State {
         let response = context.message_bus.request(&accounts_query_topic, msg).await?;
         let message = Arc::try_unwrap(response).unwrap_or_else(|arc| (*arc).clone());
 
+        // TODO: Ensure AccountsStateQueryResponse is for the correct block
         let result_map = match message {
             Message::StateQueryResponse(StateQueryResponse::Accounts(
                 AccountsStateQueryResponse::AccountsDrepDelegationsMap(map),

modules/stake_delta_filter/src/utils.rs

@@ -401,8 +401,9 @@ mod test {
     use crate::*;
     use acropolis_common::{
         messages::AddressDeltasMessage, Address, AddressDelta, BlockHash, BlockInfo, BlockStatus,
-        ByronAddress, Era, ShelleyAddress, ShelleyAddressDelegationPart, ShelleyAddressPaymentPart,
-        ShelleyAddressPointer, StakeAddress, StakeAddressPayload, UTxOIdentifier, ValueDelta, NetworkId
+        ByronAddress, Era, NetworkId, ShelleyAddress, ShelleyAddressDelegationPart,
+        ShelleyAddressPaymentPart, ShelleyAddressPointer, StakeAddress, StakeAddressPayload,
+        UTxOIdentifier, ValueDelta,
     };
     use bech32::{Bech32, Hrp};