fix set prices in tests

keep-starknet-strange · May 4, 2024 · 756a3d5 · 756a3d5
1 parent 45ed156
commit 756a3d5
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 9 deletions.
diff --git a/tests/integration/swap_test.cairo b/tests/integration/swap_test.cairo
@@ -99,7 +99,9 @@ fn test_deposit_market_integration() {
             keys::max_pool_amount_key(market.market_token, market.short_token), 500000000000000000
         );
 
-    oracle.set_price_testing_eth(5000);
+    oracle.set_primary_prices(market.long_token, 5000);
+    oracle.set_primary_prices(market.short_token, 1);
+
     // Fill the pool.
     IERC20Dispatcher { contract_address: market.long_token }.mint(market.market_token, 50000000000);
     IERC20Dispatcher { contract_address: market.short_token }
@@ -427,7 +429,8 @@ fn test_swap_18_deposit_market_integration() {
             2500000000000000000000000000000000000000000000
         );
 
-    oracle.set_price_testing_eth(5000);
+    oracle.set_primary_prices(market.long_token, 5000);
+    oracle.set_primary_prices(market.short_token, 1);
 
     'fill the pool'.print();
     // Fill the pool.

diff --git a/tests/integration/test_deposit_withdrawal.cairo b/tests/integration/test_deposit_withdrawal.cairo
@@ -507,7 +507,8 @@ fn test_deposit_market_integration() {
             keys::max_pool_amount_key(market.market_token, market.short_token), 500000000000000000
         );
 
-    oracle.set_price_testing_eth(5000);
+    oracle.set_primary_prices(market.long_token, 5000);
+    oracle.set_primary_prices(market.short_token, 1);
 
     // Fill the pool.
     IERC20Dispatcher { contract_address: market.long_token }.mint(market.market_token, 50000000000);
@@ -685,7 +686,8 @@ fn test_deposit_withdraw_integration() {
             keys::max_pool_amount_key(market.market_token, market.short_token), 500000000000000000
         );
 
-    oracle.set_price_testing_eth(5000);
+    oracle.set_primary_prices(market.long_token, 5000);
+    oracle.set_primary_prices(market.short_token, 1);
 
     // Fill the pool.
     IERC20Dispatcher { contract_address: market.long_token }.mint(market.market_token, 50000000000);

diff --git a/tests/integration/test_long_integration.cairo b/tests/integration/test_long_integration.cairo
@@ -112,7 +112,7 @@ const INITIAL_TOKENS_MINTED: felt252 = 1000;
 //             keys::max_pool_amount_key(market.market_token, market.short_token), 500000000000000000
 //         );
 
-//     oracle.set_price_testing_eth(5000);
+//     oracle.set_primary_prices(market.long_token, 5000);
 
 //     // Fill the pool.
 //     IERC20Dispatcher { contract_address: market.long_token }.mint(market.market_token, 50000000000);
@@ -382,7 +382,8 @@ const INITIAL_TOKENS_MINTED: felt252 = 1000;
 //         .balance_of(caller_address);
 //     'balance of mkt before'.print();
 //     balance_of_mkt_before.print();
-//     oracle.set_price_testing_eth(6000);
+//     oracle.set_primary_prices(market.long_token, 6000);
+
 
 //     start_prank(market.market_token, caller_address);
 //     start_prank(market.long_token, caller_address);
@@ -527,7 +528,8 @@ fn test_long_decimals_market_integration() {
             50000000000000000000000000000000000000000000000
         );
 
-    oracle.set_price_testing_eth(5000);
+    oracle.set_primary_prices(market.long_token, 5000);
+    oracle.set_primary_prices(market.short_token, 1);
 
     'fill the pool'.print();
     // Fill the pool.
@@ -788,7 +790,7 @@ fn test_long_decimals_market_integration() {
     'size in usd'.print();
     first_position.size_in_usd.print();
     'OKAAAAAYYYYYY'.print();
-    oracle.set_price_testing_eth(6000);
+    oracle.set_primary_prices(market.long_token, 6000);
     let first_position_after_pump = data_store.get_position(position_key_1);
     'size tokens after pump'.print();
     first_position_after_pump.size_in_tokens.print();

diff --git a/tests/integration/test_short_integration.cairo b/tests/integration/test_short_integration.cairo
@@ -113,7 +113,8 @@ fn test_short_market_integration() {
             keys::max_pool_amount_key(market.market_token, market.short_token), 500000000000000000
         );
 
-    oracle.set_price_testing_eth(5000);
+    oracle.set_primary_prices(market.long_token, 5000);
+    oracle.set_primary_prices(market.short_token, 1);
 
     // Fill the pool.
     IERC20Dispatcher { contract_address: market.long_token }.mint(market.market_token, 50000000000);