#!/usr/bin/env python # -*- 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 os import pytest from gns3server.modules.project import Project from gns3server.modules.vpcs import VPCS, VPCSVM @pytest.fixture(scope="module") def manager(port_manager): m = VPCS.instance() m.port_manager = port_manager return m @pytest.fixture(scope="function") def vm(project, manager): return VPCSVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager) def test_affect_uuid(): p = Project() assert len(p.uuid) == 36 p = Project(uuid='00010203-0405-0607-0809-0a0b0c0d0e0f') assert p.uuid == '00010203-0405-0607-0809-0a0b0c0d0e0f' def test_path(tmpdir): p = Project(location=str(tmpdir)) assert p.path == os.path.join(str(tmpdir), p.uuid) assert os.path.exists(os.path.join(str(tmpdir), p.uuid)) assert os.path.exists(os.path.join(str(tmpdir), p.uuid, 'vms')) def test_temporary_path(): p = Project() assert os.path.exists(p.path) def test_json(tmpdir): p = Project() assert p.__json__() == {"location": p.location, "uuid": p.uuid} def test_vm_working_directory(tmpdir, vm): p = Project(location=str(tmpdir)) assert os.path.exists(p.vm_working_directory(vm)) assert os.path.exists(os.path.join(str(tmpdir), p.uuid, vm.module_name, vm.uuid)) def test_mark_vm_for_destruction(tmpdir, vm): p = Project(location=str(tmpdir)) p.mark_vm_for_destruction(vm) assert len(p._vms_to_destroy) == 1 def test_commit(tmpdir, manager): project = Project(location=str(tmpdir)) vm = VPCSVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager) directory = project.vm_working_directory(vm) project.mark_vm_for_destruction(vm) assert len(project._vms_to_destroy) == 1 assert os.path.exists(directory) project.commit() assert len(project._vms_to_destroy) == 0 assert os.path.exists(directory) is False def test_project_delete(tmpdir): project = Project(location=str(tmpdir)) directory = project.path assert os.path.exists(directory) project.delete() assert os.path.exists(directory) is False