tests: fix subgraph datasource test failing

Author: incrypto32

tests/integration-tests/subgraph-data-sources/package.json

@@ -7,7 +7,7 @@
     "deploy:test": "graph deploy test/subgraph-data-sources --version-label v0.0.1 --ipfs $IPFS_URI --node $GRAPH_NODE_ADMIN_URI"
   },
   "devDependencies": {
-    "@graphprotocol/graph-cli": "0.91.0-alpha-20241129215038-b75cda9",
+    "@graphprotocol/graph-cli": "0.93.4-alpha-20250105163501-f401d0c57c4ba1f1af95a928d447efd63a56ecdc",
     "@graphprotocol/graph-ts": "0.36.0-alpha-20241129215038-b75cda9"
   }
 }

tests/integration-tests/subgraph-data-sources/src/mapping.ts

@@ -1,35 +1,23 @@
-import { Entity, log, store } from '@graphprotocol/graph-ts';
+import { Entity, log, store, BigInt, EntityTrigger, EntityOp } from '@graphprotocol/graph-ts';
+import { Block } from '../generated/subgraph-Qmaqf8cRxfxbduZppSHKG9DMuX5JZPMoGuwGb2DQuo48sq';
 import { MirrorBlock } from '../generated/schema';
 
-export class EntityTrigger {
-  constructor(
-    public entityOp: u32,
-    public entityType: string,
-    public entity: Entity,
-    public vid: i64,
-  ) {}
-}
-
-export function handleEntity(trigger: EntityTrigger): void {
-  let blockEntity = trigger.entity;
-  let blockNumber = blockEntity.getBigInt('number');
-  let blockHash = blockEntity.getBytes('hash');
-  let testMessage = blockEntity.get('testMessage');
-  let id = blockEntity.getString('id');
+export function handleEntity(trigger: EntityTrigger<Block>): void {
+  let blockEntity = trigger.data;
+  let id = blockEntity.id;
 
-  log.info('Block number: {}', [blockNumber.toString()]);
-
-  if (trigger.entityOp == 2) {
+  if (trigger.operation === EntityOp.Remove) {
     log.info('Removing block entity with id: {}', [id]);
     store.remove('MirrorBlock', id);
     return;
   }
 
   let block = loadOrCreateMirrorBlock(id);
-  block.number = blockNumber;
-  block.hash = blockHash;
-  if (testMessage) {
-    block.testMessage = testMessage.toString();
+  block.number = blockEntity.number;
+  block.hash = blockEntity.hash;
+  
+  if (blockEntity.testMessage) {
+    block.testMessage = blockEntity.testMessage;
   }
 
   block.save();
@@ -41,6 +29,5 @@ export function loadOrCreateMirrorBlock(id: string): MirrorBlock {
     log.info('Creating new block entity with id: {}', [id]);
     block = new MirrorBlock(id);
   }
-
   return block;
 }

tests/integration-tests/subgraph-data-sources/subgraph.yaml

@@ -6,7 +6,7 @@ dataSources:
     name: Contract
     network: test
     source:
-      address: 'Qmaqf8cRxfxbduZppSHKG9DMuX5JZPMoGuwGb2DQuo48sq'
+      address: 'QmQCiSBooPHZTrqxJknG9W1YJ8EZXj8c8baFs24eY9pXM8'
       startBlock: 0
     mapping:
       apiVersion: 0.0.7