Added run limit order test

osmosis-labs · Feb 21, 2024 · 60f9b78 · 60f9b78
1 parent febf5a6
commit 60f9b78
Show file tree

Hide file tree

Showing 3 changed files with 760 additions and 2 deletions.
diff --git a/contracts/orderbook/src/error.rs b/contracts/orderbook/src/error.rs
@@ -49,4 +49,7 @@ pub enum ContractError {
         amount_remaining: Uint128,
         reason: Option<String>,
     },
+
+    #[error("Mismatched order direction")]
+    MismatchedOrderDirection {},
 }
diff --git a/contracts/orderbook/src/order.rs b/contracts/orderbook/src/order.rs
@@ -3,8 +3,8 @@ use crate::state::*;
 use crate::state::{MAX_TICK, MIN_TICK, ORDERBOOKS};
 use crate::types::{Fulfilment, LimitOrder, MarketOrder, OrderDirection, REPLY_ID_REFUND};
 use cosmwasm_std::{
-    coin, ensure, ensure_eq, BankMsg, Decimal, DepsMut, Env, MessageInfo, Order, Response, Storage,
-    SubMsg, Uint128,
+    coin, ensure, ensure_eq, ensure_ne, BankMsg, Decimal, DepsMut, Env, MessageInfo, Order,
+    Response, Storage, SubMsg, Uint128,
 };
 use cw_storage_plus::Bound;
 use cw_utils::{must_pay, nonpayable};
@@ -225,6 +225,11 @@ pub fn run_market_order(
 
         for maybe_current_order in tick_orders {
             let current_order = maybe_current_order?.1;
+            ensure_ne!(
+                current_order.order_direction,
+                order.order_direction,
+                ContractError::MismatchedOrderDirection {}
+            );
             let fill_quantity = order.quantity.min(current_order.quantity);
             // Add to total amount fulfiled from placed order
             amount_fulfiled = amount_fulfiled.checked_add(fill_quantity)?;