Example #1
0
class TorWrapperTest(unittest.TestCase):

    def __init__(self, arg):
        super(TorWrapperTest, self).__init__(arg)
        self.env = TorEnvironment()
        self.tor = Tor(self.env)

        self.base_dir = os.path.abspath(os.path.join(os.environ['NETWORK_BASE_DIR'], 'tor'))
    
    def setUp(self):
        pass
        

    def tearDown(self):
        key_folder = os.path.join(self.base_dir, 'keys')
        #shutil.rmtree(key_folder)

    def setup_keys(self):
        p = self.tor.gen_authority_key(stdin=sp.PIPE)
        p.communicate(b'password\n')

    def test_gen_authority_key(self):
        key_folder = os.path.join(self.base_dir, 'keys')
        cmd1 = self.tor.gen_authority_key(execute=False)
        cmd2 = 'tor-gencert --create-identity-key --passphrase-fd 0 -i {0}/{1} -s {2}/{3} -c {4}/{5} -a {6}'
        cmd2 = cmd2.format(key_folder, 'authority_identity_key',
                           key_folder, 'authority_signing_key',
                           key_folder, 'authority_certificate',
                           '10.0.0.2:9000')
        self.maxDiff = None
        self.assertEqual(cmd1, cmd2)

        p = self.tor.gen_authority_key(stdin=sp.PIPE)
        p.communicate(b'password\n')
        self.assertTrue(os.path.exists(self.tor.env.abs_id_file()))
        
        
    def test_get_fingerprint_from_cert_file(self):
        self.setup_keys()
        fp = self.tor.env.fingerprint_from_cert_file()
        self.assertEqual(len(fp), 40)

    def test_gen_fingerprint(self):
        self.setup_keys()
        p = self.tor.gen_fingerprint(stdin=sp.PIPE)
        p.communicate(b'\n')
        self.assertEqual(len(self.tor.env.fingerprint_from_file()), 40)
Example #2
0
    def __init__(self, arg):
        super(TorWrapperTest, self).__init__(arg)
        self.env = TorEnvironment()
        self.tor = Tor(self.env)

        self.base_dir = os.path.abspath(os.path.join(os.environ['NETWORK_BASE_DIR'], 'tor'))