Updated link tests

This commit is contained in:
Mark Qvist 2022-06-09 19:54:20 +02:00
parent 2c1b3a0e5b
commit ec90af750d

View File

@ -1,5 +1,7 @@
import unittest import unittest
import subprocess
import shlex
import threading import threading
import time import time
import RNS import RNS
@ -15,20 +17,34 @@ fixed_keys = [
("08bb35f92b06a0832991165a0d9b4fd91af7b7765ce4572aa6222070b11b767092b61b0fd18b3a59cae6deb9db6d4bfb1c7fcfe076cfd66eea7ddd5f877543b9", "d13712efc45ef87674fb"), ("08bb35f92b06a0832991165a0d9b4fd91af7b7765ce4572aa6222070b11b767092b61b0fd18b3a59cae6deb9db6d4bfb1c7fcfe076cfd66eea7ddd5f877543b9", "d13712efc45ef87674fb"),
] ]
def targets_job(caller):
cmd = "python -c \"from tests.link import targets; targets()\""
print("Opening subprocess for "+str(caller)+"...", RNS.LOG_VERBOSE)
ppath = os.getcwd()
try:
caller.process = subprocess.Popen(shlex.split(cmd), cwd=ppath)
except Exception as e:
raise e
caller.pipe_is_open = False
c_rns = None c_rns = None
def init_rns(): def init_rns(caller=None):
global c_rns global c_rns
if c_rns == None: if c_rns == None:
targets_job(caller)
time.sleep(2)
print("Starting local RNS instance...") print("Starting local RNS instance...")
c_rns = RNS.Reticulum("./tests/rnsconfig") c_rns = RNS.Reticulum("./tests/rnsconfig")
c_rns.m_proc = caller.process
print("Done starting local RNS instance...") print("Done starting local RNS instance...")
class TestLink(unittest.TestCase): class TestLink(unittest.TestCase):
def setUp(self): def setUp(self):
pass pass
def test_establish(self): def test_0_establish(self):
init_rns() init_rns(self)
print("") print("")
id1 = RNS.Identity.from_bytes(bytes.fromhex(fixed_keys[0][0])) id1 = RNS.Identity.from_bytes(bytes.fromhex(fixed_keys[0][0]))
@ -45,8 +61,8 @@ class TestLink(unittest.TestCase):
time.sleep(0.5) time.sleep(0.5)
self.assertEqual(l1.status, RNS.Link.CLOSED) self.assertEqual(l1.status, RNS.Link.CLOSED)
def test_packets(self): def test_1_packets(self):
init_rns() init_rns(self)
print("") print("")
# TODO: Load this from public bytes only # TODO: Load this from public bytes only
@ -97,8 +113,8 @@ class TestLink(unittest.TestCase):
time.sleep(0.5) time.sleep(0.5)
self.assertEqual(l1.status, RNS.Link.CLOSED) self.assertEqual(l1.status, RNS.Link.CLOSED)
def test_micro_resource(self): def test_2_micro_resource(self):
init_rns() init_rns(self)
print("") print("")
print("Micro resource test") print("Micro resource test")
@ -131,8 +147,8 @@ class TestLink(unittest.TestCase):
time.sleep(0.5) time.sleep(0.5)
self.assertEqual(l1.status, RNS.Link.CLOSED) self.assertEqual(l1.status, RNS.Link.CLOSED)
def test_small_resource(self): def test_3_small_resource(self):
init_rns() init_rns(self)
print("") print("")
print("Small resource test") print("Small resource test")
@ -166,8 +182,8 @@ class TestLink(unittest.TestCase):
self.assertEqual(l1.status, RNS.Link.CLOSED) self.assertEqual(l1.status, RNS.Link.CLOSED)
def test_medium_resource(self): def test_4_medium_resource(self):
init_rns() init_rns(self)
print("") print("")
print("Medium resource test") print("Medium resource test")
@ -200,8 +216,8 @@ class TestLink(unittest.TestCase):
time.sleep(0.5) time.sleep(0.5)
self.assertEqual(l1.status, RNS.Link.CLOSED) self.assertEqual(l1.status, RNS.Link.CLOSED)
def test_large_resource(self): def test_5_large_resource(self):
init_rns() init_rns(self)
print("") print("")
print("Large resource test") print("Large resource test")
@ -257,8 +273,6 @@ class TestLink(unittest.TestCase):
if __name__ == '__main__': if __name__ == '__main__':
unittest.main(verbosity=1) unittest.main(verbosity=1)
def targets_job():
cmd = "python -c \"from test.link import targets; targets()\""
def targets(): def targets():
def resource_started(resource): def resource_started(resource):
@ -283,7 +297,8 @@ def targets():
d1.set_proof_strategy(RNS.Destination.PROVE_ALL) d1.set_proof_strategy(RNS.Destination.PROVE_ALL)
d1.set_link_established_callback(link_established) d1.set_link_established_callback(link_established)
input() while True:
time.sleep(1)
def targets_profiling(): def targets_profiling():
targets() targets()