squid-js/test/keeper/EventHandler.test.ts

91 lines
2.6 KiB
TypeScript

import { assert, expect, spy, use } from 'chai'
import * as spies from 'chai-spies'
import { EventHandler } from '../../src/keeper/EventHandler'
import { Ocean } from '../../src/ocean/Ocean'
import config from '../config'
use(spies)
describe('EventHandler', () => {
let ocean: Ocean
let eventHandler: EventHandler
before(async () => {
ocean = await Ocean.getInstance(config)
eventHandler = new EventHandler((ocean as any).instanceConfig)
})
afterEach(() => {
spy.restore()
})
describe('#subscribe()', () => {
it('should subscribe to an event', async () => {
const countBefore = eventHandler.count
const subscription = eventHandler.subscribe(() => null)
assert.isDefined(subscription)
const countAfter = eventHandler.count
assert.equal(
countBefore + 1,
countAfter,
'The event seems not added.'
)
subscription.unsubscribe()
})
it('should unsubscribe using the subscription', async () => {
const countBefore = eventHandler.count
const subscription = eventHandler.subscribe(() => null)
assert.isDefined(subscription)
subscription.unsubscribe()
const countAfter = eventHandler.count
assert.equal(countBefore, countAfter, 'The event seems not added.')
})
})
describe('#unsubscribe()', () => {
it('should unsubscribe from an event', async () => {
const countBefore = eventHandler.count
const callback = () => null
eventHandler.subscribe(callback)
eventHandler.unsubscribe(callback)
const countAfter = eventHandler.count
assert.equal(
countBefore,
countAfter,
'The event seems not removed.'
)
})
})
describe('#checkBlock()', () => {
it('should call the callback on each new block', async () => {
let blockNumber = 100000000000
const callbackSpy = spy()
spy.on((ocean as any).web3.eth, 'getBlockNumber', () => blockNumber)
const subscription = eventHandler.subscribe(callbackSpy)
await new Promise(_ => setTimeout(_, 300))
expect(callbackSpy).not.to.has.been.called()
blockNumber++
await new Promise(_ => setTimeout(_, 300))
expect(callbackSpy).to.has.been.called.with(blockNumber)
subscription.unsubscribe()
})
})
})