diff --git a/test/integration/concrete/nft-descriptor/map-symbol/mapSymbol.t.sol b/test/integration/concrete/nft-descriptor/map-symbol/mapSymbol.t.sol index bbceb7968..07461069b 100644 --- a/test/integration/concrete/nft-descriptor/map-symbol/mapSymbol.t.sol +++ b/test/integration/concrete/nft-descriptor/map-symbol/mapSymbol.t.sol @@ -1,7 +1,8 @@ // 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"; @@ -9,9 +10,10 @@ 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 deleted file mode 100644 index 004190ba6..000000000 --- a/test/mocks/erc721/ERC721Mock.sol +++ /dev/null @@ -1,8 +0,0 @@ -// SPDX-License-Identifier: GPL-3.0-or-later -pragma solidity >=0.8.22; - -import { ERC721 } from "@openzeppelin/contracts/token/ERC721/ERC721.sol"; - -contract ERC721Mock is ERC721 { - constructor(string memory name, string memory symbol) ERC721(name, symbol) { } -}