def test_node_owner(self):
import time
from datetime import datetime
from zilpool.database.zilnode import ZilNode, ZilNodeOwner
drop_all()
email = "*****@*****.**"
owner = ZilNodeOwner.create(email)
assert owner is not None
assert owner.email_verified is False
assert owner.pow_fee == 0.0
assert owner.balance == 0.0
assert owner.join_date <= datetime.utcnow()
key = ZilKey.generate_key_pair()
pub_key = key.keypair_str.public
node = owner.register_node(pub_key)
assert pub_key in owner.pending_nodes
node = ZilNode.get_by_pub_key(pub_key=pub_key, authorized=False)
assert node is not None
assert node.authorized is False
assert node.pow_fee == owner.pow_fee
def test_zil_nodes(self):
from zilpool.database.zilnode import ZilNode
drop_all()
def check_doc_count(i):
assert ZilNode._collection.count_documents({}) == i
key = ZilKey.generate_key_pair()
pub_key = key.keypair_str.public
node = ZilNode(pub_key=pub_key, pow_fee=0, authorized=False)
node.save()
check_doc_count(1)
node.update(set__pow_fee=1.2)
check_doc_count(1)
node2 = ZilNode.get_by_pub_key(pub_key, authorized=False)
assert node2.pow_fee == node.pow_fee == 1.2
node = ZilNode.get_by_pub_key("")
assert node is None
node = ZilNode(pub_key=pub_key)
success = node.save()
assert not success
check_doc_count(1)
fail_pub_key = rand_hex_str(9)
node2 = ZilNode(pub_key=fail_pub_key, pow_fee=6, authorized=True)
check_doc_count(1)
success = node2.save()
assert success is node2
check_doc_count(2)
success = node2.save()
assert success is node2
check_doc_count(2)
drop_all()
def admin_auth_node(pub_key, approve):
from zilpool.web import tools
node = ZilNode.get_by_pub_key(pub_key, authorized=None)
assert node, "Node not found"
if approve != node.authorized:
node = node.update(authorized=approve)
assert node, "failed to update database"
action = "Approved" if approve else "Revoked"
messages = f"Node Register {action}: {node.pub_key}"
logging.info(messages)
if node.email:
tools.send_auth_notification_email(node.email, messages=messages)
return node