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()