Skip to content

Commit ec878ed

Browse files
elmatticarajaseklemmih
authored
Market actor unit tests (part 7) (#259)
* Add publish_a_deal_with_enough_collateral_when_circulating_supply_is_superior_to_zero * Add publish_multiple_deals_for_different_clients_and_ensure_balances_are_correct * Add active_deals_multiple_times_with_different_providers Co-authored-by: Aayush Rajasekaran <[email protected]> Co-authored-by: David Himmelstrup <[email protected]>
1 parent f3d38da commit ec878ed

File tree

3 files changed

+396
-22
lines changed

3 files changed

+396
-22
lines changed

actors/market/src/state.rs

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -59,10 +59,10 @@ pub struct State {
5959

6060
/// Total Client Collateral that is locked -> unlocked when deal is terminated
6161
#[serde(with = "bigint_ser")]
62-
pub total_client_locked_colateral: TokenAmount,
62+
pub total_client_locked_collateral: TokenAmount,
6363
/// Total Provider Collateral that is locked -> unlocked when deal is terminated
6464
#[serde(with = "bigint_ser")]
65-
pub total_provider_locked_colateral: TokenAmount,
65+
pub total_provider_locked_collateral: TokenAmount,
6666
/// Total storage fee that is locked in escrow -> unlocked when payments are made
6767
#[serde(with = "bigint_ser")]
6868
pub total_client_storage_fee: TokenAmount,
@@ -98,15 +98,15 @@ impl State {
9898
deal_ops_by_epoch: empty_deal_ops_hamt,
9999
last_cron: EPOCH_UNDEFINED,
100100

101-
total_client_locked_colateral: TokenAmount::default(),
102-
total_provider_locked_colateral: TokenAmount::default(),
101+
total_client_locked_collateral: TokenAmount::default(),
102+
total_provider_locked_collateral: TokenAmount::default(),
103103
total_client_storage_fee: TokenAmount::default(),
104104
})
105105
}
106106

107107
pub fn total_locked(&self) -> TokenAmount {
108-
&self.total_client_locked_colateral
109-
+ &self.total_provider_locked_colateral
108+
&self.total_client_locked_collateral
109+
+ &self.total_provider_locked_collateral
110110
+ &self.total_client_storage_fee
111111
}
112112

@@ -182,8 +182,8 @@ pub(super) struct MarketStateMutation<'bs, 's, BS> {
182182

183183
pub(super) locked_permit: Permission,
184184
pub(super) locked_table: Option<BalanceTable<'bs, BS>>,
185-
pub(super) total_client_locked_colateral: Option<TokenAmount>,
186-
pub(super) total_provider_locked_colateral: Option<TokenAmount>,
185+
pub(super) total_client_locked_collateral: Option<TokenAmount>,
186+
pub(super) total_provider_locked_collateral: Option<TokenAmount>,
187187
pub(super) total_client_storage_fee: Option<TokenAmount>,
188188

189189
pub(super) next_deal_id: DealID,
@@ -210,8 +210,8 @@ where
210210
deals_by_epoch: None,
211211
locked_permit: Permission::Invalid,
212212
locked_table: None,
213-
total_client_locked_colateral: None,
214-
total_provider_locked_colateral: None,
213+
total_client_locked_collateral: None,
214+
total_provider_locked_collateral: None,
215215
total_client_storage_fee: None,
216216
}
217217
}
@@ -227,11 +227,11 @@ where
227227

228228
if self.locked_permit != Permission::Invalid {
229229
self.locked_table = Some(BalanceTable::from_root(self.store, &self.st.locked_table)?);
230-
self.total_client_locked_colateral =
231-
Some(self.st.total_client_locked_colateral.clone());
230+
self.total_client_locked_collateral =
231+
Some(self.st.total_client_locked_collateral.clone());
232232
self.total_client_storage_fee = Some(self.st.total_client_storage_fee.clone());
233-
self.total_provider_locked_colateral =
234-
Some(self.st.total_provider_locked_colateral.clone());
233+
self.total_provider_locked_collateral =
234+
Some(self.st.total_provider_locked_collateral.clone());
235235
}
236236

237237
if self.escrow_permit != Permission::Invalid {
@@ -302,11 +302,11 @@ where
302302
self.st.locked_table =
303303
s.root().map_err(|e| e.downcast_wrap("failed to flush locked table"))?;
304304
}
305-
if let Some(s) = &mut self.total_client_locked_colateral {
306-
self.st.total_client_locked_colateral = s.clone();
305+
if let Some(s) = &mut self.total_client_locked_collateral {
306+
self.st.total_client_locked_collateral = s.clone();
307307
}
308-
if let Some(s) = &mut self.total_provider_locked_colateral {
309-
self.st.total_provider_locked_colateral = s.clone();
308+
if let Some(s) = &mut self.total_provider_locked_collateral {
309+
self.st.total_provider_locked_collateral = s.clone();
310310
}
311311
if let Some(s) = &mut self.total_client_storage_fee {
312312
self.st.total_client_storage_fee = s.clone();
@@ -580,13 +580,13 @@ where
580580
self.maybe_lock_balance(&proposal.provider, &proposal.provider_collateral)
581581
.map_err(|e| e.wrap("failed to lock provider funds"))?;
582582

583-
if let Some(v) = self.total_client_locked_colateral.as_mut() {
583+
if let Some(v) = self.total_client_locked_collateral.as_mut() {
584584
*v += &proposal.client_collateral;
585585
}
586586
if let Some(v) = self.total_client_storage_fee.as_mut() {
587587
*v += proposal.total_storage_fee();
588588
}
589-
if let Some(v) = self.total_provider_locked_colateral.as_mut() {
589+
if let Some(v) = self.total_provider_locked_collateral.as_mut() {
590590
*v += &proposal.provider_collateral;
591591
}
592592
Ok(())
@@ -604,13 +604,13 @@ where
604604
self.locked_table.as_mut().unwrap().must_subtract(addr, amount)?;
605605

606606
match lock_reason {
607-
Reason::ClientCollateral => self.total_client_locked_colateral.as_mut().map(|v| {
607+
Reason::ClientCollateral => self.total_client_locked_collateral.as_mut().map(|v| {
608608
*v -= amount;
609609
}),
610610
Reason::ClientStorageFee => self.total_client_storage_fee.as_mut().map(|v| {
611611
*v -= amount;
612612
}),
613-
Reason::ProviderCollateral => self.total_provider_locked_colateral.as_mut().map(|v| {
613+
Reason::ProviderCollateral => self.total_provider_locked_collateral.as_mut().map(|v| {
614614
*v -= amount;
615615
}),
616616
};

0 commit comments

Comments
 (0)