add merchantNft

Author: Axel Valavaara

makefile

@@ -1,4 +1,5 @@
 -include .env
 .PHONY: all test deploy
 
-deploy :; forge script script/DeployOrderContract.s.sol:DeployOrderContract --rpc-url $(SEPOLIA_RPC) --broadcast --private-key $(PRIVATE_KEY) --verify -vvvv
+deploysepolia :; forge script script/DeployOrderContract.s.sol:DeployOrderContract --rpc-url $(SEPOLIA_RPC) --broadcast --private-key $(PRIVATE_KEY) --verify -vvvv
+deployanvil :; forge script script/DeployOrderContract.s.sol:DeployOrderContract --rpc-url $(ANVIL_RPC) --broadcast --private-key $(DEFAULT_ANVIL_KEY) -vvvv

script/DeployOrderContract.s.sol

@@ -7,16 +7,18 @@ import {OrderContract} from "../src/OrderContract.sol";
 import {Script} from "forge-std/Script.sol";
 import {HelperConfig} from "./HelperConfig.s.sol";
 import {A3AToken} from "../src/A3Atoken.sol";
+import {MerchantNft} from "../src/MerchantNft.sol";
 contract DeployOrderContract is Script {
 
 
-    function run() public returns (OrderContract, HelperConfig, A3AToken) {
+    function run() public returns (OrderContract, HelperConfig, A3AToken, MerchantNft) {
         HelperConfig helperConfig = new HelperConfig();
 
         (address pyUSD, address agentController, uint256 deployerKey, address owner)= helperConfig.activeNetworkConfig();
 
         vm.startBroadcast(deployerKey);
         A3AToken token = new A3AToken();
+        MerchantNft merchantNft = new MerchantNft();
         OrderContract orderContract = new OrderContract(
         agentController,  pyUSD, address(token), owner // Replace with actual pyUSD token address
         );
@@ -30,6 +32,6 @@ contract DeployOrderContract is Script {
 
 
         vm.stopBroadcast();
-        return (orderContract, helperConfig, token);
+        return (orderContract, helperConfig, token, merchantNft);
     }
 }

src/MerchantNft.sol

@@ -0,0 +1,30 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.18;
+
+import {ERC721} from "@openzeppelin/contracts/token/ERC721/ERC721.sol";
+
+contract MerchantNft is ERC721 {
+    error MerchantNft__MerchantIdAlreadyMinted();
+    
+
+    constructor() ERC721("MerchantNft", "MNFT") {
+        
+    }
+
+    function mintNft(uint256 merchantId) public {
+        if (_ownerOf(merchantId) != address(0)) {
+            revert MerchantNft__MerchantIdAlreadyMinted();
+        }
+        _safeMint(msg.sender, merchantId);
+        
+    }
+
+    function isMerchant(address wallet, uint256 merchantId) external view returns (bool) {
+        if (_ownerOf(merchantId) == wallet) {
+            return true;
+        } else {
+            return false;
+        }
+    }
+}

src/OrderContract.sol

@@ -27,6 +27,7 @@ import {ERC20} from "@openzeppelin/contracts/token/ERC20/ERC20.sol";
 import {ReentrancyGuard} from "@openzeppelin/contracts/utils/ReentrancyGuard.sol";
 import {A3AToken} from "./A3Atoken.sol";
 
+
 contract OrderContract is ReentrancyGuard{
 
     /*//////////////////////////////////////////////////////////////

test/fuzz/InvariantsTest.t.sol

@@ -15,7 +15,7 @@ contract InvariantsTest is StdInvariant, Test {
 
     function setUp() public {
         DeployOrderContract deployer = new DeployOrderContract();
-        (orderContract, helperConfig,) = deployer.run();
+        (orderContract, helperConfig,,) = deployer.run();
         targetContract(address(orderContract));
     }
 

test/unit/OrderContractTest.t.sol

@@ -8,6 +8,8 @@ import {HelperConfig} from "script/HelperConfig.s.sol";
 import {DeployOrderContract} from "script/DeployOrderContract.s.sol";
 import {ERC20Mock} from "@openzeppelin/contracts/mocks/token/ERC20Mock.sol";
 import {A3AToken} from "../../src/A3Atoken.sol";
+import {ERC721} from "@openzeppelin/contracts/token/ERC721/ERC721.sol";
+import {MerchantNft} from "../../src/MerchantNft.sol";
 
 contract OrderContractTest is Test {
     OrderContract orderContract;
@@ -16,6 +18,7 @@ contract OrderContractTest is Test {
     address addressController;
     A3AToken a3aToken;
     address owner;
+    MerchantNft merchantNft;
 
 
     bytes32 constant PROMPT_HASH = keccak256(abi.encodePacked("Test Prompt"));
@@ -35,7 +38,7 @@ contract OrderContractTest is Test {
 
     function setUp() public {
         DeployOrderContract deployer = new DeployOrderContract();
-        (orderContract, helperConfig, a3aToken) = deployer.run();
+        (orderContract, helperConfig, a3aToken, merchantNft) = deployer.run();
         (pyUSD, addressController,,owner) = helperConfig.activeNetworkConfig();
         ERC20Mock(pyUSD).mint(USER, DEFAULT_MINT_AMOUNT);
         ERC20Mock(pyUSD).mint(USER2, DEFAULT_MINT_AMOUNT);
@@ -350,6 +353,48 @@ contract OrderContractTest is Test {
         assertEq(uint8(offer.status), uint8(OrderContract.OrderStatus.Proposed));
     }
 
+    /*//////////////////////////////////////////////////////////////
+                           MERCHANT NFT TESTS
+    //////////////////////////////////////////////////////////////*/
+    function testMerchantNftMintingAndOwnership() public {
+        // Arrange
+        uint256 merchantId = 1;
+        // Act
+        vm.prank(SELLER);
+        merchantNft.mintNft(merchantId);
+        // Assert
+        address ownerOfNft = merchantNft.ownerOf(merchantId);
+        assertEq(ownerOfNft, SELLER);
+        bool isMerchant = merchantNft.isMerchant(SELLER, merchantId);
+        assertTrue(isMerchant);
+
+    }
+
+
+    function testMerchantNftRevertsOnDuplicateMint() public {
+        // Arrange
+        uint256 merchantId = 1;
+        vm.prank(SELLER);
+        merchantNft.mintNft(merchantId);
+        // Act / Assert
+        vm.prank(USER);
+        vm.expectRevert(MerchantNft.MerchantNft__MerchantIdAlreadyMinted.selector);
+        merchantNft.mintNft(merchantId);
+
+    }
+
+    function testMerchantNftReturnsFalseIfNotOwner() public {
+        // Arrange
+        uint256 merchantId = 1;
+        vm.prank(SELLER);
+        merchantNft.mintNft(merchantId);
+        // Act
+        bool isMerchant = merchantNft.isMerchant(USER, merchantId);
+        // Assert
+        assertFalse(isMerchant);
+    }
+
+
     /*//////////////////////////////////////////////////////////////
                        USER ORDER MAPPINGS TESTS
     //////////////////////////////////////////////////////////////*/