Link API (do nothing for the moment)

tests/controller/test_link.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2016 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 <http://www.gnu.org/licenses/>.
+
+import pytest
+
+from gns3server.controller.link import Link
+from gns3server.controller.vm import VM
+from gns3server.controller.hypervisor import Hypervisor
+from gns3server.controller.project import Project
+
+
+@pytest.fixture
+def project():
+    return Project()
+
+
+@pytest.fixture
+def hypervisor():
+    return Hypervisor("example.com")
+
+
+def test_addVM(async_run, project, hypervisor):
+    vm1 = VM(project, hypervisor)
+
+    link = Link()
+    async_run(link.addVM(vm1, 0, 4))
+    assert link._vms == [
+        {
+            "vm": vm1,
+            "adapter_number": 0,
+            "port_number": 4
+        }
+    ]
+
+
+def test_json(async_run, project, hypervisor):
+    vm1 = VM(project, hypervisor)
+    vm2 = VM(project, hypervisor)
+
+    link = Link()
+    async_run(link.addVM(vm1, 0, 4))
+    async_run(link.addVM(vm2, 1, 3))
+    assert link.__json__() == {
+        "link_id": link.id,
+        "vms": [
+            {
+                "vm_id": vm1.id,
+                "adapter_number": 0,
+                "port_number": 4
+            },
+            {
+                "vm_id": vm2.id,
+                "adapter_number": 1,
+                "port_number": 3
+            }
+        ]
+    }
+

tests/controller/test_project.py

@@ -16,6 +16,8 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+import pytest
+import aiohttp
 from unittest.mock import MagicMock
 
 
@@ -45,3 +47,34 @@ def test_addVM(async_run):
                                              'console_type': 'telnet',
                                              'startup_config': 'test.cfg',
                                              'name': 'test'})
+
+
+def test_getVM(async_run):
+    hypervisor = MagicMock()
+    project = Project()
+    vm = async_run(project.addVM(hypervisor, None, name="test", vm_type="vpcs", properties={"startup_config": "test.cfg"}))
+    assert project.getVM(vm.id) == vm
+
+    with pytest.raises(aiohttp.web_exceptions.HTTPNotFound):
+        project.getVM("test")
+
+
+def test_addLink(async_run):
+    hypervisor = MagicMock()
+    project = Project()
+    vm1 = async_run(project.addVM(hypervisor, None, name="test1", vm_type="vpcs", properties={"startup_config": "test.cfg"}))
+    vm2 = async_run(project.addVM(hypervisor, None, name="test2", vm_type="vpcs", properties={"startup_config": "test.cfg"}))
+    link = async_run(project.addLink())
+    async_run(link.addVM(vm1, 3, 1))
+    async_run(link.addVM(vm2, 4, 2))
+    assert len(link._vms) == 2
+
+
+def test_getLink(async_run):
+    hypervisor = MagicMock()
+    project = Project()
+    link = async_run(project.addLink())
+    assert project.getLink(link.id) == link
+
+    with pytest.raises(aiohttp.web_exceptions.HTTPNotFound):
+        project.getLink("test")