Orderbook mock

crates/driver/src/domain/competition/order/app_data.rs

@@ -66,7 +66,7 @@ impl AppDataRetriever {
                 let url = self_
                     .0
                     .base_url
-                    .join(&format!("v1/app_data/{:?}", app_data))?;
+                    .join(&format!("v1/app_data/{:?}", app_data.0))?;
                 let response = self_.0.client.get(url).send().await?;
 
                 let validated_app_data = match response.status() {

crates/driver/src/infra/cli.rs

@@ -21,6 +21,10 @@ pub struct Args {
     #[clap(long, env)]
     pub ethrpc: Url,
 
+    /// The orderbook API base URL.
+    #[clap(long, env)]
+    pub orderbook_url: Url,
+
     /// Path to the driver configuration file. This file should be in TOML
     /// format. For an example see
     /// https://github.com/cowprotocol/services/blob/main/crates/driver/example.toml.

crates/driver/src/infra/config/file/load.rs

@@ -376,6 +376,5 @@ pub async fn load(chain: Chain, path: &Path) -> infra::Config {
         order_priority_strategies: config.order_priority_strategies,
         archive_node_url: config.archive_node_url,
         simulation_bad_token_max_age: config.simulation_bad_token_max_age,
-        orderbook_url: config.orderbook_url,
     }
 }

crates/driver/src/infra/config/file/mod.rs

@@ -73,9 +73,6 @@ struct Config {
         default = "default_simulation_bad_token_max_age"
     )]
     simulation_bad_token_max_age: Duration,
-
-    /// URL of the orderbook API to fetch order's app-data from.
-    orderbook_url: Url,
 }
 
 #[serde_as]

crates/driver/src/infra/config/mod.rs

@@ -30,5 +30,4 @@ pub struct Config {
     pub order_priority_strategies: Vec<OrderPriorityStrategy>,
     pub archive_node_url: Option<Url>,
     pub simulation_bad_token_max_age: Duration,
-    pub orderbook_url: Url,
 }

crates/driver/src/run.rs

@@ -79,7 +79,7 @@ async fn run_with(args: cli::Args, addr_sender: Option<oneshot::Sender<SocketAdd
             let _ = shutdown_receiver.await;
         },
         config.order_priority_strategies,
-        config.orderbook_url,
+        args.orderbook_url,
     );
 
     futures::pin_mut!(serve);

crates/driver/src/tests/setup/driver.rs

@@ -3,7 +3,10 @@ use {
     crate::{
         domain::competition::order,
         infra::config::file::OrderPriorityStrategy,
-        tests::{hex_address, setup::blockchain::Trade},
+        tests::{
+            hex_address,
+            setup::{blockchain::Trade, orderbook::Orderbook},
+        },
     },
     rand::seq::SliceRandom,
     serde_json::json,
@@ -31,6 +34,7 @@ impl Driver {
         config: &Config,
         solvers: &Vec<(Solver, SocketAddr)>,
         blockchain: &Blockchain,
+        orderbook: Orderbook,
     ) -> Self {
         let (config_file, config_temp_path) = match config.config_file.as_ref() {
             Some(config_file) => (config_file.to_owned(), None),
@@ -46,6 +50,8 @@ impl Driver {
             "0.0.0.0:0".to_owned(),
             "--ethrpc".to_owned(),
             blockchain.web3_url.clone(),
+            "--orderbook-url".to_owned(),
+            format!("http://{:?}", orderbook.addr).to_owned(),
             "--config".to_owned(),
             config_file.to_str().unwrap().to_owned(),
         ];

crates/driver/src/tests/setup/mod.rs

@@ -28,7 +28,10 @@ use {
                 ETH_ORDER_AMOUNT,
             },
             hex_address,
-            setup::blockchain::{Blockchain, Interaction, Trade},
+            setup::{
+                blockchain::{Blockchain, Interaction, Trade},
+                orderbook::Orderbook,
+            },
         },
     },
     app_data::AppDataHash,
@@ -51,6 +54,7 @@ use {
 pub mod blockchain;
 mod driver;
 pub mod fee;
+mod orderbook;
 mod solver;
 
 #[derive(Debug, Clone, Copy)]
@@ -957,6 +961,7 @@ impl Setup {
             (solver.clone(), instance.addr)
         }))
         .await;
+        let orderbook = Orderbook::start();
         let driver = Driver::new(
             &driver::Config {
                 config_file,
@@ -966,6 +971,7 @@ impl Setup {
             },
             &solvers_with_address,
             &blockchain,
+            orderbook,
         )
         .await;
 

crates/driver/src/tests/setup/orderbook.rs

@@ -0,0 +1,28 @@
+use {
+    axum::{extract::Path, http::StatusCode, routing::get, Router},
+    std::net::SocketAddr,
+};
+
+pub struct Orderbook {
+    pub addr: SocketAddr,
+}
+
+impl Orderbook {
+    pub fn start() -> Self {
+        let app = Router::new().route("/v1/app_data/:app_data", get(Self::mock_handler));
+        let server =
+            axum::Server::bind(&"0.0.0.0:0".parse().unwrap()).serve(app.into_make_service());
+        let addr = server.local_addr();
+        println!("Orderbook mock server listening on {}", addr);
+
+        tokio::spawn(server);
+
+        Orderbook { addr }
+    }
+
+    /// Default mock handler that always returns 404 Not Found.
+    async fn mock_handler(Path(app_data): Path<String>) -> StatusCode {
+        println!("Received app_data request: {}", app_data);
+        StatusCode::NOT_FOUND
+    }
+}