nucypher/tests/integration/network/test_network_upgrade.py

34 lines
1.1 KiB
Python

from pathlib import Path
import pytest_twisted
import requests
from cryptography.hazmat.primitives import serialization
from twisted.internet import threads
from nucypher.characters.lawful import Ursula
@pytest_twisted.inlineCallbacks
def test_nodes_connect_via_tls_and_verify(lonely_ursula_maker):
node = lonely_ursula_maker(quantity=1).pop()
node_deployer = node.get_deployer()
node_deployer.addServices()
node_deployer.catalogServers(node_deployer.hendrix)
node_deployer.start()
cert = node_deployer.cert.to_cryptography()
cert_bytes = cert.public_bytes(serialization.Encoding.PEM)
def check_node_with_cert(node, cert_file):
response = requests.get("https://{}/public_information".format(node.rest_url()), verify=cert_file)
ursula = Ursula.from_metadata_bytes(response.content)
assert ursula == node
try:
with open("test-cert", "wb") as f:
f.write(cert_bytes)
yield threads.deferToThread(check_node_with_cert, node, "test-cert")
finally:
Path("test-cert").unlink()