import os, uuid
from random import randint
from sqlite3 import Time
import socket
from queue import Queue
from time import sleep
import secrets


TEST_DIR = 'testdata/%s-%s' % (str(uuid.uuid4())[:6], os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR

import unittest
import sys
sys.path.append(".")
sys.path.append('static-data/default-plugins/wot/')
sys.path.append("src/")
from wot import identity


def generate_graph(iden: identity.Identity, depth, max_neighbors):
    c = 0
    if depth == 0:
        return
    for i in range(randint(0, max_neighbors)):
        i = identity.Identity(secrets.token_hex(16))
        iden.trusted.add(i)
        generate_graph(i, depth - 1, max_neighbors)


class IdentityDistanceTest(unittest.TestCase):
    def test_distance(self):
        iden = identity.Identity(secrets.token_hex(16))
        generate_graph(iden, 10, 5)
        iden2 = list(list(iden.trusted)[0].trusted)[0]

        self.assertEqual(identity.get_distance(iden, iden2), 2)

unittest.main()