From cf3ab35144b779834aaab777117ac040343e5868 Mon Sep 17 00:00:00 2001
From: davidemazzocchi <davide.mazzocchi@gmail.com>
Date: Wed, 1 Oct 2025 07:56:00 +0200
Subject: [PATCH] fix: correct intended use for entity state

---
 01.workspace/heave/src/tst/intended_use.rs | 17 ++++++-----------
 1 file changed, 6 insertions(+), 11 deletions(-)

diff --git a/01.workspace/heave/src/tst/intended_use.rs b/01.workspace/heave/src/tst/intended_use.rs
index ef0a93f..34b78cd 100644
--- a/01.workspace/heave/src/tst/intended_use.rs
+++ b/01.workspace/heave/src/tst/intended_use.rs
@@ -173,22 +173,17 @@ mod tests {
             .with_attribute("int", Value::SignedInt(50))
             .with_attribute("string", Value::Text("text".to_string()));
         let id = entity.id.clone();
+        let expected_entity = Entity {
+            state: EntityState::Loaded,
+            ..entity.clone()
+        };
         catalog.insert(entity);
         catalog.persist();
         // read the entity in a new catalog
         let mut catalog = Catalog::new(path);
         catalog.load_by_id(&id);
-        let entity: Option<Entity> = catalog.get(&id);
-        assert_eq!(
-            entity,
-            Some(
-                Entity::default()
-                    .with_id(&id)
-                    .with_class("test")
-                    .with_attribute("int", Value::SignedInt(50))
-                    .with_attribute("string", Value::Text("text".to_string()))
-            )
-        );
+        let entity: Entity = catalog.get(&id).unwrap();
+        assert_eq!(expected_entity, entity,);
     }
     #[test]
     fn check_011() {