use super::FeeReserveFinalizationSummary;
use crate::internal_prelude::*;
use crate::{
    errors::CanBeAbortion,
    transaction::{AbortReason, CostingParameters},
};
use radix_engine_interface::blueprints::resource::LiquidFungibleResource;
use radix_transactions::model::TipSpecifier;
use radix_transactions::prelude::TransactionCostingParameters;
use sbor::rust::cmp::min;

#[derive(Debug, Clone, PartialEq, Eq, ScryptoSbor)]
pub enum FeeReserveError {
    InsufficientBalance {
        required: Decimal,
        remaining: Decimal,
    },
    Overflow,
    LimitExceeded {
        limit: u32,
        committed: u32,
        new: u32,
    },
    LoanRepaymentFailed {
        xrd_owed: Decimal,
    },
    Abort(AbortReason),
}

#[derive(Copy, Debug, Clone, Hash, PartialEq, Eq, PartialOrd, Ord, ScryptoSbor)]
pub enum StorageType {
    State,
    Archive,
}

impl CanBeAbortion for FeeReserveError {
    fn abortion(&self) -> Option<&AbortReason> {
        match self {
            Self::Abort(reason) => Some(reason),
            _ => None,
        }
    }
}

/// This is only allowed before a transaction properly begins.
/// After any other methods are called, this cannot be called again.
pub trait PreExecutionFeeReserve {
    fn consume_deferred_execution(&mut self, cost_units: u32) -> Result<(), FeeReserveError>;

    fn consume_deferred_finalization(&mut self, cost_units: u32) -> Result<(), FeeReserveError>;

    fn consume_deferred_storage(
        &mut self,
        storage_type: StorageType,
        size_increase: usize,
    ) -> Result<(), FeeReserveError>;
}

pub trait ExecutionFeeReserve {
    fn consume_execution(&mut self, cost_units: u32) -> Result<(), FeeReserveError>;

    fn consume_finalization(&mut self, cost_units: u32) -> Result<(), FeeReserveError>;

    fn consume_storage(
        &mut self,
        storage_type: StorageType,
        size_increase: usize,
    ) -> Result<(), FeeReserveError>;

    fn consume_royalty(
        &mut self,
        royalty_amount: RoyaltyAmount,
        recipient: RoyaltyRecipient,
    ) -> Result<(), FeeReserveError>;

    fn lock_fee(&mut self, vault_id: NodeId, fee: LiquidFungibleResource, contingent: bool);
}

pub trait FinalizingFeeReserve {
    fn finalize(
        self,
    ) -> (
        FeeReserveFinalizationSummary,
        CostingParameters,
        TransactionCostingParameters,
    );
}

pub trait FeeReserve: PreExecutionFeeReserve + ExecutionFeeReserve + FinalizingFeeReserve {}

#[derive(Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord, ScryptoSbor)]
pub enum RoyaltyRecipient {
    Package(PackageAddress, NodeId),
    Component(ComponentAddress, NodeId),
}

impl RoyaltyRecipient {
    pub fn vault_id(&self) -> NodeId {
        match self {
            RoyaltyRecipient::Package(_, v) | RoyaltyRecipient::Component(_, v) => *v,
        }
    }
}

#[derive(Debug, Clone, ScryptoSbor)]
pub struct SystemLoanFeeReserve {
    costing_parameters: CostingParameters,
    transaction_costing_parameters: TransactionCostingParameters,
    abort_when_loan_repaid: bool,

    /// (Cache) The effective execution cost unit price, with tips considered
    effective_execution_cost_unit_price: Decimal,
    /// (Cache) The effective finalization cost unit price, with tips considered
    effective_finalization_cost_unit_price: Decimal,

    /// The XRD balance
    xrd_balance: Decimal,
    /// The amount of XRD owed to the system
    xrd_owed: Decimal,

    /// Execution costs
    execution_cost_units_committed: u32,
    execution_cost_units_deferred: u32,

    // Finalization costs
    finalization_cost_units_committed: u32,
    finalization_cost_units_deferred: u32,

    /// Royalty costs
    royalty_cost_committed: Decimal,
    royalty_cost_breakdown: IndexMap<RoyaltyRecipient, Decimal>,

    /// Storage Costs
    storage_cost_committed: Decimal,
    storage_cost_deferred: IndexMap<StorageType, usize>,

    /// Payments made during the execution of a transaction.
    locked_fees: Vec<(NodeId, LiquidFungibleResource, bool)>,
}

impl Default for SystemLoanFeeReserve {
    fn default() -> Self {
        Self::new(
            CostingParameters::babylon_genesis(),
            TransactionCostingParameters::default(),
            false,
        )
    }
}

#[inline]
fn checked_add(a: u32, b: u32) -> Result<u32, FeeReserveError> {
    a.checked_add(b).ok_or(FeeReserveError::Overflow)
}

#[inline]
fn checked_add_assign(value: &mut u32, summand: u32) -> Result<(), FeeReserveError> {
    *value = checked_add(*value, summand)?;
    Ok(())
}

impl SystemLoanFeeReserve {
    pub fn new(
        costing_parameters: CostingParameters,
        transaction_costing_parameters: TransactionCostingParameters,
        abort_when_loan_repaid: bool,
    ) -> Self {
        // Sanity checks
        assert!(!costing_parameters.execution_cost_unit_price.is_negative());
        assert!(!costing_parameters
            .finalization_cost_unit_price
            .is_negative());
        assert!(!costing_parameters.usd_price.is_negative());
        assert!(!costing_parameters.state_storage_price.is_negative());
        assert!(!costing_parameters.archive_storage_price.is_negative());
        assert!(!transaction_costing_parameters
            .free_credit_in_xrd
            .is_negative());

        let tip_multiplier = transaction_costing_parameters.tip.fee_multiplier();

        let effective_execution_cost_unit_price = costing_parameters
            .execution_cost_unit_price
            .checked_mul(tip_multiplier)
            .unwrap();

        let effective_finalization_cost_unit_price = costing_parameters
            .finalization_cost_unit_price
            .checked_mul(tip_multiplier)
            .unwrap();

        let system_loan_in_xrd = effective_execution_cost_unit_price
            .checked_mul(costing_parameters.execution_cost_unit_loan)
            .unwrap();

        let starting_xrd_balance = system_loan_in_xrd
            .checked_add(transaction_costing_parameters.free_credit_in_xrd)
            .expect("Invalid system loan or free credit amount");

        Self {
            costing_parameters,
            transaction_costing_parameters,
            abort_when_loan_repaid,

            // Cache
            effective_execution_cost_unit_price,
            effective_finalization_cost_unit_price,

            // Running balance
            xrd_balance: starting_xrd_balance,
            xrd_owed: system_loan_in_xrd,

            // Internal states
            execution_cost_units_committed: 0,
            execution_cost_units_deferred: 0,

            finalization_cost_units_committed: 0,
            finalization_cost_units_deferred: 0,

            royalty_cost_breakdown: index_map_new(),
            royalty_cost_committed: Decimal::ZERO,

            storage_cost_committed: Decimal::ZERO,
            storage_cost_deferred: index_map_new(),

            locked_fees: Vec::new(),
        }
    }

    pub fn costing_parameters(&self) -> &CostingParameters {
        &self.costing_parameters
    }

    pub fn transaction_costing_parameters(&self) -> &TransactionCostingParameters {
        &self.transaction_costing_parameters
    }

    pub fn execution_cost_unit_limit(&self) -> u32 {
        self.costing_parameters.execution_cost_unit_limit
    }

    pub fn execution_cost_unit_price(&self) -> Decimal {
        self.costing_parameters.execution_cost_unit_price
    }

    pub fn finalization_cost_unit_limit(&self) -> u32 {
        self.costing_parameters.finalization_cost_unit_limit
    }

    pub fn finalization_cost_unit_price(&self) -> Decimal {
        self.costing_parameters.finalization_cost_unit_price
    }

    pub fn usd_price(&self) -> Decimal {
        self.costing_parameters.usd_price
    }

    pub fn tip(&self) -> TipSpecifier {
        self.transaction_costing_parameters.tip
    }

    pub fn fee_balance(&self) -> Decimal {
        self.xrd_balance
    }

    pub fn royalty_cost_breakdown(&self) -> &IndexMap<RoyaltyRecipient, Decimal> {
        &self.royalty_cost_breakdown
    }

    fn check_execution_cost_unit_limit(&self, cost_units: u32) -> Result<(), FeeReserveError> {
        if checked_add(self.execution_cost_units_committed, cost_units)?
            > self.costing_parameters.execution_cost_unit_limit
        {
            return Err(FeeReserveError::LimitExceeded {
                limit: self.costing_parameters.execution_cost_unit_limit,
                committed: self.execution_cost_units_committed,
                new: cost_units,
            });
        }
        Ok(())
    }

    fn check_finalization_cost_unit_limit(&self, cost_units: u32) -> Result<(), FeeReserveError> {
        if checked_add(self.finalization_cost_units_committed, cost_units)?
            > self.costing_parameters.finalization_cost_unit_limit
        {
            return Err(FeeReserveError::LimitExceeded {
                limit: self.costing_parameters.finalization_cost_unit_limit,
                committed: self.finalization_cost_units_committed,
                new: cost_units,
            });
        }
        Ok(())
    }

    fn consume_execution_internal(&mut self, cost_units: u32) -> Result<(), FeeReserveError> {
        self.check_execution_cost_unit_limit(cost_units)?;

        let amount = self
            .effective_execution_cost_unit_price
            .checked_mul(cost_units)
            .ok_or(FeeReserveError::Overflow)?;
        if self.xrd_balance < amount {
            Err(FeeReserveError::InsufficientBalance {
                required: amount,
                remaining: self.xrd_balance,
            })
        } else {
            self.xrd_balance -= amount;
            self.execution_cost_units_committed += cost_units;
            Ok(())
        }
    }

    fn consume_finalization_internal(&mut self, cost_units: u32) -> Result<(), FeeReserveError> {
        self.check_finalization_cost_unit_limit(cost_units)?;

        let amount = self
            .effective_finalization_cost_unit_price
            .checked_mul(cost_units)
            .ok_or(FeeReserveError::Overflow)?;
        if self.xrd_balance < amount {
            Err(FeeReserveError::InsufficientBalance {
                required: amount,
                remaining: self.xrd_balance,
            })
        } else {
            self.xrd_balance -= amount;
            self.finalization_cost_units_committed += cost_units;
            Ok(())
        }
    }

    fn consume_royalty_internal(
        &mut self,
        royalty_amount: RoyaltyAmount,
        recipient: RoyaltyRecipient,
    ) -> Result<(), FeeReserveError> {
        let amount = match royalty_amount {
            RoyaltyAmount::Xrd(xrd_amount) => xrd_amount,
            RoyaltyAmount::Usd(usd_amount) => usd_amount
                .checked_mul(self.costing_parameters.usd_price)
                .ok_or(FeeReserveError::Overflow)?,
            RoyaltyAmount::Free => Decimal::ZERO,
        };

        if self.xrd_balance < amount {
            Err(FeeReserveError::InsufficientBalance {
                required: amount,
                remaining: self.xrd_balance,
            })
        } else {
            self.xrd_balance -= amount;
            self.royalty_cost_breakdown
                .entry(recipient)
                .or_default()
                .add_assign(amount);
            self.royalty_cost_committed += amount;
            Ok(())
        }
    }

    pub fn repay_all(&mut self) -> Result<(), FeeReserveError> {
        // Apply deferred execution cost
        self.consume_execution_internal(self.execution_cost_units_deferred)?;
        self.execution_cost_units_deferred = 0;

        // Apply deferred finalization cost
        self.consume_finalization_internal(self.finalization_cost_units_deferred)?;
        self.finalization_cost_units_deferred = 0;

        // Apply deferred storage cost
        let types: Vec<StorageType> = self.storage_cost_deferred.keys().cloned().collect();
        for t in types {
            self.consume_storage(t, self.storage_cost_deferred.get(&t).cloned().unwrap())?;
            self.storage_cost_deferred.swap_remove(&t);
        }

        // Repay owed with balance
        let amount = min(self.xrd_balance, self.xrd_owed);
        self.xrd_owed -= amount;
        self.xrd_balance -= amount; // not used afterwards

        // Check outstanding loan
        if !self.xrd_owed.is_zero() {
            return Err(FeeReserveError::LoanRepaymentFailed {
                xrd_owed: self.xrd_owed,
            });
        }

        if self.abort_when_loan_repaid {
            return Err(FeeReserveError::Abort(
                AbortReason::ConfiguredAbortTriggeredOnFeeLoanRepayment,
            ));
        }

        Ok(())
    }

    pub fn revert_royalty(&mut self) {
        self.xrd_balance += self.royalty_cost_committed;
        self.royalty_cost_breakdown.clear();
        self.royalty_cost_committed = Decimal::ZERO;
    }

    #[inline]
    pub fn fully_repaid(&self) -> bool {
        // The xrd_owed state is not reset before all deferred costs are applied.
        // Thus, not checking the deferred balance
        self.xrd_owed == Decimal::ZERO
    }
}

impl PreExecutionFeeReserve for SystemLoanFeeReserve {
    fn consume_deferred_execution(&mut self, cost_units: u32) -> Result<(), FeeReserveError> {
        checked_add_assign(&mut self.execution_cost_units_deferred, cost_units)?;

        Ok(())
    }

    fn consume_deferred_finalization(&mut self, cost_units: u32) -> Result<(), FeeReserveError> {
        checked_add_assign(&mut self.finalization_cost_units_deferred, cost_units)?;

        Ok(())
    }

    fn consume_deferred_storage(
        &mut self,
        storage_type: StorageType,
        size_increase: usize,
    ) -> Result<(), FeeReserveError> {
        self.storage_cost_deferred
            .entry(storage_type)
            .or_default()
            .add_assign(size_increase);

        Ok(())
    }
}

impl ExecutionFeeReserve for SystemLoanFeeReserve {
    fn consume_execution(&mut self, cost_units: u32) -> Result<(), FeeReserveError> {
        if cost_units == 0 {
            return Ok(());
        }

        self.consume_execution_internal(cost_units)?;

        if !self.fully_repaid()
            && self.execution_cost_units_committed
                >= self.costing_parameters.execution_cost_unit_loan
        {
            self.repay_all()?;
        }

        Ok(())
    }

    fn consume_finalization(&mut self, cost_units: u32) -> Result<(), FeeReserveError> {
        if cost_units == 0 {
            return Ok(());
        }

        self.consume_finalization_internal(cost_units)?;

        Ok(())
    }

    fn consume_royalty(
        &mut self,
        royalty_amount: RoyaltyAmount,
        recipient: RoyaltyRecipient,
    ) -> Result<(), FeeReserveError> {
        if royalty_amount.is_zero() {
            return Ok(());
        }
        if royalty_amount.is_negative() {
            panic!("System invariant broken: Encountered negative royalty amount")
        }

        self.consume_royalty_internal(royalty_amount, recipient)?;

        Ok(())
    }

    fn consume_storage(
        &mut self,
        storage_type: StorageType,
        size_increase: usize,
    ) -> Result<(), FeeReserveError> {
        let amount = match storage_type {
            StorageType::State => self.costing_parameters.state_storage_price,
            StorageType::Archive => self.costing_parameters.archive_storage_price,
        }
        .checked_mul(size_increase)
        .ok_or(FeeReserveError::Overflow)?;

        if self.xrd_balance < amount {
            Err(FeeReserveError::InsufficientBalance {
                required: amount,
                remaining: self.xrd_balance,
            })
        } else {
            self.xrd_balance -= amount;
            self.storage_cost_committed += amount;
            Ok(())
        }
    }

    fn lock_fee(&mut self, vault_id: NodeId, mut fee: LiquidFungibleResource, contingent: bool) {
        // Update balance
        if !contingent {
            self.xrd_balance = self
                .xrd_balance
                .checked_add(fee.amount())
                .expect("No overflow due to limited XRD supply");
        }

        // Move resource
        self.locked_fees
            .push((vault_id, fee.take_all(), contingent));
    }
}

impl FinalizingFeeReserve for SystemLoanFeeReserve {
    fn finalize(
        self,
    ) -> (
        FeeReserveFinalizationSummary,
        CostingParameters,
        TransactionCostingParameters,
    ) {
        let total_execution_cost_in_xrd = self
            .costing_parameters
            .execution_cost_unit_price
            .checked_mul(Decimal::from(self.execution_cost_units_committed))
            .unwrap();

        let total_finalization_cost_in_xrd = self
            .costing_parameters
            .finalization_cost_unit_price
            .checked_mul(Decimal::from(self.finalization_cost_units_committed))
            .unwrap();

        let tip_proportion = self.transaction_costing_parameters.tip.proportion();
        let total_tipping_cost_in_xrd = total_execution_cost_in_xrd
            .checked_mul(tip_proportion)
            .unwrap()
            .checked_add(
                total_finalization_cost_in_xrd
                    .checked_mul(tip_proportion)
                    .unwrap(),
            )
            .unwrap();

        let summary = FeeReserveFinalizationSummary {
            total_execution_cost_units_consumed: self.execution_cost_units_committed,
            total_finalization_cost_units_consumed: self.finalization_cost_units_committed,

            total_execution_cost_in_xrd,
            total_finalization_cost_in_xrd,
            total_tipping_cost_in_xrd,
            total_royalty_cost_in_xrd: self.royalty_cost_committed,
            total_storage_cost_in_xrd: self.storage_cost_committed,
            total_bad_debt_in_xrd: self.xrd_owed,
            locked_fees: self.locked_fees,
            royalty_cost_breakdown: self.royalty_cost_breakdown,
        };

        (
            summary,
            self.costing_parameters,
            self.transaction_costing_parameters,
        )
    }
}

impl FeeReserve for SystemLoanFeeReserve {}

#[cfg(test)]
mod tests {
    use super::*;

    const TEST_COMPONENT: ComponentAddress =
        component_address(EntityType::GlobalGenericComponent, 5);
    const TEST_VAULT_ID: NodeId = NodeId([0u8; NodeId::LENGTH]);
    const TEST_VAULT_ID_2: NodeId = NodeId([1u8; NodeId::LENGTH]);

    fn xrd<T: Into<Decimal>>(amount: T) -> LiquidFungibleResource {
        LiquidFungibleResource::new(amount.into())
    }

    fn create_test_fee_reserve(
        execution_cost_unit_price: Decimal,
        usd_price: Decimal,
        state_storage_price: Decimal,
        tip_percentage: u16,
        execution_cost_unit_limit: u32,
        execution_cost_unit_loan: u32,
        abort_when_loan_repaid: bool,
    ) -> SystemLoanFeeReserve {
        let mut costing_parameters = CostingParameters::babylon_genesis();
        costing_parameters.execution_cost_unit_price = execution_cost_unit_price;
        costing_parameters.execution_cost_unit_limit = execution_cost_unit_limit;
        costing_parameters.execution_cost_unit_loan = execution_cost_unit_loan;
        costing_parameters.usd_price = usd_price;
        costing_parameters.state_storage_price = state_storage_price;
        let transaction_costing_parameters = TransactionCostingParameters {
            tip: TipSpecifier::Percentage(tip_percentage),
            ..Default::default()
        };

        SystemLoanFeeReserve::new(
            costing_parameters,
            transaction_costing_parameters,
            abort_when_loan_repaid,
        )
    }

    #[test]
    fn test_consume_and_repay() {
        let mut fee_reserve = create_test_fee_reserve(dec!(1), dec!(1), dec!(0), 2, 100, 5, false);
        fee_reserve.consume_execution(2).unwrap();
        fee_reserve.lock_fee(TEST_VAULT_ID, xrd(3), false);
        fee_reserve.repay_all().unwrap();
        let (summary, _, _) = fee_reserve.finalize();
        assert!(summary.loan_fully_repaid());
        assert_eq!(summary.total_execution_cost_units_consumed, 2);
        assert_eq!(summary.total_execution_cost_in_xrd, dec!("2"));
        assert_eq!(summary.total_tipping_cost_in_xrd, dec!("0.04"));
        assert_eq!(summary.total_royalty_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_bad_debt_in_xrd, dec!("0"));
    }

    #[test]
    fn test_out_of_cost_unit() {
        let mut fee_reserve = create_test_fee_reserve(dec!(1), dec!(1), dec!(0), 2, 100, 5, false);
        assert_eq!(
            fee_reserve.consume_execution(6),
            Err(FeeReserveError::InsufficientBalance {
                required: dec!("6.12"),
                remaining: dec!("5.1"),
            }),
        );
        fee_reserve.repay_all().unwrap();
        let (summary, _, _) = fee_reserve.finalize();
        assert!(summary.loan_fully_repaid());
        assert_eq!(summary.total_execution_cost_units_consumed, 0);
        assert_eq!(summary.total_execution_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_royalty_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_bad_debt_in_xrd, dec!("0"));
    }

    #[test]
    fn test_lock_fee() {
        let mut fee_reserve =
            create_test_fee_reserve(dec!(1), dec!(1), dec!(0), 2, 100, 500, false);
        fee_reserve.lock_fee(TEST_VAULT_ID, xrd(100), false);
        fee_reserve.repay_all().unwrap();
        let (summary, _, _) = fee_reserve.finalize();
        assert!(summary.loan_fully_repaid());
        assert_eq!(summary.total_execution_cost_units_consumed, 0);
        assert_eq!(summary.total_execution_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_royalty_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_bad_debt_in_xrd, dec!("0"));
    }

    #[test]
    fn test_xrd_cost_unit_conversion() {
        let mut fee_reserve =
            create_test_fee_reserve(dec!(5), dec!(1), dec!(0), 0, 100, 500, false);
        fee_reserve.lock_fee(TEST_VAULT_ID, xrd(100), false);
        fee_reserve.repay_all().unwrap();
        let (summary, _, _) = fee_reserve.finalize();
        assert!(summary.loan_fully_repaid());
        assert_eq!(summary.total_execution_cost_units_consumed, 0);
        assert_eq!(summary.total_execution_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_royalty_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_bad_debt_in_xrd, dec!("0"));
        assert_eq!(summary.locked_fees, vec![(TEST_VAULT_ID, xrd(100), false)],);
    }

    #[test]
    fn test_bad_debt() {
        let mut fee_reserve = create_test_fee_reserve(dec!(5), dec!(1), dec!(0), 1, 100, 50, false);
        fee_reserve.consume_execution(2).unwrap();
        assert_eq!(
            fee_reserve.repay_all(),
            Err(FeeReserveError::LoanRepaymentFailed {
                xrd_owed: dec!("10.1")
            })
        );
        let (summary, _, _) = fee_reserve.finalize();
        assert!(!summary.loan_fully_repaid());
        assert_eq!(summary.total_execution_cost_units_consumed, 2);
        assert_eq!(summary.total_execution_cost_in_xrd, dec!("10"));
        assert_eq!(summary.total_tipping_cost_in_xrd, dec!("0.1"));
        assert_eq!(summary.total_royalty_cost_in_xrd, dec!("0"));
        assert_eq!(summary.total_bad_debt_in_xrd, dec!("10.1"));
        assert_eq!(summary.locked_fees, vec![],);
    }

    #[test]
    fn test_royalty_execution_mix() {
        let mut fee_reserve = create_test_fee_reserve(dec!(5), dec!(2), dec!(0), 1, 100, 50, false);
        fee_reserve.consume_execution(2).unwrap();
        fee_reserve
            .consume_royalty(
                RoyaltyAmount::Xrd(2.into()),
                RoyaltyRecipient::Package(PACKAGE_PACKAGE, TEST_VAULT_ID),
            )
            .unwrap();
        fee_reserve
            .consume_royalty(
                RoyaltyAmount::Usd(7.into()),
                RoyaltyRecipient::Package(PACKAGE_PACKAGE, TEST_VAULT_ID),
            )
            .unwrap();
        fee_reserve.lock_fee(TEST_VAULT_ID, xrd(100), false);
        fee_reserve.repay_all().unwrap();
        let (summary, _, _) = fee_reserve.finalize();
        assert!(summary.loan_fully_repaid());
        assert_eq!(summary.total_execution_cost_in_xrd, dec!("10"));
        assert_eq!(summary.total_tipping_cost_in_xrd, dec!("0.1"));
        assert_eq!(summary.total_royalty_cost_in_xrd, dec!("16"));
        assert_eq!(summary.total_bad_debt_in_xrd, dec!("0"));
        assert_eq!(summary.locked_fees, vec![(TEST_VAULT_ID, xrd(100), false)]);
        assert_eq!(summary.total_execution_cost_units_consumed, 2);
        assert_eq!(
            summary.royalty_cost_breakdown,
            indexmap!(
                RoyaltyRecipient::Package(PACKAGE_PACKAGE, TEST_VAULT_ID) => dec!("16")
            )
        );
    }

    #[test]
    fn test_royalty_insufficient_balance() {
        let mut fee_reserve =
            create_test_fee_reserve(dec!(1), dec!(1), dec!(0), 0, 1000, 50, false);
        fee_reserve.lock_fee(TEST_VAULT_ID, xrd(100), false);
        fee_reserve
            .consume_royalty(
                RoyaltyAmount::Xrd(90.into()),
                RoyaltyRecipient::Package(PACKAGE_PACKAGE, TEST_VAULT_ID),
            )
            .unwrap();
        assert_eq!(
            fee_reserve.consume_royalty(
                RoyaltyAmount::Xrd(80.into()),
                RoyaltyRecipient::Component(TEST_COMPONENT, TEST_VAULT_ID_2),
            ),
            Err(FeeReserveError::InsufficientBalance {
                required: dec!("80"),
                remaining: dec!("60"),
            }),
        );
    }
}