cleanup

Author: Zoran Cvetkov

graph/src/components/store/write.rs

@@ -1017,30 +1017,29 @@ mod test {
 
             let value = value.clone();
             let key = THING_TYPE.parse_key("one").unwrap();
-            let vid = 0i64;
             match value {
                 Ins(block) => EntityModification::Insert {
                     key,
-                    data: Arc::new(entity! { SCHEMA => id: "one", count: block, vid: vid }),
+                    data: Arc::new(entity! { SCHEMA => id: "one", count: block }),
                     block,
                     end: None,
                 },
                 Ovw(block) => EntityModification::Overwrite {
                     key,
-                    data: Arc::new(entity! { SCHEMA => id: "one", count: block, vid: vid }),
+                    data: Arc::new(entity! { SCHEMA => id: "one", count: block }),
                     block,
                     end: None,
                 },
                 Rem(block) => EntityModification::Remove { key, block },
                 InsC(block, end) => EntityModification::Insert {
                     key,
-                    data: Arc::new(entity! { SCHEMA => id: "one", count: block, vid: vid }),
+                    data: Arc::new(entity! { SCHEMA => id: "one", count: block }),
                     block,
                     end: Some(end),
                 },
                 OvwC(block, end) => EntityModification::Overwrite {
                     key,
-                    data: Arc::new(entity! { SCHEMA => id: "one", count: block, vid: vid }),
+                    data: Arc::new(entity! { SCHEMA => id: "one", count: block }),
                     block,
                     end: Some(end),
                 },

graph/src/data/store/mod.rs

@@ -924,7 +924,7 @@ impl Entity {
         self.0.insert("vid", value.into())
     }
 
-    /// Sets the VID if not set. Should be used only for tests.
+    /// Sets the VID if it's not already set. Should be used only for tests.
     #[cfg(debug_assertions)]
     pub fn set_vid_if_empty(&mut self) {
         let vid = self.get("vid");
@@ -948,7 +948,7 @@ impl Entity {
     /// If a key only exists on one entity, the value from that entity is chosen.
     /// If a key is set to `Value::Null` in `update`, the key/value pair is removed.
     pub fn merge_remove_null_fields(&mut self, update: Entity) -> Result<(), InternError> {
-        for (key, value) in update.into_iter() {
+        for (key, value) in update.0.into_iter() {
             match value {
                 Value::Null => self.0.remove(&key),
                 _ => self.0.insert(&key, value)?,

store/postgres/src/relational_queries.rs

@@ -540,7 +540,7 @@ impl EntityData {
                     // block_range that `select *` pulls in but that we
                     // don't care about here
                     if key == "vid" {
-                        // VID is not in the input schema but we need it so deserialize it too
+                        // VID is not in the input schema but we need it, so deserialize it too
                         match T::Value::from_column_value(&ColumnType::Int8, json) {
                             Ok(value) if value.is_null() => None,
                             Ok(value) => Some(Ok((Word::from("vid"), value))),

store/test-store/tests/chain/ethereum/manifest.rs

@@ -416,7 +416,7 @@ specVersion: 0.0.2
             msg
         );
 
-        let thing = entity! { schema => id: "datthing", vid: 1i64 };
+        let thing = entity! { schema => id: "datthing" };
         test_store::insert_entities(
             &deployment,
             vec![(schema.entity_type("Thing").unwrap(), thing)],

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

@@ -201,9 +201,9 @@ async fn reference_interface_derived() {
 
     let query = "query { events { id transaction { id } } }";
 
-    let buy = ("BuyEvent", entity! { schema => id: "buy", vid: 0i64 });
-    let sell1 = ("SellEvent", entity! { schema => id: "sell1", vid: 0i64 });
-    let sell2 = ("SellEvent", entity! { schema => id: "sell2", vid: 1i64 });
+    let buy = ("BuyEvent", entity! { schema => id: "buy" });
+    let sell1 = ("SellEvent", entity! { schema => id: "sell1" });
+    let sell2 = ("SellEvent", entity! { schema => id: "sell2" });
     let gift = (
         "GiftEvent",
         entity! { schema => id: "gift", transaction: "txn" },

store/test-store/tests/postgres/aggregation.rs

@@ -82,7 +82,7 @@ pub async fn insert(
         .map(|mut data| {
             let data_type = schema.entity_type("Data").unwrap();
             let key = data_type.key(data.id());
-            let _ = data.set_vid_if_empty();
+            data.set_vid_if_empty();
             EntityOperation::Set { data, key }
         })
         .collect();

store/test-store/tests/postgres/relational_bytes.rs

@@ -288,7 +288,7 @@ fn find_many() {
         const ID2: &str = "0xdeadbeef02";
         const NAME2: &str = "Moo";
         insert_thing(&mut conn, layout, ID, NAME, 0);
-        insert_thing(&mut conn, layout, ID2, NAME2, 1);
+        insert_thing(&mut conn, layout, ID2, NAME2, 0);
 
         let mut id_map = BTreeMap::default();
         let ids = IdList::try_from_iter(
@@ -336,7 +336,7 @@ fn update() {
             .expect("Failed to read Thing[deadbeef]")
             .unwrap();
 
-        assert_entity_eq!(scrub(&entity), actual);
+        assert_entity_eq!(entity, actual);
     });
 }