very basic erc20 restrictions

mints full supply during construction

Signed-off-by: stadolf <stadolf@gmail.com>

Author: elmariachi111

script/Counter.s.sol

@@ -0,0 +1,16 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.25;
+
+import {Script, console} from "forge-std/Script.sol";
+import {WelToken} from "../src/WelToken.sol";
+
+contract DeployWelToken is Script {
+    
+    address sepoliaMultisig = 0x62ee749EA4d0407a2C44690933F58B05C7867e5f;
+
+    function run() public {
+        vm.broadcast();
+        WelToken token = new WelToken(sepoliaMultisig);
+        console.log("WEL_TOKEN=", address(token));
+    }
+}

script/DeployToken.sol

@@ -30,8 +30,10 @@ import "@openzeppelin/contracts/token/ERC20/extensions/ERC20Permit.sol";
 
 /// @custom:security-contact stefan@welshare.health
 contract WelToken is ERC20, ERC20Permit {
-    constructor() ERC20("WelToken", "WEL") ERC20Permit("WelToken") {
-        _mint(msg.sender, 1000000 * 10 ** decimals());
+
+    uint256 public constant WEL_SUPPLY = 2_500_000_000 ether;
+    constructor(address treasury) ERC20("WelToken", "WEL") ERC20Permit("WelToken")  {
+        _mint(treasury, WEL_SUPPLY);
     }
 }
 

src/Counter.sol

@@ -0,0 +1,30 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.25;
+
+import {Test, console} from "forge-std/Test.sol";
+import {WelToken} from "../src/WelToken.sol";
+
+contract WelTokenTest is Test {
+    
+    address deployer = makeAddr("deployer");
+    address treasury = makeAddr("treasury");
+    address welteam = makeAddr("welteam");
+
+    function setUp() public {
+    }
+
+    function test_initialize() public {
+        vm.startPrank(deployer);
+        WelToken token = new WelToken(treasury);
+        assertEq(token.name(), "WelToken");
+        assertEq(token.totalSupply(), 2_500_000_000 ether);
+        assertEq(token.balanceOf(treasury), 2_500_000_000 ether);
+
+        vm.startPrank(treasury);
+        token.transfer(welteam, 10_000 ether);
+
+        assertEq(token.balanceOf(welteam), 10_000 ether);
+        assertEq(token.balanceOf(treasury), 2_500_000_000 ether - 10_000 ether);
+        assertEq(token.totalSupply(), 2_500_000_000 ether);
+    }
+}