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

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()
        })
    })
})