fix: resolve issue with supply collateral event having null user

Swaylend · Aug 16, 2024 · 7618667 · 7618667
1 parent d0bb16b
commit 7618667
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 21 deletions.
diff --git a/apps/indexer/src/EventHandlers.ts b/apps/indexer/src/EventHandlers.ts
@@ -153,11 +153,29 @@ Market.UserSupplyCollateralEvent.loader(({ event, context }) => {
     loadCollateralAsset: false,
     loadUser: false,
   });
+
+  const address = event.data.address;
+  context.User.load(address.bits);
 });
 
 Market.UserSupplyCollateralEvent.handler(async ({ event, context }) => {
   const id = `${event.transactionId}_${event.receiptIndex}`;
 
+  // Create user if it doesn't exist
+  const user = await context.User.get(event.data.address.bits);
+
+  if (!user) {
+    context.User.set({
+      id: event.data.address.bits,
+      address: event.data.address.bits,
+      principal: BigInt(0),
+      baseTrackingIndex: BigInt(0),
+      baseTrackingAccrued: BigInt(0),
+      totalCollateralBought: BigInt(0),
+      totalValueLiquidated: BigInt(0),
+    });
+  }
+
   context.UserCollateralEvent.set({
     id,
     user_id: event.data.address.bits,

diff --git a/apps/indexer/test/tests.spec.ts b/apps/indexer/test/tests.spec.ts
@@ -408,26 +408,6 @@ describe('Market contract event tests', () => {
 
       const userAddress = ALICE_ADDRESS;
 
-      // We first need to initialize a mock user entity
-      const mockUserBasicEvent = Market.UserBasicEvent.mockData({
-        address: {
-          bits: userAddress,
-        },
-        user_basic: {
-          principal: {
-            negative: false,
-            value: BigInt(0),
-          },
-          base_tracking_index: BigInt(0),
-          base_tracking_accrued: BigInt(0),
-        },
-      });
-
-      let updatedMockDb = await Market.UserBasicEvent.processEvent({
-        event: mockUserBasicEvent,
-        mockDb: mockDbInitial,
-      });
-
       // Create a mock user collateral event
       const mockUserSupplyCollateralEvent =
         Market.UserSupplyCollateralEvent.mockData({
@@ -439,7 +419,7 @@ describe('Market contract event tests', () => {
         });
 
       // Processing the mock event on the mock database
-      updatedMockDb = await Market.UserSupplyCollateralEvent.processEvent({
+      let updatedMockDb = await Market.UserSupplyCollateralEvent.processEvent({
         event: mockUserSupplyCollateralEvent,
         mockDb: mockDbInitial,
       });