tests

Author: teogeb

packages/cli-tools/test/internal-operator.test.ts

@@ -0,0 +1,57 @@
+import { _operatorContractUtils } from '@streamr/sdk'
+import { fetchPrivateKeyWithGas, generateWalletWithGasAndTokens } from '@streamr/test-utils'
+import { createTestClient, runCommand } from './utils'
+import { parseEther, Wallet } from 'ethers'
+import { wait } from '@streamr/utils'
+
+const DELEGATION_AMOUNT = '20000'
+const STAKE_AMOUNT = '10000'
+const SELF_DELEGATION_AMOUNT = '100000'
+const MINIMUM_DELEGATION_SECONDS = 1  // the config value defined in StreamrEnvDeployer in network-contracts repo
+
+describe('operator', () => {
+
+    it('happy path', async () => {
+        const client = createTestClient(await fetchPrivateKeyWithGas())
+        const stream = await client.createStream('/test')
+        const sponsorshipContract = await _operatorContractUtils.deploySponsorshipContract({ 
+            streamId: stream.id,
+            deployer: new Wallet(await fetchPrivateKeyWithGas()).connect(_operatorContractUtils.getProvider())
+        })
+        const sponsorshipAddress: string = await sponsorshipContract.getAddress()
+        const operator = await generateWalletWithGasAndTokens()
+        const operatorContract = await _operatorContractUtils.deployOperatorContract({
+            deployer: operator
+        })
+        await _operatorContractUtils.delegate(operator, await operatorContract.getAddress(), parseEther(SELF_DELEGATION_AMOUNT))
+        const delegator = await generateWalletWithGasAndTokens()
+        const operatorAddress: string = await operatorContract.getAddress()
+
+        // delegate
+        await runCommand(`internal operator-delegate ${operatorAddress} ${DELEGATION_AMOUNT}`, {
+            privateKey: delegator.privateKey
+        })
+        expect(await operatorContract.balanceInData(await delegator.getAddress())).toEqual(parseEther(DELEGATION_AMOUNT))
+
+        // stake
+        await runCommand(`internal operator-stake ${operatorAddress} ${sponsorshipAddress} ${STAKE_AMOUNT}`, {
+            privateKey: operator.privateKey
+        })
+        expect(await operatorContract.totalStakedIntoSponsorshipsWei()).toEqual(parseEther(STAKE_AMOUNT))
+
+        // unstake
+        await runCommand(`internal operator-unstake ${operatorAddress} ${sponsorshipAddress}`, {
+            privateKey: operator.privateKey
+        })
+        expect(await operatorContract.totalStakedIntoSponsorshipsWei()).toEqual(0n)
+
+        // undelegate
+        await wait(MINIMUM_DELEGATION_SECONDS)
+        await runCommand(`internal operator-undelegate ${operatorAddress} ${DELEGATION_AMOUNT}`, {
+            privateKey: delegator.privateKey
+        })
+        expect(await operatorContract.balanceInData(await delegator.getAddress())).toEqual(0n)
+
+        await client.destroy()
+    }, 10 * 1000)
+})

packages/cli-tools/test/internal-sponsorship-sponsor.test.ts

@@ -0,0 +1,28 @@
+import { _operatorContractUtils } from '@streamr/sdk'
+import { fetchPrivateKeyWithGas, generateWalletWithGasAndTokens } from '@streamr/test-utils'
+import { createTestClient, runCommand } from './utils'
+import { parseEther, Wallet } from 'ethers'
+
+const SPONSOR_AMOUNT = '12345'
+
+describe('sponsorship-sponsor', () => {
+
+    it('happy path', async () => {
+        const client = createTestClient(await fetchPrivateKeyWithGas())
+        const stream = await client.createStream('/test')
+        const sponsorshipContract = await _operatorContractUtils.deploySponsorshipContract({ 
+            streamId: stream.id,
+            deployer: new Wallet(await fetchPrivateKeyWithGas()).connect(_operatorContractUtils.getProvider())
+        })
+
+        const sponsorer = await generateWalletWithGasAndTokens()
+        const sponsorshipAddress: string = await sponsorshipContract.getAddress()
+        await runCommand(`internal sponsorship-sponsor ${sponsorshipAddress} ${SPONSOR_AMOUNT}`, {
+            privateKey: sponsorer.privateKey
+        })
+
+        const remainingWei = await sponsorshipContract.connect(sponsorer, _operatorContractUtils.getProvider()).remainingWei()
+        expect(remainingWei).toEqual(parseEther(SPONSOR_AMOUNT))
+        await client.destroy()
+    })
+})