pywiki/tests/sdk-pyhmy/test_validator.py

import pytest
from decimal import Decimal

from pyhmy import validator

from pyhmy.numbers import convert_one_to_atto

from pyhmy.exceptions import InvalidValidatorError

test_epoch_number = 0
genesis_block_number = 0
test_block_number = 1
test_validator_object = None
test_validator_loaded = False


def test_instantiate_validator(setup_blockchain):
    global test_validator_object
    test_validator_object = validator.Validator(
        "one1a0x3d6xpmr6f8wsyaxd9v36pytvp48zckswvv9"
    )
    assert isinstance(test_validator_object, validator.Validator)


def test_load_validator(setup_blockchain):
    if not test_validator_object:
        pytest.skip("Validator not instantiated yet")
    info = {
        "name": "Alice",
        "identity": "alice",
        "website": "alice.harmony.one",
        "details": "Don't mess with me!!!",
        "security-contact": "Bob",
        "min-self-delegation": convert_one_to_atto(10000),
        "amount": convert_one_to_atto(10001),
        "max-rate": "0.9",
        "max-change-rate": "0.05",
        "rate": "0.01",
        "bls-public-keys": [
            "0x30b2c38b1316da91e068ac3bd8751c0901ef6c02a1d58bc712104918302c6ed03d5894671d0c816dad2b4d303320f202"
        ],
        "bls-key-sigs": [
            "0x68f800b6adf657b674903e04708060912b893b7c7b500788808247550ab3e186e56a44ebf3ca488f8ed1a42f6cef3a04bd5d2b2b7eb5a767848d3135b362e668ce6bba42c7b9d5666d8e3a83be707b5708e722c58939fe9b07c170f3b7062414"
        ],
        "max-total-delegation": convert_one_to_atto(40000),
    }
    test_validator_object.load(info)
    global test_validator_loaded
    test_validator_loaded = True


"""
TypeScript signature source (is outdated because the JS SDK has not been updated for SlotKeySigs)
For now I have checked that the below transaction to localnet works
---
const description: Description = new Description('Alice', 'alice', 'alice.harmony.one', 'Bob', "Don't mess with me!!!")
const commissionRates: CommissionRate = new CommissionRate(new Decimal('0.01'), new Decimal('0.9'), new Decimal('0.05'))
const stakeMsg: CreateValidator = new CreateValidator(
        'one1a0x3d6xpmr6f8wsyaxd9v36pytvp48zckswvv9',
        description,
        commissionRates,
        numberToHex(new Unit('10000').asOne().toWei()),    // minSelfDelegation
        numberToHex(new Unit('40000').asOne().toWei()),    // maxTotalDelegation
        [ '0xb9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b62247608611' ],
        numberToHex(new Unit('10001').asOne().toWei())    // amount
  )
const stakingTx: StakingTransaction = new StakingTransaction(
  Directive.DirectiveCreateValidator,
  stakeMsg,
  2,    // nonce
  numberToHex(new Unit('1').asOne().toWei()),    // gasPrice
  100,      // gasLimit
  null,     // chainId
);
const signed = stakingTx.rlpSign('4edef2c24995d15b0e25cbd152fb0e2c05d3b79b9c2afd134e6f59f91bf99e48')
console.log( 'Signed transaction' )
console.log(signed)
"""


def test_create_validator_sign(setup_blockchain):
    if not (test_validator_object or test_validator_loaded):
        pytest.skip("Validator not instantiated yet")
    signed_hash = test_validator_object.sign_create_validator_transaction(
        2,
        int(convert_one_to_atto(1)),
        100,
        "4edef2c24995d15b0e25cbd152fb0e2c05d3b79b9c2afd134e6f59f91bf99e48",
        2,
    ).rawTransaction.hex()
    assert (
        signed_hash
        == "0xf9017580f9012394ebcd16e8c1d8f493ba04e99a56474122d81a9c58f83885416c69636585616c69636591616c6963652e6861726d6f6e792e6f6e6583426f6295446f6e2774206d6573732077697468206d65212121dcc8872386f26fc10000c9880c7d713b49da0000c887b1a2bc2ec500008a021e19e0c9bab24000008a0878678326eac9000000f1b030b2c38b1316da91e068ac3bd8751c0901ef6c02a1d58bc712104918302c6ed03d5894671d0c816dad2b4d303320f202f862b86068f800b6adf657b674903e04708060912b893b7c7b500788808247550ab3e186e56a44ebf3ca488f8ed1a42f6cef3a04bd5d2b2b7eb5a767848d3135b362e668ce6bba42c7b9d5666d8e3a83be707b5708e722c58939fe9b07c170f3b70624148a021e27c1806e59a4000002880de0b6b3a76400006428a0c6c7e62f02331df0afd4699ec514a2fc4548c920d77ad74d98caeec8c924c09aa02b27b999a724b1d341d6bbb0e877611d0047542cb7e380f9a6a272d204b450cd"
    )


"""
Signature matched from TypeScript (is outdated because the JS SDK has not been updated for SlotKeyToAddSig)
For now I have checked that the below transaction to localnet works
---
import {
  CreateValidator,
  EditValidator,
  Delegate,
  Undelegate,
  CollectRewards,
  Directive,
  Description,
  CommissionRate,
  Decimal,
  StakingTransaction,
} from '@harmony-js/staking'
const { numberToHex, Unit } = require('@harmony-js/utils');

const description: Description = new Description('Alice', 'alice', 'alice.harmony.one', 'Bob', "Don't mess with me!!!")
const commissionRates: CommissionRate = new CommissionRate(new Decimal('0.01'), new Decimal('0.9'), new Decimal('0.05'))
const stakeMsg: EditValidator = new EditValidator(
        'one1a0x3d6xpmr6f8wsyaxd9v36pytvp48zckswvv9',
        description,
        new Decimal('0.06'),
        numberToHex(new Unit('10000').asOne().toWei()),    // minSelfDelegation
        numberToHex(new Unit('40000').asOne().toWei()),    // maxTotalDelegation
        '0xb9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b62247608611', // remove key
        '0xb9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b62247608612'  // add key
  )
const stakingTx: StakingTransaction = new StakingTransaction(
  Directive.DirectiveEditValidator,
  stakeMsg,
  2,    // nonce
  numberToHex(new Unit('1').asOne().toWei()),    // gasPrice
  100,  // gasLimit
  2,    // chainId
);
const signed = stakingTx.rlpSign('4edef2c24995d15b0e25cbd152fb0e2c05d3b79b9c2afd134e6f59f91bf99e48')
console.log( 'Signed transaction' )
console.log(signed)
"""


def test_edit_validator_sign(setup_blockchain):
    if not (test_validator_object or test_validator_loaded):
        pytest.skip("Validator not instantiated yet")
    signed_hash = test_validator_object.sign_edit_validator_transaction(
        2,
        int(convert_one_to_atto(1)),
        100,
        "0.06",
        "0xb9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b62247608612",  # remove key
        "0xb9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b62247608611",  # add key
        "0x68f800b6adf657b674903e04708060912b893b7c7b500788808247550ab3e186e56a44ebf3ca488f8ed1a42f6cef3a04bd5d2b2b7eb5a767848d3135b362e668ce6bba42c7b9d5666d8e3a83be707b5708e722c58939fe9b07c170f3b7062414",  # add key sig
        "4edef2c24995d15b0e25cbd152fb0e2c05d3b79b9c2afd134e6f59f91bf99e48",
        2,
    ).rawTransaction.hex()
    assert (
        signed_hash
        == "0xf9012101f8d094ebcd16e8c1d8f493ba04e99a56474122d81a9c58f83885416c69636585616c69636591616c6963652e6861726d6f6e792e6f6e6583426f6295446f6e2774206d6573732077697468206d65212121c887d529ae9e8600008a021e19e0c9bab24000008a0878678326eac9000000b0b9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b62247608612b0b9486167ab9087ab818dc4ce026edb5bf216863364c32e42df2af03c5ced1ad181e7d12f0e6dd5307a73b6224760861102880de0b6b3a76400006428a0782ed9f909b5c68eb050a917a274d9187a4c8f13799f21ba351bdcb11290c6c7a00a3b4ec8431290565acbbbb8231cafe32ed7c0b6211e7cf570b459cb733e0995"
    )


def test_invalid_validator(setup_blockchain):
    if not (test_validator_object or test_validator_loaded):
        pytest.skip("Validator not instantiated yet")
    with pytest.raises(InvalidValidatorError):
        info = {
            "name": "Alice",
        }
        test_validator_object.load(info)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_name("a" * 141)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_identity("a" * 141)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_website("a" * 141)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_security_contact("a" * 141)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_details("a" * 281)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_min_self_delegation(1)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_max_total_delegation(1)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_amount(1)
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_max_rate("2.0")
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_max_change_rate("-2.0")
    with pytest.raises(InvalidValidatorError):
        test_validator_object.set_rate("-2.0")


def test_validator_getters(setup_blockchain):
    if not (test_validator_object or test_validator_loaded):
        pytest.skip("Validator not instantiated yet")
    assert (
        test_validator_object.get_address()
        == "one1a0x3d6xpmr6f8wsyaxd9v36pytvp48zckswvv9"
    )
    assert test_validator_object.add_bls_key("5")
    assert test_validator_object.remove_bls_key("5")
    assert test_validator_object.get_name() == "Alice"
    assert test_validator_object.get_identity() == "alice"
    assert test_validator_object.get_website() == "alice.harmony.one"
    assert test_validator_object.get_security_contact() == "Bob"
    assert test_validator_object.get_details() == "Don't mess with me!!!"
    assert isinstance(test_validator_object.get_min_self_delegation(), Decimal)
    assert isinstance(test_validator_object.get_max_total_delegation(), Decimal)
    assert isinstance(test_validator_object.get_amount(), Decimal)
    assert isinstance(test_validator_object.get_max_rate(), Decimal)
    assert isinstance(test_validator_object.get_max_change_rate(), Decimal)
    assert isinstance(test_validator_object.get_rate(), Decimal)
    assert len(test_validator_object.get_bls_keys()) > 0


def test_validator_load_from_blockchain(setup_blockchain):
    test_validator_object2 = validator.Validator(
        "one155jp2y76nazx8uw5sa94fr0m4s5aj8e5xm6fu3"
    )
    test_validator_object2.load_from_blockchain()