2
from wicd import wnettools
4
class TestWnettools(unittest.TestCase):
6
self.interface = wnettools.BaseInterface('eth0')
8
def test_find_wireless_interface(self):
9
interfaces = wnettools.GetWirelessInterfaces()
10
# wlan0 may change depending on your system
11
self.assertTrue('wlan0' in interfaces)
13
def test_find_wired_interface(self):
14
interfaces = wnettools.GetWiredInterfaces()
15
# eth0 may change depending on your system
16
self.assertTrue('eth0' in interfaces)
18
def test_wext_is_valid_wpasupplicant_driver(self):
19
self.assertTrue(wnettools.IsValidWpaSuppDriver('wext'))
21
def test_needs_external_calls_not_implemented(self):
22
self.assertRaises(NotImplementedError, wnettools.NeedsExternalCalls)
24
def test_get_ip_not_implemented(self):
25
self.assertRaises(NotImplementedError, self.interface.GetIP)
27
def test_is_up_not_implemented(self):
28
self.assertRaises(NotImplementedError, self.interface.IsUp)
30
def test_enable_debug_mode(self):
31
self.interface.SetDebugMode(True)
32
self.assertTrue(self.interface.verbose)
34
def test_disable_debug_mode(self):
35
self.interface.SetDebugMode(False)
36
self.assertFalse(self.interface.verbose)
38
def test_interface_name_sanitation(self):
39
interface = wnettools.BaseInterface('blahblah; uptime > /tmp/blah | cat')
40
self.assertEquals(interface.iface, 'blahblahuptimetmpblahcat')
42
def test_freq_translation_low(self):
44
interface = wnettools.BaseWirelessInterface('wlan0')
45
self.assertEquals(interface._FreqToChannel(freq), 1)
47
def test_freq_translation_high(self):
49
interface = wnettools.BaseWirelessInterface('wlan0')
50
self.assertEquals(interface._FreqToChannel(freq), 14)
52
def test_generate_psk(self):
53
interface = wnettools.BaseWirelessInterface('wlan0')
54
psk = interface.GeneratePSK({'essid' : 'Network 1', 'key' : 'arandompassphrase'})
55
self.assertEquals(psk, 'd70463014514f4b4ebb8e3aebbdec13f4437ac3a9af084b3433f3710e658a7be')
58
suite = unittest.TestSuite()
60
[ tests.append(test) for test in dir(TestWnettools) if test.startswith('test') ]
62
suite.addTest(TestWnettools(test))
65
if __name__ == '__main__':