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
3 years ago
|
import chai, { expect } from "chai";
|
||
|
|
import chaiBN from "chai-bn";
|
||
|
|
import BN from "bn.js";
|
||
|
|
chai.use(chaiBN(BN));
|
||
|
|
|
||
|
|
import * as fs from "fs";
|
||
|
|
import { Cell, toNano } from "ton";
|
||
|
|
import { SmartContract } from "ton-contract-executor";
|
||
|
|
import * as main from "../contracts/main";
|
||
|
|
import { internalMessage, randomAddress, setBalance } from "./helpers";
|
||
|
|
|
||
|
|
describe("Deposit and withdraw tests", () => {
|
||
|
|
let contract: SmartContract;
|
||
|
|
|
||
|
|
beforeEach(async () => {
|
||
|
|
contract = await SmartContract.fromCell(
|
||
|
|
Cell.fromBoc(fs.readFileSync("build/main.cell"))[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
|
||
|
|
});
|
||
|
|
});
|