fix tests

open-web3-stack · Oct 23, 2024 · d7b0755 · d7b0755
1 parent 34449c3
commit d7b0755
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/Blockchain/Tests/BlockchainTests/ExtrinsicPoolServiceTests.swift b/Blockchain/Tests/BlockchainTests/ExtrinsicPoolServiceTests.swift
@@ -56,7 +56,11 @@ struct ExtrinsicPoolServiceTests {
 
             allTickets.append(contentsOf: tickets)
 
-            let event = RuntimeEvents.SafroleTicketsGenerated(items: tickets, publicKey: secretKey.publicKey)
+            let event = RuntimeEvents.SafroleTicketsGenerated(
+                epochIndex: state.value.timeslot.timeslotToEpochIndex(config: config),
+                items: tickets,
+                publicKey: secretKey.publicKey
+            )
             await eventBus.publish(event)
 
             // Wait for the event to be processed
@@ -126,7 +130,11 @@ struct ExtrinsicPoolServiceTests {
             idx: 0
         )
 
-        let addEvent = RuntimeEvents.SafroleTicketsGenerated(items: tickets, publicKey: secretKey.publicKey)
+        let addEvent = RuntimeEvents.SafroleTicketsGenerated(
+            epochIndex: state.value.timeslot.timeslotToEpochIndex(config: config),
+            items: tickets,
+            publicKey: secretKey.publicKey
+        )
         await eventBus.publish(addEvent)
 
         // Wait for the event to be processed
@@ -173,7 +181,11 @@ struct ExtrinsicPoolServiceTests {
             idx: 0
         )
 
-        let addEvent = RuntimeEvents.SafroleTicketsGenerated(items: oldTickets, publicKey: secretKey.publicKey)
+        let addEvent = RuntimeEvents.SafroleTicketsGenerated(
+            epochIndex: state.value.timeslot.timeslotToEpochIndex(config: config),
+            items: oldTickets,
+            publicKey: secretKey.publicKey
+        )
         await eventBus.publish(addEvent)
         await storeMiddleware.wait()
 
@@ -216,7 +228,11 @@ struct ExtrinsicPoolServiceTests {
         )
 
         // Ensure new tickets are accepted
-        let newAddEvent = RuntimeEvents.SafroleTicketsGenerated(items: newTickets, publicKey: secretKey.publicKey)
+        let newAddEvent = RuntimeEvents.SafroleTicketsGenerated(
+            epochIndex: newState.value.timeslot.timeslotToEpochIndex(config: config),
+            items: newTickets,
+            publicKey: secretKey.publicKey
+        )
         await eventBus.publish(newAddEvent)
         await storeMiddleware.wait()