|
2 | 2 |
|
3 | 3 | pragma solidity ^0.8.8; |
4 | 4 |
|
5 | | -import { ERC2981 } from './ERC2981.sol'; |
| 5 | +import { ERC2981, IERC2981 } from './ERC2981.sol'; |
6 | 6 | import { ERC2981Storage } from './ERC2981Storage.sol'; |
7 | 7 |
|
8 | | -import { ERC165 } from '../../../introspection/ERC165.sol'; |
| 8 | +import { IERC165, ERC165, ERC165Storage } from '../../../introspection/ERC165.sol'; |
9 | 9 |
|
10 | 10 | contract ERC2981Mock is ERC2981, ERC165 { |
| 11 | + using ERC165Storage for ERC165Storage.Layout; |
| 12 | + |
11 | 13 | constructor( |
12 | 14 | uint16 defaultRoyaltyBPS, |
13 | 15 | uint16[] memory royaltiesBPS, |
14 | 16 | address defaultRoyaltyReceiver |
15 | 17 | ) { |
16 | | - ERC2981Storage.Layout storage l = ERC2981Storage.layout(); |
| 18 | + { |
| 19 | + ERC2981Storage.Layout storage l = ERC2981Storage.layout(); |
| 20 | + l.defaultRoyaltyBPS = defaultRoyaltyBPS; |
| 21 | + l.defaultRoyaltyReceiver = defaultRoyaltyReceiver; |
17 | 22 |
|
18 | | - l.defaultRoyaltyBPS = defaultRoyaltyBPS; |
19 | | - l.defaultRoyaltyReceiver = defaultRoyaltyReceiver; |
| 23 | + for (uint8 i = 0; i < royaltiesBPS.length; i++) { |
| 24 | + l.royaltiesBPS[i] = royaltiesBPS[i]; |
| 25 | + } |
| 26 | + } |
20 | 27 |
|
21 | | - for (uint8 i = 0; i < royaltiesBPS.length; i++) { |
22 | | - l.royaltiesBPS[i] = royaltiesBPS[i]; |
| 28 | + { |
| 29 | + ERC165Storage.Layout storage l = ERC165Storage.layout(); |
| 30 | + l.setSupportedInterface(type(IERC165).interfaceId, true); |
| 31 | + l.setSupportedInterface(type(IERC2981).interfaceId, true); |
23 | 32 | } |
24 | 33 | } |
25 | 34 |
|
|
0 commit comments