def setup(self): print("setting up test of task runner class") self.config = parse_config(good_config)
import argparse from nc.taskrunner import Discovery from nc.config import parse_config if __name__ == "__main__": parser = argparse.ArgumentParser(description='Cisco Network Discovery.') parser.add_argument('--config', dest='config_path', action='store', help='Path to configuration file') parser.add_argument("--devices", metavar="device-a.demo.local", type=str, nargs="+", help="Device hostnames to run discovery against") args = parser.parse_args() config = parse_config(args.config_path) disco = Discovery( commands=config["commands"], devices=config["devices"] ) disco.run() for dev, cmd in disco.results.items(): print "Device: %s discovery information: " % dev for cmd, data in cmd.items(): print cmd, data
def test_good_load_config(): print parse_config(good_config) assert parse_config(good_config)