Fix upcoming

e2e/block.test.ts

@@ -0,0 +1,37 @@
+import { describe, expect, it } from 'vitest'
+
+import { chain, delay, dev, setupApi } from './helper'
+
+setupApi({ endpoint: 'wss://rpc.polkadot.io' })
+
+describe('block', () => {
+  it('upcoming block works', async () => {
+    const blockNumber = chain.head.number
+
+    setTimeout(() => {
+      dev.newBlock()
+    }, 1000)
+    {
+      const next = await chain.upcomingBlock()
+      expect(next.number).toEqual(blockNumber + 1)
+    }
+
+    setTimeout(() => {
+      dev.newBlock()
+    }, 1000)
+    {
+      const next = await chain.upcomingBlock()
+      expect(next.number).toEqual(blockNumber + 2)
+    }
+
+    setTimeout(() => {
+      dev.newBlock({ count: 3 })
+    }, 1000)
+    {
+      const next = await chain.upcomingBlock(2)
+      expect(next.number).toEqual(blockNumber + 5)
+    }
+
+    await delay(1000)
+  })
+})

e2e/helper.ts

@@ -121,7 +121,8 @@ export const setupApi = (option: SetupOption) => {
   beforeAll(async () => {
     const res = await setupAll(option)
     setup = res.setup
-    return () => res.teardownAll()
+
+    return res.teardownAll
   })
 
   beforeEach(async () => {

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@acala-network/chopsticks",
-	"version": "0.3.8",
+	"version": "0.3.9",
 	"main": "./dist/index.js",
 	"types": "./dist/index.d.ts",
 	"author": "Bryan Chen <[email protected]>",

src/blockchain/index.ts

@@ -161,8 +161,8 @@ export class Blockchain {
     return this.#head
   }
 
-  async upcomingBlock(count = 1) {
-    return this.#txpool.upcomingBlock(count)
+  async upcomingBlock(skipCount = 0) {
+    return this.#txpool.upcomingBlock(skipCount)
   }
 
   async dryRunExtrinsic(

src/blockchain/txpool.ts

@@ -1,6 +1,6 @@
+import { BehaviorSubject, firstValueFrom } from 'rxjs'
 import { HexString } from '@polkadot/util/types'
-import { ReplaySubject, firstValueFrom, share } from 'rxjs'
-import { first, skip } from 'rxjs/operators'
+import { skip, take } from 'rxjs/operators'
 import _ from 'lodash'
 
 import { Block } from './block'
@@ -37,13 +37,12 @@ export class TxPool {
   readonly #mode: BuildBlockMode
   readonly #inherentProvider: InherentProvider
 
+  #last: BehaviorSubject<Block>
   #lastBuildBlockPromise: Promise<void> = Promise.resolve()
 
-  #last = new ReplaySubject<Block>(1)
-  #upcoming = this.#last.pipe(share())
-
   constructor(chain: Blockchain, inherentProvider: InherentProvider, mode: BuildBlockMode = BuildBlockMode.Batch) {
     this.#chain = chain
+    this.#last = new BehaviorSubject<Block>(chain.head)
     this.#mode = mode
     this.#inherentProvider = inherentProvider
   }
@@ -77,9 +76,9 @@ export class TxPool {
     this.#last.next(this.#chain.head)
   }
 
-  async upcomingBlock(count = 1) {
-    if (count < 1) throw new Error('count needs to be greater than 0')
-    return firstValueFrom(this.#upcoming.pipe(skip(count - 1), first()))
+  async upcomingBlock(skipCount = 0) {
+    if (skipCount < 0) throw new Error('skipCount needs to be greater or equal to 0')
+    return firstValueFrom(this.#last.pipe(skip(1 + skipCount), take(1)))
   }
 
   async #buildBlock(wait: Promise<void>, params?: BuildBlockParams) {