You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
3.1 KiB
85 lines
3.1 KiB
import chai, { expect } from "chai"; |
|
import chaiBN from "chai-bn"; |
|
import BN from "bn.js"; |
|
chai.use(chaiBN(BN)); |
|
|
|
import { Cell, toNano } from "ton"; |
|
import { SmartContract } from "ton-contract-executor"; |
|
import * as main from "../contracts/main"; |
|
import { internalMessage, randomAddress, setBalance } from "./helpers"; |
|
|
|
import { hex } from "../build/main-bitcode.json"; |
|
|
|
describe("Deposit and withdraw tests", () => { |
|
let contract: SmartContract; |
|
|
|
beforeEach(async () => { |
|
contract = await SmartContract.fromCell( |
|
Cell.fromBoc(hex)[0], // code cell from build output |
|
main.data({ |
|
ownerAddress: randomAddress("owner"), |
|
counter: 17, |
|
}) |
|
); |
|
}); |
|
|
|
it("should get balance", async () => { |
|
setBalance(contract, toNano(37)); |
|
const call = await contract.invokeGetMethod("balance", []); |
|
expect(call.result[0]).to.be.bignumber.equal(toNano(37)); |
|
}); |
|
|
|
it("should allow the owner to withdraw when balance is high", async () => { |
|
setBalance(contract, toNano(37)); |
|
const send = await contract.sendInternalMessage( |
|
internalMessage({ |
|
from: randomAddress("owner"), |
|
body: main.withdraw({ withdrawAmount: toNano(20) }), |
|
}) |
|
); |
|
expect(send.type).to.equal("success"); |
|
expect(send.actionList).to.have.lengthOf(1); |
|
const resultMessage = (send.actionList[0] as any)?.message?.info; |
|
expect(resultMessage?.dest?.equals(randomAddress("owner"))).to.equal(true); |
|
expect(resultMessage?.value?.coins).to.be.bignumber.equal(toNano(20)); |
|
}); |
|
|
|
it("should prevent others from withdrawing when balance is high", async () => { |
|
setBalance(contract, toNano(37)); |
|
const send = await contract.sendInternalMessage( |
|
internalMessage({ |
|
from: randomAddress("notowner"), |
|
body: main.withdraw({ withdrawAmount: toNano(20) }), |
|
}) |
|
); |
|
expect(send.type).to.equal("failed"); |
|
expect(send.exit_code).to.equal(102); // access_denied in contracts/imports/constants.fc |
|
}); |
|
|
|
it("should prevent the owner to withdraw when balance is low", async () => { |
|
setBalance(contract, toNano(10)); |
|
const send = await contract.sendInternalMessage( |
|
internalMessage({ |
|
from: randomAddress("owner"), |
|
body: main.withdraw({ withdrawAmount: toNano(20) }), |
|
}) |
|
); |
|
expect(send.type).to.equal("failed"); |
|
expect(send.exit_code).to.equal(103); // insufficient_balance in contracts/imports/constants.fc |
|
}); |
|
|
|
it("should leave enough balance for rent", async () => { |
|
setBalance(contract, toNano(20)); |
|
const send = await contract.sendInternalMessage( |
|
internalMessage({ |
|
from: randomAddress("owner"), |
|
body: main.withdraw({ withdrawAmount: toNano(20) }), |
|
}) |
|
); |
|
expect(send.type).to.equal("success"); |
|
expect(send.actionList).to.have.lengthOf(1); |
|
const resultMessage = (send.actionList[0] as any)?.message?.info; |
|
expect(resultMessage?.dest?.equals(randomAddress("owner"))).to.equal(true); |
|
expect(resultMessage?.value?.coins).to.be.bignumber.equal(toNano(20).sub(toNano(0.01))); // min_tons_for_storage in contracts/imports/constants.fc |
|
}); |
|
});
|
|
|