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

import {assert} from "chai"
import DIDRegistry from "../../src/keeper/contracts/DIDRegistry"
import Account from "../../src/ocean/Account"
import { Ocean } from "../../src/ocean/Ocean"
import { generateId } from "../../src/utils/GeneratorHelpers"
import config from "../config"
import TestContractHandler from "./TestContractHandler"

let ocean: Ocean
let didRegistry: DIDRegistry

describe("DIDRegistry", () => {

    before(async () => {
        await TestContractHandler.prepareContracts()
        ocean = await Ocean.getInstance(config)
        didRegistry = ocean.keeper.didRegistry
    })

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

        it("should register an attribute in a new did", async () => {
            const ownerAccount: Account = (await ocean.accounts.list())[0]
            const did = generateId()
            const data = "my nice provider, is nice"
            const receipt = await didRegistry.registerAttribute(did, `0123456789abcdef`, [], data, ownerAccount.getId())
            assert(receipt.status)
            assert(receipt.events.DIDAttributeRegistered)
        })

        it("should register another attribute in the same did", async () => {
            const ownerAccount: Account = (await ocean.accounts.list())[0]
            const did = generateId()
            {
                // register the first attribute
                const data = "my nice provider, is nice"
                await didRegistry.registerAttribute(did, "0123456789abcdef", [], data, ownerAccount.getId())
            }
            {
                // register the second attribute with the same did
                const data = "asdsad"
                const receipt = await didRegistry.registerAttribute(did, "0123456789abcdef", [], data, ownerAccount.getId())
                assert.isTrue(receipt.status)
                assert.isDefined(receipt.events.DIDAttributeRegistered)
            }
        })

    })

    describe("#getDIDOwner()", () => {
        it("should get the owner of a did properly", async () => {
            const ownerAccount: Account = (await ocean.accounts.list())[0]
            const did = generateId()
            const data = "my nice provider, is nice"
            await didRegistry.registerAttribute(did, "0123456789abcdef", [], data, ownerAccount.getId())

            const owner = await didRegistry.getDIDOwner(did)

            assert.equal(owner, ownerAccount.getId(), `Got ${owner} but expected ${ownerAccount.getId()}`)
        })

        it("should get 0x0 for a not registered did", async () => {
            const owner = await didRegistry.getDIDOwner("1234")
            assert.equal(owner, `0x${"0".repeat(40)}`)
        })
    })
})