58 lines
2.3 KiB
Solidity
58 lines
2.3 KiB
Solidity
// SPDX-License-Identifier: MIT
|
|
pragma solidity ^0.8.13;
|
|
|
|
import {Test, console} from "forge-std/Test.sol";
|
|
import {Army, Raider} from "../src/RaidGeld.sol";
|
|
import "../src/RaidGeldUtils.sol";
|
|
|
|
contract raid_geldTest is Test {
|
|
function test_0_unit_price() public {
|
|
// buying 1 unit of moloch_denier
|
|
uint256 basePriceMolochDenier = RaidGeldUtils.calculateUnitPrice(0, 0, 1);
|
|
assertEq(basePriceMolochDenier, RaidGeldUtils.BASE_PRICE);
|
|
|
|
// buying 3 units
|
|
// has to be a bit more than 3 * 38 = 114
|
|
uint256 price = RaidGeldUtils.calculateUnitPrice(0, 0, 3);
|
|
assertGt(price, 3 * basePriceMolochDenier);
|
|
|
|
uint256 basePriceChamp = RaidGeldUtils.calculateUnitPrice(0, 0, 3);
|
|
// buying 3 units of champions
|
|
// has to be a bit more than 4 * 38 * 3 = 114
|
|
price = RaidGeldUtils.calculateUnitPrice(3, 0, 3);
|
|
assertGt(price, 3 * basePriceChamp);
|
|
|
|
// buying 12 units of champions when u already have 10
|
|
price = RaidGeldUtils.calculateUnitPrice(3, 10, 12);
|
|
// price should scale up
|
|
assertGt(price, basePriceChamp * 12);
|
|
}
|
|
|
|
function test_1_profits_per_second(uint16 _dLvl, uint16 _apLvl, uint16 _anLvl, uint16 _cLvl) public {
|
|
Army memory army = Army({
|
|
moloch_denier: Raider({ level: 1}),
|
|
apprentice: Raider({ level: 0}),
|
|
anointed: Raider({ level: 0}),
|
|
champion: Raider({ level: 0}),
|
|
profit_per_second: 0 // irrelevant for this test
|
|
});
|
|
uint256 profits_per_second = RaidGeldUtils.calculateProfitsPerSecond(army);
|
|
assertEq(profits_per_second, RaidGeldUtils.PRECISION);
|
|
|
|
army = Army({
|
|
moloch_denier: Raider({ level: _dLvl}),
|
|
apprentice: Raider({ level: _apLvl}),
|
|
anointed: Raider({ level: _anLvl}),
|
|
champion: Raider({ level: _cLvl}),
|
|
profit_per_second: 0 // irrelevant for this test
|
|
});
|
|
profits_per_second = RaidGeldUtils.calculateProfitsPerSecond(army);
|
|
uint256 expected =
|
|
_dLvl * RaidGeldUtils.PRECISION +
|
|
_apLvl * RaidGeldUtils.APPRENTICE_PROFIT +
|
|
_anLvl * RaidGeldUtils.ANOINTED_PROFIT +
|
|
_cLvl * RaidGeldUtils.CHAMPION_PROFIT;
|
|
assertEq(profits_per_second, expected);
|
|
}
|
|
}
|