# -*- coding: utf-8 -*- # # Copyright (C) 2015 GNS3 Technologies Inc. # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . import uuid import os import pytest from unittest.mock import patch from gns3server.modules.vpcs import VPCS def test_create_vm_new_topology(loop, project, port_manager): VPCS._instance = None vpcs = VPCS.instance() vpcs.port_manager = port_manager vm_id = str(uuid.uuid4()) vm = loop.run_until_complete(vpcs.create_vm("PC 1", project.id, vm_id)) assert vm in project.vms def test_create_vm_new_topology_without_uuid(loop, project, port_manager): VPCS._instance = None vpcs = VPCS.instance() vpcs.port_manager = port_manager vm = loop.run_until_complete(vpcs.create_vm("PC 1", project.id, None)) assert vm in project.vms assert len(vm.id) == 36 def test_create_vm_old_topology(loop, project, tmpdir, port_manager): with patch("gns3server.config.Config.get_section_config", return_value={"local": True}): # Create an old topology directory project_dir = str(tmpdir / "testold") vm_dir = os.path.join(project_dir, "testold-files", "vpcs", "pc-1") project.path = project_dir os.makedirs(vm_dir, exist_ok=True) with open(os.path.join(vm_dir, "startup.vpc"), "w+") as f: f.write("1") VPCS._instance = None vpcs = VPCS.instance() vpcs.port_manager = port_manager vm_id = 1 vm = loop.run_until_complete(vpcs.create_vm("PC 1", project.id, vm_id)) assert len(vm.id) == 36 vm_dir = os.path.join(project_dir, "project-files", "vpcs", vm.id) with open(os.path.join(vm_dir, "startup.vpc")) as f: assert f.read() == "1"