class Testcases(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.grv = Gravity(
nobroadcast=True,
# We want to bundle many operations into a single transaction
bundle=True,
# Overwrite wallet to use this list of wifs only
wif=[wif])
self.grv.set_default_account("init0")
set_shared_gravity_instance(self.grv)
def test_account(self):
Account("witness-account")
Account("1.2.3")
asset = Asset("1.3.0")
symbol = asset["symbol"]
account = Account("witness-account", full=True)
self.assertEqual(account.name, "witness-account")
self.assertEqual(account["name"], account.name)
self.assertEqual(account["id"], "1.2.1")
self.assertIsInstance(account.balance("1.3.0"), Amount)
self.assertIsInstance(account.balance({"symbol": symbol}), Amount)
self.assertIsInstance(account.balances, list)
for h in account.history(limit=1):
pass
# BlockchainObjects method
account.cached = False
self.assertTrue(account.items())
account.cached = False
self.assertIn("id", account)
account.cached = False
self.assertEqual(account["id"], "1.2.1")
self.assertEqual(str(account), "<Account 1.2.1>")
self.assertIsInstance(Account(account), Account)
def test_account_upgrade(self):
account = Account("witness-account")
tx = account.upgrade()
ops = tx["operations"]
op = ops[0][1]
self.assertEqual(len(ops), 1)
self.assertEqual(getOperationNameForId(ops[0][0]), "account_upgrade")
self.assertTrue(op["upgrade_to_lifetime_member"])
self.assertEqual(
op["account_to_upgrade"],
"1.2.1",
)
class Testcases(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.grv = Gravity(
nobroadcast=True,
# We want to bundle many operations into a single transaction
bundle=True,
# Overwrite wallet to use this list of wifs only
wif=[wif],
offline=True
)
self.grv.set_default_account("init0")
set_shared_gravity_instance(self.grv)
def test_get_pub_from_wif(self):
self.assertEqual(
self.grv.wallet._get_pub_from_wif(wif),
"ZGV6MRyAjQq8ud7hVNYcfnVPJqcVpscN5So8BhtHuGYqET5GDW5CV"
)
class Testcases(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.grv = Gravity(nobroadcast=True, wif=[wif])
# from getpass import getpass
# self.grv.wallet.unlock(getpass())
set_shared_gravity_instance(self.grv)
self.grv.set_default_account("init0")
def test_finalizeOps_proposal(self):
grv = self.grv
# proposal = grv.new_proposal(grv.tx())
proposal = grv.proposal()
self.grv.transfer("init1", 1, "ZGV", append_to=proposal)
tx = grv.tx().json() # default tx buffer
ops = tx["operations"]
self.assertEqual(len(ops), 1)
self.assertEqual(getOperationNameForId(ops[0][0]), "proposal_create")
prop = ops[0][1]
self.assertEqual(len(prop["proposed_ops"]), 1)
self.assertEqual(
getOperationNameForId(prop["proposed_ops"][0]["op"][0]),
"transfer")
def test_finalizeOps_proposal2(self):
grv = self.grv
proposal = grv.new_proposal()
# proposal = grv.proposal()
self.grv.transfer("init1", 1, "ZGV", append_to=proposal)
tx = grv.tx().json() # default tx buffer
ops = tx["operations"]
self.assertEqual(len(ops), 1)
self.assertEqual(getOperationNameForId(ops[0][0]), "proposal_create")
prop = ops[0][1]
self.assertEqual(len(prop["proposed_ops"]), 1)
self.assertEqual(
getOperationNameForId(prop["proposed_ops"][0]["op"][0]),
"transfer")
def test_finalizeOps_combined_proposal(self):
grv = self.grv
parent = grv.new_tx()
proposal = grv.new_proposal(parent)
self.grv.transfer("init1", 1, "ZGV", append_to=proposal)
self.grv.transfer("init1", 1, "ZGV", append_to=parent)
tx = parent.json()
ops = tx["operations"]
self.assertEqual(len(ops), 2)
self.assertEqual(getOperationNameForId(ops[0][0]), "proposal_create")
self.assertEqual(getOperationNameForId(ops[1][0]), "transfer")
prop = ops[0][1]
self.assertEqual(len(prop["proposed_ops"]), 1)
self.assertEqual(
getOperationNameForId(prop["proposed_ops"][0]["op"][0]),
"transfer")
def test_finalizeOps_changeproposer_new(self):
grv = self.grv
proposal = grv.proposal(proposer="init5")
grv.transfer("init1", 1, "ZGV", append_to=proposal)
tx = grv.tx().json()
ops = tx["operations"]
self.assertEqual(len(ops), 1)
self.assertEqual(getOperationNameForId(ops[0][0]), "proposal_create")
prop = ops[0][1]
self.assertEqual(len(prop["proposed_ops"]), 1)
self.assertEqual(prop["fee_paying_account"], "1.2.12")
self.assertEqual(
getOperationNameForId(prop["proposed_ops"][0]["op"][0]),
"transfer")
def test_finalizeOps_changeproposer_legacy(self):
grv = self.grv
grv.proposer = "init5"
tx = grv.transfer("init1", 1, "ZGV")
ops = tx["operations"]
self.assertEqual(len(ops), 1)
self.assertEqual(getOperationNameForId(ops[0][0]), "proposal_create")
prop = ops[0][1]
self.assertEqual(len(prop["proposed_ops"]), 1)
self.assertEqual(prop["fee_paying_account"], "1.2.12")
self.assertEqual(
getOperationNameForId(prop["proposed_ops"][0]["op"][0]),
"transfer")
def test_new_proposals(self):
grv = self.grv
p1 = grv.new_proposal()
p2 = grv.new_proposal()
self.assertIsNotNone(id(p1), id(p2))
def test_new_txs(self):
grv = self.grv
p1 = grv.new_tx()
p2 = grv.new_tx()
self.assertIsNotNone(id(p1), id(p2))
class Testcases(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.grv = Gravity(
nobroadcast=True,
wif={"active": wif}
)
set_shared_gravity_instance(self.grv)
self.grv.set_default_account("init0")
self.mock()
def mock(self):
# Inject test data into cache
_cache = ObjectCache(default_expiration=60 * 60 * 1, no_overwrite=True)
for i in _objects:
_cache[i["id"]] = i
BlockchainObject._cache = _cache
def test_connect(self):
self.grv.connect()
def test_set_default_account(self):
self.grv.set_default_account("init0")
def test_info(self):
info = self.grv.info()
for key in ['current_witness',
'head_block_id',
'head_block_number',
'id',
'last_irreversible_block_num',
'next_maintenance_time',
'random',
'recently_missed_count',
'time']:
self.assertTrue(key in info)
def test_finalizeOps(self):
grv = self.grv
tx1 = grv.new_tx()
tx2 = grv.new_tx()
self.grv.transfer("init1", 1, core_unit, append_to=tx1)
self.grv.transfer("init1", 2, core_unit, append_to=tx2)
self.grv.transfer("init1", 3, core_unit, append_to=tx1)
tx1 = tx1.json()
tx2 = tx2.json()
ops1 = tx1["operations"]
ops2 = tx2["operations"]
self.assertEqual(len(ops1), 2)
self.assertEqual(len(ops2), 1)
def test_transfer(self):
grv = self.grv
tx = grv.transfer(
"1.2.8", 1.33, core_unit, memo="Foobar", account="1.2.7")
self.assertEqual(
getOperationNameForId(tx["operations"][0][0]),
"transfer"
)
op = tx["operations"][0][1]
self.assertIn("memo", op)
self.assertEqual(op["from"], "1.2.7")
self.assertEqual(op["to"], "1.2.8")
amount = Amount(op["amount"])
self.assertEqual(float(amount), 1.33)
def test_create_account(self):
grv = self.grv
name = ''.join(random.choice(string.ascii_lowercase) for _ in range(12))
key1 = PrivateKey()
key2 = PrivateKey()
key3 = PrivateKey()
key4 = PrivateKey()
tx = grv.create_account(
name,
registrar="init0", # 1.2.7
referrer="init1", # 1.2.8
referrer_percent=33,
owner_key=str(key1.pubkey),
active_key=str(key2.pubkey),
memo_key=str(key3.pubkey),
additional_owner_keys=[str(key4.pubkey)],
additional_active_keys=[str(key4.pubkey)],
additional_owner_accounts=["committee-account"], # 1.2.0
additional_active_accounts=["committee-account"],
proxy_account="init0",
storekeys=False
)
self.assertEqual(
getOperationNameForId(tx["operations"][0][0]),
"account_create"
)
op = tx["operations"][0][1]
role = "active"
self.assertIn(
str(key2.pubkey),
[x[0] for x in op[role]["key_auths"]])
self.assertIn(
str(key4.pubkey),
[x[0] for x in op[role]["key_auths"]])
self.assertIn(
"1.2.0",
[x[0] for x in op[role]["account_auths"]])
role = "owner"
self.assertIn(
str(key1.pubkey),
[x[0] for x in op[role]["key_auths"]])
self.assertIn(
str(key4.pubkey),
[x[0] for x in op[role]["key_auths"]])
self.assertIn(
"1.2.0",
[x[0] for x in op[role]["account_auths"]])
self.assertEqual(
op["options"]["voting_account"],
"1.2.7")
self.assertEqual(
op["registrar"],
"1.2.7")
self.assertEqual(
op["referrer"],
"1.2.8")
self.assertEqual(
op["referrer_percent"],
33 * 100)
def test_weight_threshold(self):
grv = self.grv
auth = {'account_auths': [['1.2.0', '1']],
'extensions': [],
'key_auths': [
['ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n', 1],
['ZGV7GM9YXcsoAJAgKbqW2oVj7bnNXFNL4pk9NugqKWPmuhoEDbkDv', 1]],
'weight_threshold': 3} # threshold fine
grv._test_weights_treshold(auth)
auth = {'account_auths': [['1.2.0', '1']],
'extensions': [],
'key_auths': [
['ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n', 1],
['ZGV7GM9YXcsoAJAgKbqW2oVj7bnNXFNL4pk9NugqKWPmuhoEDbkDv', 1]],
'weight_threshold': 4} # too high
with self.assertRaises(ValueError):
grv._test_weights_treshold(auth)
def test_allow(self):
grv = self.grv
tx = grv.allow(
"ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n",
weight=1,
threshold=1,
permission="owner"
)
self.assertEqual(
getOperationNameForId(tx["operations"][0][0]),
"account_update"
)
op = tx["operations"][0][1]
self.assertIn("owner", op)
self.assertIn(
["ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n", '1'],
op["owner"]["key_auths"])
self.assertEqual(op["owner"]["weight_threshold"], 1)
""" Disable this test until we can test with an actual setup on the
main/testnet
def test_disallow(self):
grv = self.grv
with self.assertRaisesRegex(ValueError, ".*Changes nothing.*"):
grv.disallow(
"ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n",
weight=1,
threshold=1,
permission="owner"
)
with self.assertRaisesRegex(ValueError, ".*Cannot have threshold of 0.*"):
grv.disallow(
"ZGV6MRyAjQq8ud7hVNYcfnVPJqcVpscN5So8BhtHuGYqET5GDW5CV",
weight=1,
threshold=1,
permission="owner"
)
"""
def test_update_memo_key(self):
grv = self.grv
tx = grv.update_memo_key("ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n")
self.assertEqual(
getOperationNameForId(tx["operations"][0][0]),
"account_update"
)
op = tx["operations"][0][1]
self.assertEqual(
op["new_options"]["memo_key"],
"ZGV55VCzsb47NZwWe5F3qyQKedX9iHBHMVVFSc96PDvV7wuj7W86n")
def test_approvewitness(self):
grv = self.grv
tx = grv.approvewitness("1.6.1")
self.assertEqual(
getOperationNameForId(tx["operations"][0][0]),
"account_update"
)
op = tx["operations"][0][1]
self.assertIn(
"1:0",
op["new_options"]["votes"])
def test_approvecommittee(self):
grv = self.grv
tx = grv.approvecommittee("1.5.0")
self.assertEqual(
getOperationNameForId(tx["operations"][0][0]),
"account_update"
)
op = tx["operations"][0][1]
self.assertIn(
"0:11",
op["new_options"]["votes"])
class Testcases(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.grv = Gravity(
nobroadcast=True,
wif={
# Force signing with this key
"active": wif
})
set_shared_gravity_instance(self.grv)
self.grv.set_default_account("init0")
def test_add_one_proposal_one_op(self):
grv = self.grv
tx1 = grv.new_tx()
proposal1 = grv.new_proposal(tx1, proposer="init0")
op = operations.Transfer(
**{
"fee": {
"amount": 0,
"asset_id": "1.3.0"
},
"from": "1.2.0",
"to": "1.2.0",
"amount": {
"amount": 0,
"asset_id": "1.3.0"
},
"prefix": "ZGV"
})
proposal1.appendOps(op)
tx = tx1.json()
self.assertEqual(tx["operations"][0][0], 22)
self.assertEqual(len(tx["operations"]), 1)
ps = tx["operations"][0][1]
self.assertEqual(len(ps["proposed_ops"]), 1)
self.assertEqual(ps["proposed_ops"][0]["op"][0], 0)
def test_add_one_proposal_two_ops(self):
grv = self.grv
tx1 = grv.new_tx()
proposal1 = grv.new_proposal(tx1, proposer="init0")
op = operations.Transfer(
**{
"fee": {
"amount": 0,
"asset_id": "1.3.0"
},
"from": "1.2.0",
"to": "1.2.0",
"amount": {
"amount": 0,
"asset_id": "1.3.0"
},
"prefix": "ZGV"
})
proposal1.appendOps(op)
proposal1.appendOps(op)
tx = tx1.json()
self.assertEqual(tx["operations"][0][0], 22)
self.assertEqual(len(tx["operations"]), 1)
ps = tx["operations"][0][1]
self.assertEqual(len(ps["proposed_ops"]), 2)
self.assertEqual(ps["proposed_ops"][0]["op"][0], 0)
self.assertEqual(ps["proposed_ops"][1]["op"][0], 0)
def test_have_two_proposals(self):
grv = self.grv
tx1 = grv.new_tx()
# Proposal 1
proposal1 = grv.new_proposal(tx1, proposer="init0")
op = operations.Transfer(
**{
"fee": {
"amount": 0,
"asset_id": "1.3.0"
},
"from": "1.2.0",
"to": "1.2.0",
"amount": {
"amount": 0,
"asset_id": "1.3.0"
},
"prefix": "ZGV"
})
for i in range(0, 3):
proposal1.appendOps(op)
# Proposal 1
proposal2 = grv.new_proposal(tx1, proposer="init0")
op = operations.Transfer(
**{
"fee": {
"amount": 0,
"asset_id": "1.3.0"
},
"from": "1.2.0",
"to": "1.2.0",
"amount": {
"amount": 5555555,
"asset_id": "1.3.0"
},
"prefix": "ZGV"
})
for i in range(0, 2):
proposal2.appendOps(op)
tx = tx1.json()
self.assertEqual(len(tx["operations"]), 2) # 2 proposals
# Test proposal 1
prop = tx["operations"][0]
self.assertEqual(prop[0], 22)
ps = prop[1]
self.assertEqual(len(ps["proposed_ops"]), 3)
for i in range(0, 3):
self.assertEqual(ps["proposed_ops"][i]["op"][0], 0)
# Test proposal 2
prop = tx["operations"][1]
self.assertEqual(prop[0], 22)
ps = prop[1]
self.assertEqual(len(ps["proposed_ops"]), 2)
for i in range(0, 2):
self.assertEqual(ps["proposed_ops"][i]["op"][0], 0)
