test: add unit test for AssetTransferredToDestination

Author: gleiser-oliveira

subgraphs/protocol-reserve/tests/TokenConverter/events.ts

@@ -2,6 +2,7 @@ import { Address, BigInt, ethereum } from '@graphprotocol/graph-ts';
 import { newMockEvent } from 'matchstick-as';
 
 import {
+  AssetTransferredToDestination,
   BaseAssetUpdated,
   ConvertedExactTokens,
   ConversionConfigUpdated,
@@ -195,3 +196,35 @@ export const createConvertedEvent = (
 
   return event;
 };
+
+export const createAssetTransferredToDestinationEvent = (
+  tokenConverterAddress: Address,
+  receiver: Address,
+  comptroller: Address,
+  asset: Address,
+  amount: string,
+): AssetTransferredToDestination => {
+  const event = changetype<AssetTransferredToDestination>(newMockEvent());
+  event.address = tokenConverterAddress;
+  event.parameters = [];
+
+  const receiverParam = new ethereum.EventParam('receiver', ethereum.Value.fromAddress(receiver));
+  event.parameters.push(receiverParam);
+
+  const comptrollerParam = new ethereum.EventParam(
+    'comptroller',
+    ethereum.Value.fromAddress(comptroller),
+  );
+  event.parameters.push(comptrollerParam);
+
+  const assetParam = new ethereum.EventParam('asset', ethereum.Value.fromAddress(asset));
+  event.parameters.push(assetParam);
+
+  const amountParam = new ethereum.EventParam(
+    'amount',
+    ethereum.Value.fromUnsignedBigInt(BigInt.fromString(amount)),
+  );
+  event.parameters.push(amountParam);
+
+  return event;
+};

subgraphs/protocol-reserve/tests/TokenConverter/index.test.ts

@@ -3,6 +3,7 @@ import { assert, beforeAll, describe, test } from 'matchstick-as/assembly/index'
 
 import { TokenConverter } from '../../generated/schema';
 import {
+  handleAssetTransferredToDestination,
   handleBaseAssetUpdated,
   handleConversionConfigUpdated,
   handleConversionEvent,
@@ -20,6 +21,7 @@ import {
   createConverterNetworkAddressUpdatedEvent,
   createConvertedEvent,
   createDestinationAddressUpdatedEvent,
+  createAssetTransferredToDestinationEvent,
 } from './events';
 import { createTokenConverterMock, createTokenMock } from './mocks';
 
@@ -197,4 +199,27 @@ describe('Token Converter', () => {
       tokenConverter2Address,
     );
   });
+
+  test('should handle an AssetTransferredToDestination event', () => {
+    handleAssetTransferredToDestination(
+      createAssetTransferredToDestinationEvent(
+        tokenConverter2Address,
+        destination1Address,
+        Address.fromString('0x0000000000000000000000000000000000000bca'),
+        token2Address,
+        '12345',
+      ),
+    );
+
+    const tokenConverter = TokenConverter.load(getTokenConverterId(tokenConverter2Address))!;
+    const destinationAmounts = tokenConverter.destinationAmounts.load();
+    assert.i32Equals(destinationAmounts.length, 2);
+    assert.bigIntEquals(destinationAmounts[1].amount, BigInt.fromString('12345'));
+    assert.addressEquals(Address.fromBytes(destinationAmounts[1].token), token2Address);
+    assert.addressEquals(Address.fromBytes(destinationAmounts[1].address), destination1Address);
+    assert.addressEquals(
+      Address.fromBytes(destinationAmounts[1].tokenConverter),
+      tokenConverter2Address,
+    );
+  });
 });