Skip to content

Commit b2bfb5e

Browse files
committed
chore: use STXBalance
1 parent f3bd41d commit b2bfb5e

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

stackslib/src/net/api/tests/getclaritymarfvalue.rs

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515

1616
use std::net::{IpAddr, Ipv4Addr, SocketAddr};
1717

18+
use clarity::vm::database::{ClarityDeserializable, STXBalance};
1819
use clarity::vm::types::{QualifiedContractIdentifier, StacksAddressExtensions, TypeSignature};
1920
use clarity::vm::{ClarityName, ContractName, Value};
2021
use stacks_common::codec::StacksMessageCodec;
@@ -149,7 +150,7 @@ fn test_try_make_response() {
149150
Some(1)
150151
);
151152

152-
let resp = response.decode_data_var_response().unwrap();
153+
let resp = response.decode_clarity_marf_response().unwrap();
153154
assert_eq!(resp.data, "0x0000000000000000000000000000000000");
154155
assert!(resp.marf_proof.is_some());
155156

@@ -165,7 +166,7 @@ fn test_try_make_response() {
165166
Some(1)
166167
);
167168

168-
let resp = response.decode_data_var_response().unwrap();
169+
let resp = response.decode_clarity_marf_response().unwrap();
169170
assert_eq!(resp.data, "0x0100000000000000000000000000000001");
170171
assert!(resp.marf_proof.is_some());
171172

@@ -189,15 +190,16 @@ fn test_try_make_response() {
189190
let (preamble, body) = response.destruct();
190191
assert_eq!(preamble.status_code, 404);
191192

192-
// vm-account blaance
193+
// vm-account balance
193194
let response = responses.remove(0);
194195
debug!(
195196
"Response:\n{}\n",
196197
std::str::from_utf8(&response.try_serialize().unwrap()).unwrap()
197198
);
198199

199-
let resp = response.decode_data_var_response().unwrap();
200-
let balance = Value::try_deserialize_hex(&resp.data[2..], &TypeSignature::IntType, false);
201-
assert_eq!(balance, Ok(Value::Int(256_000_000_000)));
202-
assert!(resp.marf_proof.is_some());
200+
let resp = response.decode_clarity_marf_response().unwrap();
201+
let balance = STXBalance::deserialize(&resp.data[2..]).unwrap();
202+
203+
assert_eq!(balance.amount_unlocked(), 1_000_000_000);
204+
assert_eq!(balance.amount_locked(), 0);
203205
}

0 commit comments

Comments
 (0)