mirror of
https://github.com/GNS3/gns3-server
synced 2024-11-24 09:18:08 +00:00
Close a project
This commit is contained in:
parent
986a7f55ef
commit
3f5c2390cd
@ -56,6 +56,24 @@ class ProjectHandler:
|
||||
project.commit()
|
||||
response.set_status(204)
|
||||
|
||||
@classmethod
|
||||
@Route.post(
|
||||
r"/project/{uuid}/close",
|
||||
description="Close project",
|
||||
parameters={
|
||||
"uuid": "Project instance UUID",
|
||||
},
|
||||
status_codes={
|
||||
204: "Project closed",
|
||||
404: "Project instance doesn't exist"
|
||||
})
|
||||
def close(request, response):
|
||||
|
||||
pm = ProjectManager.instance()
|
||||
project = pm.get_project(request.match_info["uuid"])
|
||||
project.close()
|
||||
response.set_status(204)
|
||||
|
||||
@classmethod
|
||||
@Route.delete(
|
||||
r"/project/{uuid}",
|
||||
|
@ -27,6 +27,7 @@ class BaseVM:
|
||||
self._uuid = uuid
|
||||
self._project = project
|
||||
self._manager = manager
|
||||
project.add_vm(self)
|
||||
|
||||
log.debug("{module}: {name} [{uuid}] initialized".format(module=self.manager.module_name,
|
||||
name=self.name,
|
||||
|
@ -46,6 +46,7 @@ class Project:
|
||||
if location is None:
|
||||
self._location = tempfile.mkdtemp()
|
||||
|
||||
self._vms = set()
|
||||
self._vms_to_destroy = set()
|
||||
self._path = os.path.join(self._location, self._uuid)
|
||||
try:
|
||||
@ -68,6 +69,11 @@ class Project:
|
||||
|
||||
return self._path
|
||||
|
||||
@property
|
||||
def vms(self):
|
||||
|
||||
return self._vms
|
||||
|
||||
def vm_working_directory(self, vm):
|
||||
"""
|
||||
Return a working directory for a specific VM.
|
||||
@ -98,10 +104,21 @@ class Project:
|
||||
"location": self._location
|
||||
}
|
||||
|
||||
def add_vm(self, vm):
|
||||
"""
|
||||
Add a VM to the project. In theory this should be called by
|
||||
the VM initializer.
|
||||
|
||||
:params vm: A VM instance
|
||||
"""
|
||||
|
||||
self._vms.add(vm)
|
||||
|
||||
def close(self):
|
||||
"""Close the project, but keep informations on disk"""
|
||||
|
||||
pass
|
||||
for vm in self._vms:
|
||||
vm.close()
|
||||
|
||||
def commit(self):
|
||||
"""Write project changes on disk"""
|
||||
|
@ -20,6 +20,7 @@ This test suite check /project endpoint
|
||||
"""
|
||||
|
||||
import uuid
|
||||
from tests.utils import asyncio_patch
|
||||
|
||||
|
||||
def test_create_project_with_dir(server, tmpdir):
|
||||
@ -51,8 +52,10 @@ def test_create_project_with_uuid(server):
|
||||
|
||||
|
||||
def test_commit_project(server, project):
|
||||
response = server.post("/project/{uuid}/commit".format(uuid=project.uuid), example=True)
|
||||
with asyncio_patch("gns3server.modules.project.Project.commit", return_value=True) as mock:
|
||||
response = server.post("/project/{uuid}/commit".format(uuid=project.uuid), example=True)
|
||||
assert response.status == 204
|
||||
assert mock.called
|
||||
|
||||
|
||||
def test_commit_project_invalid_project_uuid(server, project):
|
||||
@ -60,14 +63,25 @@ def test_commit_project_invalid_project_uuid(server, project):
|
||||
assert response.status == 404
|
||||
|
||||
|
||||
def test_delete_project(server):
|
||||
query = {"uuid": "00010203-0405-0607-0809-0a0b0c0d0e0f"}
|
||||
response = server.post("/project", query)
|
||||
assert response.status == 200
|
||||
response = server.delete("/project/00010203-0405-0607-0809-0a0b0c0d0e0f")
|
||||
assert response.status == 204
|
||||
def test_delete_project(server, project):
|
||||
with asyncio_patch("gns3server.modules.project.Project.delete", return_value=True) as mock:
|
||||
response = server.delete("/project/{uuid}".format(uuid=project.uuid), example=True)
|
||||
assert response.status == 204
|
||||
assert mock.called
|
||||
|
||||
|
||||
def test_delete_project_invalid_uuid(server, project):
|
||||
response = server.delete("/project/{uuid}".format(uuid=uuid.uuid4()))
|
||||
assert response.status == 404
|
||||
|
||||
|
||||
def test_close_project(server, project):
|
||||
with asyncio_patch("gns3server.modules.project.Project.close", return_value=True) as mock:
|
||||
response = server.post("/project/{uuid}/close".format(uuid=project.uuid), example=True)
|
||||
assert response.status == 204
|
||||
assert mock.called
|
||||
|
||||
|
||||
def test_close_project_invalid_uuid(server, project):
|
||||
response = server.post("/project/{uuid}/close".format(uuid=uuid.uuid4()))
|
||||
assert response.status == 404
|
||||
|
@ -18,6 +18,8 @@
|
||||
|
||||
import os
|
||||
import pytest
|
||||
from unittest.mock import patch
|
||||
|
||||
from gns3server.modules.project import Project
|
||||
from gns3server.modules.vpcs import VPCS, VPCSVM
|
||||
|
||||
@ -89,3 +91,19 @@ def test_project_delete(tmpdir):
|
||||
assert os.path.exists(directory)
|
||||
project.delete()
|
||||
assert os.path.exists(directory) is False
|
||||
|
||||
|
||||
def test_project_add_vm(tmpdir, manager):
|
||||
project = Project(location=str(tmpdir))
|
||||
# The VM initalizer call the add_vm method
|
||||
vm = VPCSVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
|
||||
assert len(project.vms) == 1
|
||||
|
||||
|
||||
def test_project_close(tmpdir, manager):
|
||||
project = Project(location=str(tmpdir))
|
||||
vm = VPCSVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
|
||||
project.add_vm(vm)
|
||||
with patch("gns3server.modules.vpcs.vpcs_vm.VPCSVM.close") as mock:
|
||||
project.close()
|
||||
assert mock.called
|
||||
|
Loading…
Reference in New Issue
Block a user