add list primitives test

unavi-xyz · Oct 11, 2024 · 0b4db59 · 0b4db59
1 parent 387d187
commit 0b4db59
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/crates/unavi-scripting/src/api/wired/scene/mesh.rs b/crates/unavi-scripting/src/api/wired/scene/mesh.rs
@@ -302,7 +302,7 @@ mod tests {
     }
 
     #[test]
-    fn test_create_remove_primitive() {
+    fn test_primitives() {
         let (mut app, mut data) = init_test_data();
 
         let node = HostNode::new(&mut data).unwrap();
@@ -321,6 +321,14 @@ mod tests {
 
         test_create_node_primitives(&mut app, &mut data, &node, &primitive);
 
+        let primitives =
+            HostMesh::list_primitives(&mut data, Resource::new_own(mesh.rep())).unwrap();
+        assert_eq!(primitives.len(), 1);
+        assert_eq!(
+            data.table.get(&primitives[0]).unwrap().read().id,
+            primitive_data.read().id
+        );
+
         HostMesh::remove_primitive(&mut data, mesh, primitive).unwrap();
 
         test_remove_node_primitives(&mut app, &mut data, &node, &primitive_data);

diff --git a/crates/unavi-scripting/src/api/wired/scene/primitive.rs b/crates/unavi-scripting/src/api/wired/scene/primitive.rs
@@ -9,7 +9,7 @@ use wasm_bridge::component::Resource;
 use crate::{
     api::{
         id::ResourceId,
-        wired::scene::bindings::mesh::{HostMesh, HostPrimitive, Material},
+        wired::scene::bindings::mesh::{HostPrimitive, Material},
     },
     data::ScriptData,
 };
@@ -211,7 +211,7 @@ impl HostPrimitive for ScriptData {
 mod tests {
     use crate::api::{
         tests::init_test_data,
-        wired::scene::bindings::{material::HostMaterial, node::HostNode},
+        wired::scene::bindings::{material::HostMaterial, mesh::HostMesh, node::HostNode},
     };
 
     use super::*;