Rename modules => hypervisor

tests/hypervisor/virtualbox/test_virtualbox_manager.py

@@ -0,0 +1,99 @@
+# -*- 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 <http://www.gnu.org/licenses/>.
+
+
+import pytest
+import tempfile
+import os
+import stat
+import asyncio
+
+
+from unittest.mock import patch
+
+from gns3server.hypervisor.virtualbox import VirtualBox
+from gns3server.hypervisor.virtualbox.virtualbox_error import VirtualBoxError
+from tests.utils import asyncio_patch
+
+
+@pytest.fixture(scope="module")
+def manager(port_manager):
+    m = VirtualBox.instance()
+    m.port_manager = port_manager
+    return m
+
+
+def test_vm_invalid_vboxmanage_path(manager):
+    with patch("gns3server.config.Config.get_section_config", return_value={"vboxmanage_path": "/bin/test_fake"}):
+        with pytest.raises(VirtualBoxError):
+            manager.find_vboxmanage()
+
+
+def test_vm_non_executable_vboxmanage_path(manager):
+    tmpfile = tempfile.NamedTemporaryFile()
+    with patch("gns3server.config.Config.get_section_config", return_value={"vboxmanage_path": tmpfile.name}):
+        with pytest.raises(VirtualBoxError):
+            manager.find_vboxmanage()
+
+
+def test_vm_invalid_executable_name_vboxmanage_path(manager, tmpdir):
+    path = str(tmpdir / "vpcs")
+    with open(path, "w+") as f:
+        f.write(path)
+    os.chmod(path, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
+    tmpfile = tempfile.NamedTemporaryFile()
+    with patch("gns3server.config.Config.get_section_config", return_value={"vboxmanage_path": path}):
+        with pytest.raises(VirtualBoxError):
+            manager.find_vboxmanage()
+
+
+def test_vboxmanage_path(manager, tmpdir):
+    path = str(tmpdir / "VBoxManage")
+    with open(path, "w+") as f:
+        f.write(path)
+    os.chmod(path, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
+    tmpfile = tempfile.NamedTemporaryFile()
+    with patch("gns3server.config.Config.get_section_config", return_value={"vboxmanage_path": path}):
+        assert manager.find_vboxmanage() == path
+
+
+def test_list_images(manager, loop):
+    vm_list = ['"Windows 8.1" {27b4d095-ff5f-4ac4-bb9d-5f2c7861c1f1}',
+               '"Carriage',
+               'Return" {27b4d095-ff5f-4ac4-bb9d-5f2c7861c1f1}',
+               '',
+               '"<inaccessible>" {42b4d095-ff5f-4ac4-bb9d-5f2c7861c1f1}',
+               '"Linux Microcore 4.7.1" {ccd8c50b-c172-457d-99fa-dd69371ede0e}']
+
+    @asyncio.coroutine
+    def execute_mock(cmd, args):
+        if cmd == "list":
+            return vm_list
+        else:
+            if args[0] == "Windows 8.1":
+                return ["memory=512"]
+            elif args[0] == "Linux Microcore 4.7.1":
+                return ["memory=256"]
+        assert False, "Unknow {} {}".format(cmd, args)
+
+    with asyncio_patch("gns3server.hypervisor.virtualbox.VirtualBox.execute") as mock:
+        mock.side_effect = execute_mock
+        vms = loop.run_until_complete(asyncio.async(manager.list_images()))
+    assert vms == [
+        {"vmname": "Windows 8.1", "ram": 512},
+        {"vmname": "Linux Microcore 4.7.1", "ram": 256}
+    ]

tests/hypervisor/virtualbox/test_virtualbox_vm.py

@@ -0,0 +1,69 @@
+# -*- 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 <http://www.gnu.org/licenses/>.
+
+import pytest
+import asyncio
+from tests.utils import asyncio_patch
+
+from gns3server.hypervisor.virtualbox.virtualbox_vm import VirtualBoxVM
+from gns3server.hypervisor.virtualbox.virtualbox_error import VirtualBoxError
+from gns3server.hypervisor.virtualbox import VirtualBox
+
+
+@pytest.fixture(scope="module")
+def manager(port_manager):
+    m = VirtualBox.instance()
+    m.port_manager = port_manager
+    return m
+
+
+@pytest.fixture(scope="function")
+def vm(project, manager):
+    return VirtualBoxVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager, "test", False)
+
+
+def test_vm(project, manager):
+    vm = VirtualBoxVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager, "test", False)
+    assert vm.name == "test"
+    assert vm.id == "00010203-0405-0607-0809-0a0b0c0d0e0f"
+    assert vm.vmname == "test"
+
+
+def test_vm_valid_virtualbox_api_version(loop, project, manager):
+    with asyncio_patch("gns3server.hypervisor.virtualbox.VirtualBox.execute", return_value=["API version:  4_3"]):
+        vm = VirtualBoxVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager, "test", False)
+        loop.run_until_complete(asyncio.async(vm.create()))
+
+
+def test_vm_invalid_virtualbox_api_version(loop, project, manager):
+    with asyncio_patch("gns3server.hypervisor.virtualbox.VirtualBox.execute", return_value=["API version:  4_2"]):
+        with pytest.raises(VirtualBoxError):
+            vm = VirtualBoxVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager, "test", False)
+            loop.run_until_complete(asyncio.async(vm.create()))
+
+
+def test_vm_adapter_add_nio_binding_adapter_not_exist(loop, vm, manager, free_console_port):
+    nio = manager.create_nio(manager.vboxmanage_path, {"type": "nio_udp", "lport": free_console_port, "rport": free_console_port, "rhost": "192.168.1.2"})
+    with pytest.raises(VirtualBoxError):
+        loop.run_until_complete(asyncio.async(vm.adapter_add_nio_binding(15, nio)))
+
+
+def test_json(vm, tmpdir, project):
+    assert vm.__json__()["vm_directory"] is None
+    project._path = str(tmpdir)
+    vm._linked_clone = True
+    assert vm.__json__()["vm_directory"] is not None