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

import {assert} from "chai"
import ConfigProvider from "../../src/ConfigProvider"
import EventListener from "../../src/keeper/EventListener"
import Keeper from "../../src/keeper/Keeper"
import Account from "../../src/ocean/Account"
import { Ocean } from "../../src/ocean/Ocean"
import config from "../config"
import TestContractHandler from "./TestContractHandler"

let keeper: Keeper
let ocean: Ocean
let accounts: Account[]

describe("EventListener", () => {

    before(async () => {
        ConfigProvider.setConfig(config)
        await TestContractHandler.prepareContracts()
        keeper = await Keeper.getInstance()
        assert(keeper)
        ocean = await Ocean.getInstance(config)
        assert(ocean)
        accounts = await ocean.getAccounts()
        assert(accounts)
    })

    describe("#listen()", () => {

        it("should listen to an event", (done) => {

            const acc = accounts[1]

            const event = EventListener.subscribe("OceanToken",
                "Transfer",
                {
                    to: acc.getId(),
                })

            let doneCalled: boolean
            event.listen((events) => {
                assert(events)
                assert(events.length === 2)
                EventListener.unsubscribe(event)
                if (!doneCalled) {
                    doneCalled = true
                    done()
                }
            })

            const {dispenser} = keeper

            dispenser.requestTokens(10, acc.getId())
            dispenser.requestTokens(10, acc.getId())
        })
    })

    describe("#listenOnce()", () => {

        it("should listen once", (done) => {

            const acc = accounts[1]

            const countBefore = EventListener.count()
            const event = EventListener.subscribe("OceanToken",
                "Transfer",
                {
                    to: acc.getId(),
                })

            event.listenOnce(
                (data: any) => {

                    assert(data)
                    assert(data.blockNumber)
                    assert(EventListener.count() === countBefore)
                    done()
                })

            const {dispenser} = keeper

            dispenser.requestTokens(10, acc.getId())
        })
    })

})