32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
|
import sys, os
|
||
|
sys.path.append(".")
|
||
|
sys.path.append("src/")
|
||
|
import unittest, uuid
|
||
|
|
||
|
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
|
||
|
import onionrsetup as setup
|
||
|
from netcontroller import customtorrc
|
||
|
createdirs.create_dirs()
|
||
|
setup.setup_config()
|
||
|
|
||
|
class TestCustomTorrc(unittest.TestCase):
|
||
|
def test_torrc_get(self):
|
||
|
torrc = identifyhome.identify_home() + '/torrc-custom'
|
||
|
self.assertEqual(customtorrc.get_custom_torrc(), '\n')
|
||
|
with open(torrc, 'w') as torrc_file:
|
||
|
torrc_file.write('test')
|
||
|
self.assertEqual(customtorrc.get_custom_torrc(), '\ntest')
|
||
|
os.remove(torrc)
|
||
|
|
||
|
def test_torrc_set(self):
|
||
|
torrc = identifyhome.identify_home() + '/torrc-custom'
|
||
|
customtorrc.set_custom_torrc('test2')
|
||
|
with open(torrc, 'r') as torrc_file:
|
||
|
self.assertEqual(torrc_file.read().splitlines()[2], 'test2')
|
||
|
os.remove(torrc)
|
||
|
|
||
|
unittest.main()
|