Added the pool amount for long (#638)

* added pool amount * fmt check
keep-starknet-strange · Apr 1, 2024 · 1021dee · 1021dee
1 parent 18dd623
commit 1021dee
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/tests/integration/test_long_integration.cairo b/tests/integration/test_long_integration.cairo
@@ -120,9 +120,6 @@ fn test_long_market_integration() {
     // TODO Check why we don't need to set pool_amount_key
     // // Set pool amount in data_store.
     // let mut key = keys::pool_amount_key(market.market_token, contract_address_const::<'ETH'>());
-    // data_store.set_u256(key, 50000000000);
-    // key = keys::pool_amount_key(market.market_token, contract_address_const::<'USDC'>());
-    // data_store.set_u256(key, 50000000000);
 
     // Send token to deposit in the deposit vault (this should be in a multi call with create_deposit)
     IERC20Dispatcher { contract_address: market.long_token }
@@ -285,6 +282,17 @@ fn test_long_market_integration() {
     // data_store.set_u256(keys::pool_amount_key(market.market_token, contract_address_const::<'ETH'>()), 1000000);
     // Execute the swap order.
 
+    data_store
+        .set_u256(
+            keys::pool_amount_key(market.market_token, contract_address_const::<'USDC'>()),
+            10000000000
+        );
+    data_store
+        .set_u256(
+            keys::pool_amount_key(market.market_token, contract_address_const::<'ETH'>()),
+            10000000000
+        );
+
     let signatures: Span<felt252> = array![0].span();
     let set_price_params = SetPricesParams {
         signer_info: 2,