feat: add support for model from other libraries

bin/sozo/src/commands/build.rs

@@ -214,8 +214,7 @@ fn create_stats_table(mut contracts_statistics: Vec<ContractStatistics>) -> Tabl
 
 #[cfg(test)]
 mod tests {
-    use camino::Utf8PathBuf;
-    use dojo_test_utils::compiler;
+    use dojo_test_utils::compiler::CompilerTestSetup;
     use prettytable::format::consts::FORMAT_NO_LINESEP_WITH_TITLE;
     use prettytable::{format, Cell, Row, Table};
     use scarb::compiler::Profile;
@@ -227,10 +226,9 @@ mod tests {
     // Uncomment once bindings support arrays.
     #[test]
     fn build_example_with_typescript_and_unity_bindings() {
-        let source_project_dir = Utf8PathBuf::from("../../examples/spawn-and-move/");
-        let dojo_core_path = Utf8PathBuf::from("../../crates/dojo-core");
+        let setup = CompilerTestSetup::from_examples("../../crates/dojo-core", "../../examples/");
 
-        let config = compiler::copy_tmp_config(&source_project_dir, &dojo_core_path, Profile::DEV);
+        let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
         let build_args = BuildArgs {
             bindings_output: "generated".to_string(),

bin/sozo/src/commands/clean.rs

@@ -91,7 +91,7 @@ impl CleanArgs {
 // which are normally built by the `build.rs` of `dojo-test-utils`.
 #[cfg(test)]
 mod tests {
-    use dojo_test_utils::compiler;
+    use dojo_test_utils::compiler::CompilerTestSetup;
     use dojo_world::manifest::DEPLOYMENT_DIR;
     use dojo_world::metadata::ABIS_DIR;
     use scarb::compiler::Profile;
@@ -100,16 +100,8 @@ mod tests {
 
     #[test]
     fn default_clean_works() {
-        let source_project = "../../examples/spawn-and-move";
-        let dojo_core_path = "../../crates/dojo-core";
-
-        let config = compiler::copy_tmp_config(
-            &Utf8PathBuf::from(source_project),
-            &Utf8PathBuf::from(dojo_core_path),
-            Profile::DEV,
-        );
-
-        println!("path {:?}", config.manifest_path());
+        let setup = CompilerTestSetup::from_examples("../../crates/dojo-core", "../../examples/");
+        let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
         let temp_project_dir = config.manifest_path().parent().unwrap().to_path_buf();
 
@@ -180,14 +172,9 @@ mod tests {
 
     #[test]
     fn all_profile_clean_works() {
-        let source_project = "../../examples/spawn-and-move";
-        let dojo_core_path = "../../crates/dojo-core";
+        let setup = CompilerTestSetup::from_examples("../../crates/dojo-core", "../../examples/");
 
-        let config = compiler::copy_tmp_config(
-            &Utf8PathBuf::from(source_project),
-            &Utf8PathBuf::from(dojo_core_path),
-            Profile::DEV,
-        );
+        let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
         let temp_project_dir = config.manifest_path().parent().unwrap().to_path_buf();
 

bin/sozo/src/commands/migrate.rs

@@ -88,36 +88,42 @@ impl MigrateArgs {
         })?;
 
         match self.command {
-            MigrateCommand::Plan => config.tokio_handle().block_on(async {
-                trace!(name, "Planning migration.");
-                migration::migrate(
-                    &ws,
-                    world_address,
-                    rpc_url,
-                    account,
-                    &name,
-                    true,
-                    TxnConfig::default(),
-                    dojo_metadata.skip_migration,
-                )
-                .await
-            }),
-            MigrateCommand::Apply { transaction } => config.tokio_handle().block_on(async {
-                trace!(name, "Applying migration.");
-                let txn_config: TxnConfig = transaction.into();
-
-                migration::migrate(
-                    &ws,
-                    world_address,
-                    rpc_url,
-                    account,
-                    &name,
-                    false,
-                    txn_config,
-                    dojo_metadata.skip_migration,
-                )
-                .await
-            }),
+            MigrateCommand::Plan => config
+                .tokio_handle()
+                .block_on(async {
+                    trace!(name, "Planning migration.");
+                    migration::migrate(
+                        &ws,
+                        world_address,
+                        rpc_url,
+                        account,
+                        &name,
+                        true,
+                        TxnConfig::default(),
+                        dojo_metadata.skip_migration,
+                    )
+                    .await
+                })
+                .map(|_| ()),
+            MigrateCommand::Apply { transaction } => config
+                .tokio_handle()
+                .block_on(async {
+                    trace!(name, "Applying migration.");
+                    let txn_config: TxnConfig = transaction.into();
+
+                    migration::migrate(
+                        &ws,
+                        world_address,
+                        rpc_url,
+                        account,
+                        &name,
+                        false,
+                        txn_config,
+                        dojo_metadata.skip_migration,
+                    )
+                    .await
+                })
+                .map(|_| ()),
         }
     }
 }

bin/sozo/tests/register_test.rs

@@ -1,7 +1,6 @@
 mod utils;
 
-use camino::Utf8PathBuf;
-use dojo_test_utils::compiler;
+use dojo_test_utils::compiler::CompilerTestSetup;
 use dojo_test_utils::migration::prepare_migration;
 use dojo_world::metadata::{dojo_metadata_from_workspace, get_default_namespace_from_ws};
 use dojo_world::migration::TxnConfig;
@@ -15,10 +14,8 @@ use utils::snapbox::get_snapbox;
 
 #[tokio::test(flavor = "multi_thread")]
 async fn reregister_models() {
-    let source_project_dir = Utf8PathBuf::from("../../examples/spawn-and-move/");
-    let dojo_core_path = Utf8PathBuf::from("../../crates/dojo-core");
-
-    let config = compiler::copy_tmp_config(&source_project_dir, &dojo_core_path, Profile::DEV);
+    let setup = CompilerTestSetup::from_examples("../../crates/dojo-core", "../../examples/");
+    let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
     let ws = ops::read_workspace(config.manifest_path(), &config)
         .unwrap_or_else(|op| panic!("Error building workspace: {op:?}"));
@@ -31,7 +28,7 @@ async fn reregister_models() {
     let default_namespace = get_default_namespace_from_ws(&ws).unwrap();
 
     let migration = prepare_migration(
-        source_project_dir.clone(),
+        config.manifest_path().parent().unwrap().into(),
         target_path,
         dojo_metadata.skip_migration,
         &default_namespace,

bin/sozo/tests/test_migrate.rs

@@ -1,7 +1,6 @@
 mod utils;
 
-use camino::Utf8PathBuf;
-use dojo_test_utils::compiler;
+use dojo_test_utils::compiler::CompilerTestSetup;
 use katana_runner::KatanaRunner;
 use scarb::compiler::Profile;
 use starknet::accounts::Account;
@@ -10,10 +9,8 @@ use utils::snapbox::get_snapbox;
 
 #[tokio::test(flavor = "multi_thread")]
 async fn migrate_dry_run() {
-    let source_project_dir = Utf8PathBuf::from("../../examples/spawn-and-move/");
-    let dojo_core_path = Utf8PathBuf::from("../../crates/dojo-core");
-
-    let config = compiler::copy_tmp_config(&source_project_dir, &dojo_core_path, Profile::DEV);
+    let setup = CompilerTestSetup::from_examples("../../crates/dojo-core", "../../examples/");
+    let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
     let sequencer = KatanaRunner::new().expect("Failed to start runner.");
 
@@ -45,7 +42,7 @@ async fn migrate_dry_run() {
 
     assert!(output.contains("Migration Strategy"));
     assert!(output.contains("# Base Contract"));
-    assert!(output.contains("# Models (8)"));
+    assert!(output.contains("# Models (10)"));
     assert!(output.contains("# World"));
     assert!(output.contains("# Contracts (4)"));
 }

crates/benches/contracts/Scarb.toml

@@ -19,3 +19,6 @@ initializer_class_hash = "0xbeef"
 # Default account for katana with seed = 0
 account_address = "0x6162896d1d7ab204c7ccac6dd5f8e9e7c25ecd5ae4fcb4ad32e57786bb46e03"
 private_key = "0x1800000000300000180000000000030000000000003006001800006600"
+
+[tool.dojo.world.namespace]
+default = "benches"

crates/dojo-bindgen/src/lib.rs

@@ -213,33 +213,32 @@ fn filter_model_tokens(tokens: &TokenizedAbi) -> TokenizedAbi {
 
 #[cfg(test)]
 mod tests {
-    use dojo_test_utils::compiler;
+    use dojo_test_utils::compiler::CompilerTestSetup;
     use dojo_world::metadata::dojo_metadata_from_workspace;
     use scarb::compiler::Profile;
 
     use super::*;
 
     #[test]
     fn gather_data_ok() {
-        let manifest_path = Utf8PathBuf::from("src/test_data/spawn-and-move/Scarb.toml");
-
-        let config = compiler::copy_tmp_config(
-            &Utf8PathBuf::from("../../examples/spawn-and-move"),
-            &Utf8PathBuf::from("../dojo-core"),
-            Profile::DEV,
-        );
+        let setup = CompilerTestSetup::from_examples("../dojo-core", "../../examples/");
+        let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
         let ws = scarb::ops::read_workspace(config.manifest_path(), &config).unwrap();
         let dojo_metadata = dojo_metadata_from_workspace(&ws).expect(
             "No current package with dojo metadata found, bindgen is not yet supported for \
              workspaces.",
         );
 
-        let data =
-            gather_dojo_data(&manifest_path, "dojo_example", "dev", dojo_metadata.skip_migration)
-                .unwrap();
-
-        assert_eq!(data.models.len(), 8);
+        let data = gather_dojo_data(
+            &config.manifest_path().to_path_buf(),
+            "dojo_example",
+            "dev",
+            dojo_metadata.skip_migration,
+        )
+        .unwrap();
+
+        assert_eq!(data.models.len(), 10);
 
         assert_eq!(data.world.name, "dojo_example");
 

crates/dojo-bindgen/src/plugins/typescript_v2/mod.rs

@@ -627,8 +627,7 @@ mod tests {
     use std::fs;
     use std::io::Read;
 
-    use camino::Utf8PathBuf;
-    use dojo_test_utils::compiler;
+    use dojo_test_utils::compiler::CompilerTestSetup;
     use dojo_world::metadata::dojo_metadata_from_workspace;
     use scarb::compiler::Profile;
 
@@ -647,21 +646,21 @@ mod tests {
         let expected_output_without_header =
             expected_output.lines().skip(1).collect::<Vec<&str>>().join("\n");
 
-        let manifest_path = Utf8PathBuf::from("src/test_data/spawn-and-move/Scarb.toml");
-        let config = compiler::copy_tmp_config(
-            &Utf8PathBuf::from("../../examples/spawn-and-move"),
-            &Utf8PathBuf::from("../dojo-core"),
-            Profile::DEV,
-        );
+        let setup = CompilerTestSetup::from_examples("../dojo-core", "../../examples/");
+        let config = setup.build_test_config("spawn-and-move", Profile::DEV);
 
         let ws = scarb::ops::read_workspace(config.manifest_path(), &config).unwrap();
         let dojo_metadata = dojo_metadata_from_workspace(&ws).expect(
             "No current package with dojo metadata found, bindgen is not yet support for \
              workspaces.",
         );
-        let data =
-            gather_dojo_data(&manifest_path, "dojo_examples", "dev", dojo_metadata.skip_migration)
-                .unwrap();
+        let data = gather_dojo_data(
+            &config.manifest_path().to_path_buf(),
+            "dojo_examples",
+            "dev",
+            dojo_metadata.skip_migration,
+        )
+        .unwrap();
 
         let actual_output = TypeScriptV2Plugin::generate_code_content(&data);
         let actual_output_without_header =

crates/dojo-core/Scarb.toml

@@ -7,3 +7,8 @@ version = "0.7.3"
 [dependencies]
 dojo_plugin = { git = "https://github.com/dojoengine/dojo", tag = "v0.7.3" }
 starknet = "=2.6.4"
+
+# Dojo core is tested with sozo, hence we need a namespace for the test
+# command to work.
+[tool.dojo.world]
+namespace = { default = "dojo" }

crates/dojo-core/src/world_test.cairo

@@ -45,7 +45,7 @@ struct Foo {
 }
 
 #[derive(Copy, Drop, Serde)]
-#[dojo::model(namespace: "another_namespace")]
+#[dojo::model(namespace: "another_namespace", nomapping: true)]
 struct Buzz {
     #[key]
     caller: ContractAddress,
@@ -940,7 +940,7 @@ trait IDojoInit<ContractState> {
 #[dojo::contract]
 mod test_contract {}
 
-#[dojo::contract(namespace: "buzz_namespace")]
+#[dojo::contract(namespace: "buzz_namespace", nomapping: true)]
 mod buzz_contract {}
 
 #[test]

crates/dojo-lang/Cargo.toml

@@ -55,6 +55,7 @@ tracing.workspace = true
 url.workspace = true
 
 [dev-dependencies]
+assert_fs.workspace = true
 cairo-lang-semantic.workspace = true
 cairo-lang-test-utils.workspace = true
 dojo-test-utils = { path = "../dojo-test-utils" }