diff --git a/gns3server/modules/project_manager.py b/gns3server/modules/project_manager.py new file mode 100644 index 00000000..0ff089bc --- /dev/null +++ b/gns3server/modules/project_manager.py @@ -0,0 +1,63 @@ +# -*- 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 aiohttp +from .project import Project + + +class ProjectManager: + """ + This singleton, keep track of available projects. + """ + + def __init__(self): + self._projects = {} + + @classmethod + def instance(cls): + """ + Singleton to return only one instance of BaseManager. + + :returns: instance of Manager + """ + + if not hasattr(cls, "_instance"): + cls._instance = cls() + return cls._instance + + def get_project(self, project_id): + """ + Returns a Project instance. + + :param project_id: Project identifier + + :returns: Project instance + """ + + if project_id not in self._projects: + raise aiohttp.web.HTTPNotFound(text="Project UUID {} doesn't exist".format(project_id)) + return self._projects[project_id] + + def create_project(self, **kwargs): + """ + Create a project and keep a references to it in project manager. + + See documentation of Project for arguments + """ + project = Project(**kwargs) + self._projects[project.uuid] = project + return project diff --git a/tests/modules/test_project.py b/tests/modules/test_project.py index 07245afd..2d6ea0b3 100644 --- a/tests/modules/test_project.py +++ b/tests/modules/test_project.py @@ -23,7 +23,7 @@ def test_affect_uuid(): p = Project() assert len(p.uuid) == 36 - p = Project(uuid = '00010203-0405-0607-0809-0a0b0c0d0e0f') + p = Project(uuid='00010203-0405-0607-0809-0a0b0c0d0e0f') assert p.uuid == '00010203-0405-0607-0809-0a0b0c0d0e0f' def test_path(tmpdir): diff --git a/tests/modules/test_project_manager.py b/tests/modules/test_project_manager.py new file mode 100644 index 00000000..64009928 --- /dev/null +++ b/tests/modules/test_project_manager.py @@ -0,0 +1,32 @@ +# -*- 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 aiohttp +import pytest +from gns3server.modules.project_manager import ProjectManager + + +def test_create_project(): + pm = ProjectManager.instance() + project = pm.create_project(uuid='00010203-0405-0607-0809-0a0b0c0d0e0f') + assert project == pm.get_project('00010203-0405-0607-0809-0a0b0c0d0e0f') + +def test_project_not_found(): + pm = ProjectManager.instance() + with pytest.raises(aiohttp.web.HTTPNotFound): + pm.get_project('00010203-0405-0607-0809-000000000000') +