35 lines
959 B
Python
35 lines
959 B
Python
|
import sys, os
|
||
|
|
sys.path.append(".")
|
||
|
|
sys.path.append("src/")
|
||
|
|
import unittest, uuid
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
|
||
|
|
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
|
||
|
|
print("Test directory:", TEST_DIR)
|
||
|
|
os.environ["ONIONR_HOME"] = TEST_DIR
|
||
|
|
from utils import createdirs, identifyhome
|
||
|
|
from streamfill import identify_neighbors
|
||
|
|
|
||
|
|
onions = []
|
||
|
|
p = subprocess.Popen(["scripts/generate-onions.py", '50000'],
|
||
|
|
stdout=subprocess.PIPE,
|
||
|
|
stderr=subprocess.PIPE)
|
||
|
|
for line in iter(p.stdout.readline, b''):
|
||
|
|
line = line.decode()
|
||
|
|
onions.append(line.strip())
|
||
|
|
p.terminate()
|
||
|
|
|
||
|
|
p = subprocess.Popen(["scripts/generate-onions.py", '1'],
|
||
|
|
stdout=subprocess.PIPE,
|
||
|
|
stderr=subprocess.PIPE)
|
||
|
|
for line in iter(p.stdout.readline, b''):
|
||
|
|
us = line.decode().strip()
|
||
|
|
p.terminate()
|
||
|
|
|
||
|
|
start = time.time()
|
||
|
|
identify_neighbors(us, onions, 5)
|
||
|
|
print(time.time() - start)
|
||
|
|
|
||
|
|
|