test: remove ERC721Mock and use forge mock

sablier-labs · Apr 9, 2024 · 188e244 · 188e244
1 parent 351ae01
commit 188e244
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/test/integration/concrete/nft-descriptor/map-symbol/mapSymbol.t.sol b/test/integration/concrete/nft-descriptor/map-symbol/mapSymbol.t.sol
@@ -1,17 +1,19 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity >=0.8.22 <0.9.0;
 
-import { ERC721Mock } from "../../../../mocks/erc721/ERC721Mock.sol";
+import { IERC721Metadata } from "@openzeppelin/contracts/token/ERC721/extensions/IERC721Metadata.sol";
+import { MockERC721 } from "forge-std/src/mocks/MockERC721.sol";
 
 import { Errors } from "src/libraries/Errors.sol";
 
 import { NFTDescriptor_Integration_Concrete_Test } from "../NFTDescriptor.t.sol";
 
 contract MapSymbol_Integration_Concrete_Test is NFTDescriptor_Integration_Concrete_Test {
     function test_RevertGiven_UnknownNFT() external {
-        ERC721Mock nft = new ERC721Mock("Foo NFT", "FOO");
+        MockERC721 nft = new MockERC721();
+        nft.initialize("Foo", "FOO");
         vm.expectRevert(abi.encodeWithSelector(Errors.SablierV2NFTDescriptor_UnknownNFT.selector, nft, "FOO"));
-        nftDescriptorMock.mapSymbol_(nft);
+        nftDescriptorMock.mapSymbol_(IERC721Metadata(address(nft)));
     }
 
     modifier givenKnownNFT() {

diff --git a/test/mocks/erc721/ERC721Mock.sol b/test/mocks/erc721/ERC721Mock.sol