def menu():
confs = ['.vimrc', '.tcshrc', '.aliases', '.promptrc']
cit.ask('Which config file to update:')
choice = cit.get_choice(['ALL'] + confs + ['exit'])
if choice == 'ALL':
for conf in confs:
config(conf)
cit.info('Done')
cit.pause('Press Enter to exit ...')
cit.bye()
elif choice == 'exit':
cit.bye()
else:
return choice
'213.248.126.137',
'213.248.126.155',
'111.108.54.16',
'52.76.139.242',
]
if not ip_list:
cit.err('Ip List is empty').bye()
vpn_route = '10.100.0.1' # duetime
cit.info('VPN route = {}'.format(vpn_route))
# get mode, delete / set / print
available_modes = ['set', 'delete', 'print']
cit.ask("Choose mode:")
mode = cit.get_choice(available_modes)
if mode not in available_modes:
cit.err('Mode {} is not supported, available modes: {}'.format(mode, available_modes)).bye()
cit.info('Mode = {}'.format(mode))
if mode == 'set':
for ip in ip_list:
cmd = 'route add {ip} mask 255.255.255.255 {vpn_route}'.format(vpn_route=vpn_route, ip=ip)
ktk.runCmd(cmd)
elif mode == 'delete':
for ip in ip_list:
cmd = 'route delete {ip}'.format(ip=ip)
ktk.runCmd(cmd)
elif mode == 'print':
cmd = 'route print'
ktk.runCmd(cmd)
cit.pause()
def test_pause(self):
with patch("sys.stdout", new=StringIO()) as fake_out, patch("sys.stdin", new=StringIO("\n")): # simulate press enter
cit.pause()
expect_word = "\nPress Enter to Continue..."
self.assertEqual(fake_out.getvalue(), expect_word)
def test_pause(self):
self.fakein.write() # simulate press enter
cit.pause()
expect_word = "\nPress Enter to Continue..."
self.assertEqual(self.fakeout.readline(ansi=False), expect_word)
self.fakein.write() # simulate press enter
cit.pause()
expect_word = "\nPress Enter to Continue..."
self.assertEqual(self.fakeout.readline(ansi=False), expect_word)
def test_bye(self):
self.assertRaises(SystemExit, cit.bye, None)
def test_get_input(self):
self.fakein.write("ABC")
userinput = cit.get_input()
self.assertEqual(self.fakeout.readline(ansi=False), "> ")
self.assertEqual(userinput, "ABC")
def test_get_choice_1(self):
self.fakein.write("1")
self.assertEqual(cit.get_choice(["ABC", "DEF"]), "ABC")
expect_word = "| 1) ABC\n| 2) DEF\n> "
self.assertEqual(self.fakeout.readline(ansi=False), expect_word)
def test_get_choice_2(self):
self.fakein.write("ABC")
self.assertEqual(cit.get_choice(["ABC", "DEF"]), "ABC")
expect_word = "| 1) ABC\n| 2) DEF\n> "
self.assertEqual(self.fakeout.readline(ansi=False), expect_word)
if __name__ == '__main__':
unittest.main(verbosity=2, exit=False)
cit.pause()
