Test for multi-version contract deployment

pull/1470/head
vzotova 2019-12-02 15:44:39 +03:00
parent 98185a99c4
commit adb11d6785
5 changed files with 96 additions and 19 deletions

View File

@ -664,13 +664,13 @@ class BlockchainDeployerInterface(BlockchainInterface):
major, minor, patch = [int(v) for v in version[1:].split(".", 3)]
if current_version_parsed[0] == -1 or \
requested_version == 'latest' and \
(current_version_parsed[0] == -1 or
major > current_version_parsed[0] or
major == current_version_parsed[0] and minor > current_version_parsed[1] or
major == current_version_parsed[0] and minor == current_version_parsed[1] and patch > current_version_parsed[2]) or \
(major > current_version_parsed[0] or
major == current_version_parsed[0] and minor > current_version_parsed[1] or
major == current_version_parsed[0] and minor == current_version_parsed[1] and patch > current_version_parsed[2]) or \
requested_version == 'earliest' and \
(major < current_version_parsed[0] or
major == current_version_parsed[0] and minor < current_version_parsed[1] or
major == current_version_parsed[0] and minor == current_version_parsed[1] and patch < current_version_parsed[2]):
major == current_version_parsed[0] and minor < current_version_parsed[1] or
major == current_version_parsed[0] and minor == current_version_parsed[1] and patch < current_version_parsed[2]):
current_version_parsed = (major, minor, patch)
current_data = data
current_version = version

View File

@ -3,6 +3,8 @@ pragma solidity ^0.5.3;
/**
* @notice Test extracting version number from dev docs
* @dev |v1.2.3|
* @dev |v1.1.4|
*/
contract VersionTest {}
contract VersionTest {
uint16 public constant VERSION = 1;
}

View File

@ -3,6 +3,8 @@ pragma solidity ^0.5.3;
/**
* @notice Test extracting version number from dev docs
* @dev |v2.3.4|
* @dev |v1.2.3|
*/
contract VersionTest {}
contract VersionTest {
uint16 public constant VERSION = 2;
}

View File

@ -14,21 +14,24 @@ GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with nucypher. If not, see <https://www.gnu.org/licenses/>.
"""
import os
from os.path import dirname, abspath
import pytest
from nucypher.blockchain.eth.interfaces import BlockchainDeployerInterface
from nucypher.blockchain.eth.interfaces import BlockchainDeployerInterface, BlockchainInterfaceFactory
from nucypher.blockchain.eth.registry import InMemoryContractRegistry
# Prevents TesterBlockchain to be picked up by py.test as a test class
from nucypher.blockchain.eth.sol.compile import SolidityCompiler, SourceDirs
from nucypher.crypto.powers import TransactingPower
from nucypher.utilities.sandbox.blockchain import TesterBlockchain as _TesterBlockchain
from nucypher.utilities.sandbox.constants import (
DEVELOPMENT_ETH_AIRDROP_AMOUNT,
NUMBER_OF_ETH_TEST_ACCOUNTS,
NUMBER_OF_STAKERS_IN_BLOCKCHAIN_TESTS,
NUMBER_OF_URSULAS_IN_BLOCKCHAIN_TESTS,
TEST_PROVIDER_URI
)
TEST_PROVIDER_URI,
INSECURE_DEVELOPMENT_PASSWORD)
@pytest.fixture()
@ -86,3 +89,73 @@ def test_testerchain_creation(testerchain, another_testerchain):
tx = {'to': etherbase, 'from': account, 'value': 100}
txhash = chain.client.send_transaction(tx)
_receipt = chain.wait_for_receipt(txhash)
def test_multiversion_contract():
# Prepare compiler
base_dir = os.path.join(dirname(abspath(__file__)), "contracts", "multiversion")
v1_dir = os.path.join(base_dir, "v1")
v2_dir = os.path.join(base_dir, "v2")
root_dir = SolidityCompiler.default_contract_dir()
solidity_compiler = SolidityCompiler(source_dirs=[SourceDirs(root_dir, {v2_dir}),
SourceDirs(root_dir, {v1_dir})])
# Prepare chain
testerchain = _TesterBlockchain(free_transactions=True, compiler=solidity_compiler)
origin = testerchain.etherbase_account
BlockchainInterfaceFactory.register_interface(interface=testerchain)
testerchain.transacting_power = TransactingPower(password=INSECURE_DEVELOPMENT_PASSWORD,
account=origin)
testerchain.transacting_power.activate()
testerchain.connect()
# Searching both contract through raw data
contract_name = "VersionTest"
requested_version = "v1.2.3"
version, _data = testerchain.find_raw_contract_data(contract_name=contract_name,
requested_version=requested_version)
assert version == requested_version
version, _data = testerchain.find_raw_contract_data(contract_name=contract_name,
requested_version="latest")
assert version == requested_version
requested_version = "v1.1.4"
version, _data = testerchain.find_raw_contract_data(contract_name=contract_name,
requested_version=requested_version)
assert version == requested_version
version, _data = testerchain.find_raw_contract_data(contract_name=contract_name,
requested_version="earliest")
assert version == requested_version
# Deploy different contracts and check their versions
registry = InMemoryContractRegistry()
contract, receipt = testerchain.deploy_contract(deployer_address=origin,
registry=registry,
contract_name=contract_name,
contract_version="v1.1.4")
assert contract.version == "v1.1.4"
assert contract.functions.VERSION().call() == 1
contract, receipt = testerchain.deploy_contract(deployer_address=origin,
registry=registry,
contract_name=contract_name,
contract_version="earliest")
assert contract.version == "v1.1.4"
assert contract.functions.VERSION().call() == 1
contract, receipt = testerchain.deploy_contract(deployer_address=origin,
registry=registry,
contract_name=contract_name,
contract_version="v1.2.3")
assert contract.version == "v1.2.3"
assert contract.functions.VERSION().call() == 2
contract, receipt = testerchain.deploy_contract(deployer_address=origin,
registry=registry,
contract_name=contract_name,
contract_version="latest")
assert contract.version == "v1.2.3"
assert contract.functions.VERSION().call() == 2
contract, receipt = testerchain.deploy_contract(deployer_address=origin,
registry=registry,
contract_name=contract_name)
assert contract.version == "v1.2.3"
assert contract.functions.VERSION().call() == 2

View File

@ -27,8 +27,8 @@ def test_nucypher_contract_compiled(testerchain, test_registry):
origin, *everybody_else = testerchain.client.accounts
token_contract_identifier = NucypherTokenDeployer(registry=test_registry, deployer_address=origin).contract_name
assert token_contract_identifier in testerchain._BlockchainDeployerInterface__raw_contract_cache
token_data = testerchain._BlockchainDeployerInterface__raw_contract_cache[token_contract_identifier]
assert token_contract_identifier in testerchain._raw_contract_cache
token_data = testerchain._raw_contract_cache[token_contract_identifier]
assert len(token_data) == 1
assert "v0.0.0" in token_data
@ -39,7 +39,7 @@ def test_multi_source_compilation(testerchain):
(SolidityCompiler.default_contract_dir(), {TesterBlockchain.TEST_CONTRACTS_DIR})
])
interfaces = solidity_compiler.compile()
assert interfaces == testerchain._BlockchainDeployerInterface__raw_contract_cache
assert interfaces == testerchain._raw_contract_cache
def test_multi_versions():
@ -54,5 +54,5 @@ def test_multi_versions():
contract_data = interfaces["VersionTest"]
assert len(contract_data) == 2
assert "v1.2.3" in contract_data
assert "v2.3.4" in contract_data
assert contract_data["v1.2.3"]["devdoc"] != contract_data["v2.3.4"]["devdoc"]
assert "v1.1.4" in contract_data
assert contract_data["v1.2.3"]["devdoc"] != contract_data["v1.1.4"]["devdoc"]