cleanup

Author: Zoran Cvetkov

graph/src/data/store/mod.rs

@@ -913,10 +913,12 @@ impl Entity {
         Id::try_from(self.get("id").unwrap().clone()).expect("the id is set to a valid value")
     }
 
-    // TODO: only for tests!
+    /// Convenience methods to modify individual attributes for tests.
+    /// Production code should not use/need this.
+    #[cfg(debug_assertions)]
     pub fn vid(&self) -> i64 {
         self.get("vid")
-            .expect("the vid is set")
+            .unwrap_or(&Value::Int8(0))
             .as_int8()
             .expect("the vid is set to a valid value")
     }

store/test-store/tests/core/interfaces.rs

@@ -1114,11 +1114,11 @@ async fn enums() {
     let entities = vec![
         (
             "Trajectory",
-            entity! { schema =>  id: "1", direction: "EAST", meters: 10, vid: 0i64},
+            entity! { schema =>  id: "1", direction: "EAST", meters: 10, vid: 0i64 },
         ),
         (
             "Trajectory",
-            entity! { schema =>  id: "2", direction: "NORTH", meters: 15, vid: 1i64},
+            entity! { schema =>  id: "2", direction: "NORTH", meters: 15, vid: 1i64 },
         ),
     ];
     let query = "query { trajectories { id, direction, meters } }";
@@ -1365,11 +1365,11 @@ async fn derived_interface_bytes() {
         ("Pool", entity! { schema =>  id: b("0xf001"), vid: 0i64 }),
         (
             "Sell",
-            entity! { schema =>  id: b("0xc0"), pool: "0xf001", vid: 0i64},
+            entity! { schema =>  id: b("0xc0"), pool: "0xf001", vid: 0i64 },
         ),
         (
             "Buy",
-            entity! { schema =>  id: b("0xb0"), pool: "0xf001", vid: 0i64},
+            entity! { schema =>  id: b("0xb0"), pool: "0xf001", vid: 0i64 },
         ),
     ];