Fix integration tests

graphprotocol · Jul 22, 2024 · 4f9938a · 4f9938a
1 parent 970df8f
commit 4f9938a
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/graph/src/blockchain/block_stream.rs b/graph/src/blockchain/block_stream.rs
@@ -360,27 +360,24 @@ fn create_mock_trigger<C: Blockchain>() -> Trigger<C> {
 
 fn create_mock_entity() -> Entity {
     let schema = InputSchema::parse_latest(
-        "type User @entity { id: String!, val: String! }",
+        "type Block @entity { id: Bytes!, number: BigInt!, hash: Bytes! }",
         DeploymentHash::new("test").unwrap(),
     )
     .unwrap();
 
+    let hash = Value::Bytes(
+        Bytes::from_str("0xd66ea6a52c13884f2a57596e09760905a0cbd3b8ad84af8bb213ad77d79149d0")
+            .unwrap(),
+    );
+
     schema
         .make_entity(vec![
-            ("id".into(), Value::String("id".to_owned())),
+            ("id".into(), hash.clone()),
             (
                 "number".into(),
                 Value::BigInt(BigInt::from_str("54321").unwrap()),
             ),
-            (
-                "hash".into(),
-                Value::Bytes(
-                    Bytes::from_str(
-                        "0xd66ea6a52c13884f2a57596e09760905a0cbd3b8ad84af8bb213ad77d79149d0",
-                    )
-                    .unwrap(),
-                ),
-            ),
+            ("hash".into(), hash),
         ])
         .unwrap()
 }