67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys, os
|
||
|
|
import subprocess
|
||
|
|
sys.path.append(".")
|
||
|
|
sys.path.append("src/")
|
||
|
|
import uuid
|
||
|
|
import binascii
|
||
|
|
from base64 import b32decode
|
||
|
|
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
|
||
|
|
print("Test directory:", TEST_DIR)
|
||
|
|
os.environ["ONIONR_HOME"] = TEST_DIR
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from streamfill import identify_neighbors, extract_ed25519_from_onion_address
|
||
|
|
|
||
|
|
class TestStreamfillNeighbors(unittest.TestCase):
|
||
|
|
def test_neighbor_closeness_consistent(self):
|
||
|
|
onions = []
|
||
|
|
# p = subprocess.Popen(["scripts/generate-onions.py", '5'],
|
||
|
|
# stdout=subprocess.PIPE,
|
||
|
|
# stderr=subprocess.PIPE)
|
||
|
|
# for line in iter(p.stdout.readline, b''):
|
||
|
|
# line = line.decode()
|
||
|
|
# onions.append(line.strip())
|
||
|
|
main = '7uifxsgidchopmdwmtip6x4ydra6cpf2ov4ghj2lzx5uydyssduh5qid.onion'
|
||
|
|
others = ['bxxajpimlonmbxb5jzjre3go3dvfobqyayqwpksd6zpjz4s4mknstwyd.onion', '2zofaifd6s3flwbv5wl4vtgnesbprc4f2ptljl4a47dfkvrbmw3e5iqd.onion', '6umslj7jtzu27n4jgf3byn55ztz5mkoqocx32zwjya6rbnxqjpyysyyd.onion']
|
||
|
|
main_num = int.from_bytes(extract_ed25519_from_onion_address(main), 'big')
|
||
|
|
|
||
|
|
test_data = identify_neighbors(main, others, 3)
|
||
|
|
|
||
|
|
my_result = []
|
||
|
|
for i in others:
|
||
|
|
my_result.append((i, abs(main_num - int.from_bytes(extract_ed25519_from_onion_address(i), 'big'))))
|
||
|
|
my_result.sort()
|
||
|
|
final = []
|
||
|
|
for i in my_result:
|
||
|
|
final.append(i[0])
|
||
|
|
self.assertTrue(len(test_data) == 3)
|
||
|
|
self.assertListEqual(test_data, final)
|
||
|
|
|
||
|
|
def test_neighbor_closeness_random(self):
|
||
|
|
onions = []
|
||
|
|
p = subprocess.Popen(["scripts/generate-onions.py", '100'],
|
||
|
|
stdout=subprocess.PIPE,
|
||
|
|
stderr=subprocess.PIPE)
|
||
|
|
for line in iter(p.stdout.readline, b''):
|
||
|
|
line = line.decode()
|
||
|
|
onions.append(line.strip())
|
||
|
|
p.terminate()
|
||
|
|
main = '7uifxsgidchopmdwmtip6x4ydra6cpf2ov4ghj2lzx5uydyssduh5qid.onion'
|
||
|
|
main_num = int.from_bytes(extract_ed25519_from_onion_address(main), 'big')
|
||
|
|
|
||
|
|
test_data = identify_neighbors(main, onions, 100)
|
||
|
|
|
||
|
|
my_result = []
|
||
|
|
for i in onions:
|
||
|
|
my_result.append((i, abs(main_num - int.from_bytes(extract_ed25519_from_onion_address(i), 'big'))))
|
||
|
|
my_result.sort()
|
||
|
|
final = []
|
||
|
|
for i in my_result:
|
||
|
|
final.append(i[0])
|
||
|
|
self.assertTrue(len(test_data) == 100)
|
||
|
|
self.assertListEqual(test_data, final)
|
||
|
|
|
||
|
|
|
||
|
|
unittest.main()
|