def nodes(self, setup_node_controller, request): if hasattr(request, 'param'): node_vars = self.override_node_parameters(**request.param) else: node_vars = env_variables net_asset = None needs_nat = node_vars.get("platform") == consts.Platforms.NONE try: if not qe_env: net_asset = NetworkAssets() node_vars["net_asset"] = net_asset.get() controller = setup_node_controller(**node_vars) nodes = Nodes(controller, node_vars["private_ssh_key_path"]) nodes.prepare_nodes() if needs_nat: nodes.configure_nat() yield nodes if env_variables['test_teardown']: logging.info('--- TEARDOWN --- node controller\n') nodes.destroy_all_nodes() if needs_nat: nodes.unconfigure_nat() finally: if not qe_env: net_asset.release_all()
def get_nodes_func(config: TerraformConfig = TerraformConfig()): if "nodes" in nodes_data: return nodes_data["nodes"] nodes_data["needs_nat"] = config.platform == consts.Platforms.NONE nodes_data["net_asset"] = NetworkAssets() config.net_asset = nodes_data["net_asset"].get() nodes = Nodes(TerraformController(config), config.private_ssh_key_path) nodes.prepare_nodes() if nodes_data["needs_nat"]: nodes.configure_nat() nodes_data["nodes"] = nodes return nodes