class TestTunnelConf(TestStartingSingleTunnel):
"""
Run the same tests as TestStartingSingleTunnel with a tunnel
loaded from the configuration file. Also adds a few extra tests.
"""
def setUp(self):
self.executable = 'cal_run_forever'
conf = SafeConfigParser()
sec1 = 'tunnel:test'
conf.add_section(sec1)
conf.set(sec1, 'tunnel_type', 'base')
conf.set(sec1, 'cmd', self.executable)
conf.set(sec1, 'executable', self.executable)
self.tm = TunnelManager()
self.tm.load_tunnels(conf)
self.tm.start_tunnels()
self.t = self.tm.tunnels[0]
def test_already_loaded(self):
conf = SafeConfigParser()
self.assertRaises(TunnelsAlreadyLoadedException, self.tm.load_tunnels, *[conf])
def test_minimums(self):
self.conf = SafeConfigParser()
self.sec1 = 'tunnel:testvpnc'
self.conf.add_section(self.sec1)
self.conf.set(self.sec1, 'tunnel_type', 'vpnc')
self.conf.set(self.sec1, 'conf_file', '/path/to/conf.conf')
tm = TunnelManager()
tm.load_tunnels(self.conf)
self.assertEqual(len(tm.tunnels), 1)
class TestBaseParser(TearDownRunForever):
def setUp(self):
self.executable = 'cal_run_forever'
self.conf = SafeConfigParser()
self.sec1 = 'tunnel:test'
self.conf.add_section(self.sec1)
self.conf.set(self.sec1, 'tunnel_type', 'base')
self.conf.set(self.sec1, 'cmd', self.executable)
self.conf.set(self.sec1, 'executable', self.executable)
self.tm = TunnelManager()
def test_base_parse(self):
self.tm.load_tunnels(self.conf)
self.tm.start_tunnels()
def test_name(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
self.assertEqual(t.name, 'test')
def test_cmd(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
self.assertEqual(t.cmd, [self.executable])
def test_executable(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
self.assertEqual(t.executable, self.executable)
def test_runs(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
self.tm.start_tunnels()
self.assertTrue(t.is_running())
class TestVpncParser(unittest.TestCase):
def setUp(self):
self.conf = SafeConfigParser()
self.sec1 = 'tunnel:testvpnc'
self.conf.add_section(self.sec1)
self.conf.set(self.sec1, 'tunnel_type', 'vpnc')
self.conf.set(self.sec1, 'conf_file', '/path/to/conf.conf')
self.conf.set(self.sec1, 'ips', '10.10.10.10, 5.5.5.5')
self.vpnc_sec = 'vpnc'
self.conf.add_section(self.vpnc_sec)
self.conf.set(self.vpnc_sec, 'bin', '/path/to/vpnc.bin')
self.conf.set(self.vpnc_sec, 'base_conn_script', '/path/to/vpnc-conn-script')
self.tm = TunnelManager()
def test_base_parse(self):
self.tm.load_tunnels(self.conf)
self.assertEqual(len(self.tm.tunnels), 1)
def test_name(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
self.assertEqual(t.name, 'testvpnc')
def test_cmd(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
expected_cmd = ['calabar_vpnc', '/path/to/conf.conf', '--no-detach',
'--local-port', '0', '--non-inter', '--script']
self.assertEqual(len(t.cmd), len(expected_cmd)+1) # +1 for the actual script
for expected_arg in expected_cmd:
self.assertTrue(expected_arg in t.cmd)
# Now check the last arg and make sure it's to /tmp/...
self.assertTrue(t.cmd[-1].startswith('/tmp/'))
def test_executable(self):
self.tm.load_tunnels(self.conf)
t = self.tm.tunnels[0]
self.assertEqual(t.executable, '/path/to/vpnc.bin')
