fix graft test

Author: Zoran Cvetkov

store/postgres/src/relational_queries.rs

@@ -4963,6 +4963,7 @@ impl<'a> QueryFragment<Pg> for CopyEntityBatchQuery<'a> {
             out.push_sql(", ");
             out.push_sql(CAUSALITY_REGION_COLUMN);
         };
+        out.push_sql(", vid");
 
         out.push_sql(")\nselect ");
         for column in &self.columns {
@@ -5028,6 +5029,8 @@ impl<'a> QueryFragment<Pg> for CopyEntityBatchQuery<'a> {
                 ));
             }
         }
+        out.push_sql(", vid");
+
         out.push_sql(" from ");
         out.push_sql(self.src.qualified_name.as_str());
         out.push_sql(" where vid >= ");

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

@@ -330,6 +330,7 @@ async fn check_graft(
 
     // Make our own entries for block 2
     shaq.set("email", "[email protected]").unwrap();
+    shaq.set("vid", 5i64).unwrap();
     let op = EntityOperation::Set {
         key: user_type.parse_key("3").unwrap(),
         data: shaq,