Clones.test.js 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. const { expectEvent, expectRevert } = require('@openzeppelin/test-helpers');
  2. const { computeCreate2Address } = require('../helpers/create2');
  3. const { expect } = require('chai');
  4. const shouldBehaveLikeClone = require('./Clones.behaviour');
  5. const Clones = artifacts.require('$Clones');
  6. contract('Clones', function (accounts) {
  7. const [deployer] = accounts;
  8. describe('clone', function () {
  9. shouldBehaveLikeClone(async (implementation, initData, opts = {}) => {
  10. const factory = await Clones.new();
  11. const receipt = await factory.$clone(implementation);
  12. const address = receipt.logs.find(({ event }) => event === 'return$clone').args.instance;
  13. await web3.eth.sendTransaction({ from: deployer, to: address, value: opts.value, data: initData });
  14. return { address };
  15. });
  16. });
  17. describe('cloneDeterministic', function () {
  18. shouldBehaveLikeClone(async (implementation, initData, opts = {}) => {
  19. const salt = web3.utils.randomHex(32);
  20. const factory = await Clones.new();
  21. const receipt = await factory.$cloneDeterministic(implementation, salt);
  22. const address = receipt.logs.find(({ event }) => event === 'return$cloneDeterministic').args.instance;
  23. await web3.eth.sendTransaction({ from: deployer, to: address, value: opts.value, data: initData });
  24. return { address };
  25. });
  26. it('address already used', async function () {
  27. const implementation = web3.utils.randomHex(20);
  28. const salt = web3.utils.randomHex(32);
  29. const factory = await Clones.new();
  30. // deploy once
  31. expectEvent(await factory.$cloneDeterministic(implementation, salt), 'return$cloneDeterministic');
  32. // deploy twice
  33. await expectRevert(factory.$cloneDeterministic(implementation, salt), 'ERC1167: create2 failed');
  34. });
  35. it('address prediction', async function () {
  36. const implementation = web3.utils.randomHex(20);
  37. const salt = web3.utils.randomHex(32);
  38. const factory = await Clones.new();
  39. const predicted = await factory.$predictDeterministicAddress(implementation, salt);
  40. const creationCode = [
  41. '0x3d602d80600a3d3981f3363d3d373d3d3d363d73',
  42. implementation.replace(/0x/, '').toLowerCase(),
  43. '5af43d82803e903d91602b57fd5bf3',
  44. ].join('');
  45. expect(computeCreate2Address(salt, creationCode, factory.address)).to.be.equal(predicted);
  46. expectEvent(await factory.$cloneDeterministic(implementation, salt), 'return$cloneDeterministic', {
  47. instance: predicted,
  48. });
  49. });
  50. });
  51. });