squid-js/test/mocks/Aquarius.mock.ts

import { Aquarius } from "../../src/aquarius/Aquarius"
import { DDO } from "../../src/ddo/DDO"
import DID from "../../src/ocean/DID"

const ddoStore: Map<string, any> = new Map<string, any>()

export default class AquariusMock extends Aquarius {

    public async getAccessUrl(accessToken: any, payload: any): Promise<string> {
        return "http://test/test"
    }

    public async storeDDO(ddo: DDO): Promise<DDO> {
        ddoStore.set(ddo.id, ddo)
        return ddo
    }

    public async retrieveDDO(did: DID): Promise<DDO> {
        return ddoStore.get(did.getDid())
    }
}