fix: updated ado mock

andromedaprotocol · Aug 7, 2024 · 53e2818 · 53e2818
1 parent ab455bb
commit 53e2818
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/contracts/data-storage/andromeda-graph/src/mock.rs b/contracts/data-storage/andromeda-graph/src/mock.rs
@@ -4,6 +4,7 @@ use andromeda_data_storage::graph::{
     Coordinate, GetAllPointsResponse, GetMapInfoResponse, GetMaxPointResponse, MapInfo,
 };
 use andromeda_data_storage::graph::{ExecuteMsg, InstantiateMsg, QueryMsg};
+use andromeda_std::error::ContractError;
 use andromeda_testing::mock::MockApp;
 use andromeda_testing::{
     mock_ado,
@@ -58,9 +59,13 @@ impl MockGraph {
         app: &mut MockApp,
         sender: Addr,
         coordinate: Coordinate,
+        is_timestamp_allowed: bool,
         funds: Option<Coin>,
     ) -> ExecuteResult {
-        let msg = ExecuteMsg::StoreCoordinate { coordinate };
+        let msg = ExecuteMsg::StoreCoordinate {
+            coordinate,
+            is_timestamp_allowed,
+        };
         if let Some(funds) = funds {
             app.execute_contract(sender, self.addr().clone(), &msg, &[funds])
         } else {