Merge pull request #123 from dev-protocol/add-migrate-test

add library migrate test

Author: Akira-Taniguchi

contracts/test/common/proxy/UpgradeabilityLibraryV1.sol

@@ -0,0 +1,43 @@
+// SPDX-License-Identifier: MPL-2.0
+pragma solidity =0.8.9;
+
+import "@openzeppelin/contracts/utils/Counters.sol";
+import "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
+import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
+import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
+
+contract UpgradeabilityLibraryV1 is OwnableUpgradeable, UUPSUpgradeable {
+	uint256 public testValue;
+	Counters.Counter private tokenIdCounter;
+	EnumerableSet.UintSet private enumerableSet;
+
+	using Counters for Counters.Counter;
+	using EnumerableSet for EnumerableSet.UintSet;
+
+	function initialize() external initializer {
+		__Ownable_init();
+		__UUPSUpgradeable_init();
+	}
+
+	function upCounter() public {
+		tokenIdCounter.increment();
+	}
+
+	function getCounter() public view returns (uint256) {
+		return tokenIdCounter.current();
+	}
+
+	function addEnumerableSet(uint256 _value) public {
+		enumerableSet.add(_value);
+	}
+
+	function getEnumerableSet() public view returns (uint256[] memory) {
+		return enumerableSet.values();
+	}
+
+	function setTestValue(uint256 _value) public {
+		testValue = _value;
+	}
+
+	function _authorizeUpgrade(address) internal override onlyOwner {}
+}

contracts/test/common/proxy/UpgradeabilityLibraryV2.sol

@@ -0,0 +1,43 @@
+// SPDX-License-Identifier: MPL-2.0
+pragma solidity =0.8.9;
+
+import "@openzeppelin/contracts/utils/Counters.sol";
+import "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
+import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
+import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
+
+contract UpgradeabilityLibraryV2 is OwnableUpgradeable, UUPSUpgradeable {
+	uint256 public testValue;
+	Counters.Counter private tokenIdCounter;
+	EnumerableSet.UintSet private enumerableSet;
+
+	using Counters for Counters.Counter;
+	using EnumerableSet for EnumerableSet.UintSet;
+
+	function initialize() external initializer {
+		__Ownable_init();
+		__UUPSUpgradeable_init();
+	}
+
+	function upCounter() public {
+		tokenIdCounter.increment();
+	}
+
+	function getCounter() public view returns (uint256) {
+		return tokenIdCounter.current();
+	}
+
+	function addEnumerableSet(uint256 _value) public {
+		enumerableSet.add(_value);
+	}
+
+	function getEnumerableSet() public view returns (uint256[] memory) {
+		return enumerableSet.values();
+	}
+
+	function setTestValue(uint256 _value) public {
+		testValue = _value;
+	}
+
+	function _authorizeUpgrade(address) internal override onlyOwner {}
+}

test/common/proxy/upgradeability-library.ts

@@ -0,0 +1,49 @@
+/* eslint-disable @typescript-eslint/no-floating-promises */
+import { UpgradeabilityLibraryV1Instance } from '../../../types/truffle-contracts'
+import { deployProxy } from '../../test-lib/instance'
+import { getEventValue } from '../../test-lib/utils/event'
+
+contract('UpgradeabilityLibrary ', ([deployer, address, user]) => {
+	describe('Same data', () => {
+		let contract: UpgradeabilityLibraryV1Instance
+
+		before(async () => {
+			contract = await deployProxy(
+				artifacts.require('UpgradeabilityLibraryV1'),
+				deployer
+			)
+			await contract.initialize()
+		})
+		it('Store data', async () => {
+			await Promise.all([
+				contract.setTestValue(5),
+				contract.upCounter(),
+				contract.addEnumerableSet(10),
+				contract.addEnumerableSet(20),
+			])
+
+			expect((await contract.getCounter()).toString()).to.equal('1')
+			const eSet = await contract.getEnumerableSet()
+			expect(eSet.length).to.equal(2)
+			const converted = eSet.map((v) => v.toNumber())
+			expect(converted.includes(10)).to.equal(true)
+			expect(converted.includes(20)).to.equal(true)
+			expect((await contract.testValue()).toString()).to.equal('5')
+		})
+		it('Should data be upgradable', async () => {
+			const newImpl = await artifacts.require('UpgradeabilityLibraryV2').new()
+			contract.upgradeTo(newImpl.address)
+			const [implementation] = await Promise.all([
+				getEventValue(contract)('Upgraded', 'implementation'),
+			])
+			expect(implementation).to.equal(newImpl.address)
+			expect((await contract.getCounter()).toString()).to.equal('1')
+			const eSet = await contract.getEnumerableSet()
+			expect(eSet.length).to.equal(2)
+			const converted = eSet.map((v) => v.toNumber())
+			expect(converted.includes(10)).to.equal(true)
+			expect(converted.includes(20)).to.equal(true)
+			expect((await contract.testValue()).toString()).to.equal('5')
+		})
+	})
+})