xtask: generate katana test db

.cargo/config.toml

@@ -0,0 +1,2 @@
+[alias]
+generate-test-db = "run --manifest-path ./xtask/generate-test-db/Cargo.toml --"

Cargo.toml

@@ -42,6 +42,7 @@ members = [
 	"crates/torii/server",
 	"crates/torii/types-test",
 	"examples/spawn-and-move",
+	"xtask/generate-test-db",
 ]
 
 [workspace.package]

crates/katana/runner/src/lib.rs

@@ -45,6 +45,8 @@
     pub log_path: Option<PathBuf>,
     /// The messaging config file
     pub messaging: Option<String>,
+    /// The path to the database dir.
+    pub db_dir: Option<PathBuf>,
 }
 
 impl Default for KatanaRunnerConfig {
@@ -58,6 +60,7 @@
             run_name: None,
             log_path: None,
             messaging: None,
+            db_dir: None,
         }
     }
 }
@@ -99,6 +102,10 @@
             builder = builder.messaging(messaging_file);
         }
 
+        if let Some(path) = config.db_dir {
+            builder = builder.db_dir(path);
+        }
+
         let mut katana = builder.spawn();
 
         let stdout =

xtask/generate-test-db/Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+edition.workspace = true
+license.workspace = true
+name = "xtask-generate-test-db"
+repository.workspace = true
+version.workspace = true
+
+[dependencies]
+dojo-test-utils.workspace = true
+dojo-world = { workspace = true, features = [ "migration" ] }
+katana-runner.workspace = true
+scarb.workspace = true
+sozo-ops = { workspace = true, features = [ "test-utils" ] }
+tokio.workspace = true

xtask/generate-test-db/src/main.rs

@@ -0,0 +1,54 @@
+use std::path::PathBuf;
+use std::process::Command;
+
+use dojo_test_utils::compiler::CompilerTestSetup;
+use dojo_world::migration::TxnConfig;
+use katana_runner::{KatanaRunner, KatanaRunnerConfig};
+use scarb::compiler::Profile;
+use sozo_ops::migration::{self, MigrationOutput};
+use sozo_ops::test_utils;
+
+type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;
+
+async fn migrate(runner: &KatanaRunner) -> Result<MigrationOutput> {
+    // migrate the example project
+    let acc = runner.account(0);
+
+    // setup scarb workspace
+    let setup = CompilerTestSetup::from_examples("crates/dojo-core", "examples/");
+    let cfg = setup.build_test_config("spawn-and-move", Profile::DEV);
+    let ws = scarb::ops::read_workspace(cfg.manifest_path(), &cfg)?;
+
+    // migrate the example project
+    let (strat, _) = test_utils::setup::setup_migration(&cfg)?;
+    let output = migration::execute_strategy(&ws, &strat, &acc, TxnConfig::init_wait()).await?;
+
+    Ok(output)
+}
+
+#[tokio::main]
+async fn main() -> Result<()> {
+    let db_path = PathBuf::from("test-db");
+    let compressed_path = "test-db.tar.gz";
+
+    // Instantiate Katana with db-dir at ./test-db
+    let cfg = KatanaRunnerConfig { db_dir: Some(db_path.clone()), ..Default::default() };
+    let runner = KatanaRunner::new_with_config(cfg)?;
+
+    let _ = migrate(&runner).await?;
+    drop(runner);
+
+    // ensure the test-db directory have been created
+    assert!(db_path.exists(), "test-db directory does not exist");
+
+    // Compress the ./test-db directory using tar
+    Command::new("tar")
+        .args(&["-czf", compressed_path, "-C", ".", "test-db"])
+        .status()
+        .expect("Failed to compress test-db directory");
+
+    // ensure the test-db directory have been created
+    assert!(PathBuf::from(db_path).exists(), "test-db directory does not exist");
+
+    Ok(())
+}