def run(self): super(PlaybookGrapherCLI, self).run() playbook = self.args[0] loader, inventory, variable_manager = self._play_prereqs(self.options) grapher = Grapher(data_loader=loader, inventory_manager=inventory, variable_manager=variable_manager, playbook_filename=playbook, output_filename=self.options.output_filename) grapher.make_graph(include_role_tasks=self.options.include_role_tasks, tags=self.options.tags, skip_tags=self.options.skip_tags) grapher.render_graph(save_dot_file=self.options.save_dot_file) grapher.post_process_svg()
def test_grapher_simple_playbook(data_loader, inventory_manager, variable_manager, tmpdir): playbook = FIXTURES_DIR + "simple_playbook.yml" output_filepath = tmpdir.join('output') grapher = Grapher(data_loader=data_loader, inventory_manager=inventory_manager, variable_manager=variable_manager, playbook_filename=playbook, output_filename=output_filepath.strpath) grapher.make_graph() grapher.render_graph() grapher.post_process_svg() svg_filepath = output_filepath.strpath + ".svg" # test if the file exist. It will exist only if we write in it assert os.path.isfile(svg_filepath) tree = etree.parse(svg_filepath)
def run(self): super(PlaybookGrapherCLI29, self).run() loader, inventory, variable_manager = self._play_prereqs() grapher = Grapher(data_loader=loader, inventory_manager=inventory, variable_manager=variable_manager, playbook_filename=self.options.playbook_filename, options=self.options) grapher.make_graph() grapher.render_graph() return grapher.post_process_svg()
def fixture_simple_grapher(data_loader, inventory_manager, variable_manager, request): return Grapher(data_loader=data_loader, inventory_manager=inventory_manager, variable_manager=variable_manager, playbook_filename=request.param)