PEP8

tests/api/base.py

@@ -15,6 +15,7 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+
 """Base code use for all API tests"""
 
 import json
@@ -27,7 +28,7 @@ import aiohttp
 
 
 from gns3server.web.route import Route
-#TODO: get rid of *
+# TODO: get rid of *
 from gns3server.handlers import *
 from gns3server.modules import MODULES
 from gns3server.modules.port_manager import PortManager
@@ -143,6 +144,7 @@ def loop(request):
     request.addfinalizer(tear_down)
     return loop
 
+
 @pytest.fixture(scope="session")
 def server(request, loop):
     port = _get_unused_port()
@@ -167,5 +169,4 @@ def server(request, loop):
 
 @pytest.fixture(scope="module")
 def project():
-    return ProjectManager.instance().create_project(uuid = "a1e920ca-338a-4e9f-b363-aa607b09dd80")
-
+    return ProjectManager.instance().create_project(uuid="a1e920ca-338a-4e9f-b363-aa607b09dd80")

tests/api/test_project.py

@@ -30,12 +30,14 @@ def test_create_project_with_dir(server, tmpdir):
     assert response.status == 200
     assert response.json['location'] == str(tmpdir)
 
+
 def test_create_project_without_dir(server):
     query = {}
     response = server.post('/project', query)
     assert response.status == 200
     assert response.json['uuid'] is not None
 
+
 def test_create_project_with_uuid(server):
     query = {'uuid': '00010203-0405-0607-0809-0a0b0c0d0e0f'}
     response = server.post('/project', query)

tests/api/test_vpcs.py

@@ -38,12 +38,11 @@ def test_vpcs_create(server, project):
     assert response.json["project_uuid"] == "61d61bdd-aa7d-4912-817f-65a9eb54d3ab"
 
 
-#FIXME
 def test_vpcs_nio_create_udp(server, vm):
     response = server.post("/vpcs/{}/ports/0/nio".format(vm["uuid"]), {"type": "nio_udp",
-                                                                            "lport": 4242,
-                                                                            "rport": 4343,
-                                                                            "rhost": "127.0.0.1"},
+                                                                       "lport": 4242,
+                                                                       "rport": 4343,
+                                                                       "rhost": "127.0.0.1"},
                            example=True)
     assert response.status == 200
     assert response.route == "/vpcs/{uuid}/ports/{port_id}/nio"
@@ -53,18 +52,17 @@ def test_vpcs_nio_create_udp(server, vm):
 @patch("gns3server.modules.vpcs.vpcs_device.has_privileged_access", return_value=True)
 def test_vpcs_nio_create_tap(mock, server, vm):
     response = server.post("/vpcs/{}/ports/0/nio".format(vm["uuid"]), {"type": "nio_tap",
-                                                                            "tap_device": "test"})
+                                                                       "tap_device": "test"})
     assert response.status == 200
     assert response.route == "/vpcs/{uuid}/ports/{port_id}/nio"
     assert response.json["type"] == "nio_tap"
 
 
-#FIXME
 def test_vpcs_delete_nio(server, vm):
     response = server.post("/vpcs/{}/ports/0/nio".format(vm["uuid"]), {"type": "nio_udp",
-                                                                            "lport": 4242,
-                                                                            "rport": 4343,
-                                                                            "rhost": "127.0.0.1"})
+                                                                       "lport": 4242,
+                                                                       "rport": 4343,
+                                                                       "rhost": "127.0.0.1"})
     response = server.delete("/vpcs/{}/ports/0/nio".format(vm["uuid"]), example=True)
     assert response.status == 200
     assert response.route == "/vpcs/{uuid}/ports/{port_id}/nio"

tests/conftest.py

@@ -1,19 +0,0 @@
-import sys
-import os
-import pytest
-import subprocess
-import time
-
-
-@pytest.fixture(scope="session", autouse=True)
-def server(request):
-    """
-    Starts GNS3 server for all the tests.
-    """
-
-    cwd = os.path.dirname(os.path.abspath(__file__))
-    server_script = os.path.join(cwd, "../gns3server/main.py")
-    process = subprocess.Popen([sys.executable, server_script, "--port=8000"])
-    #time.sleep(1)  # give some time for the process to start
-    request.addfinalizer(process.terminate)
-    return process

tests/modules/test_project.py

@@ -16,8 +16,9 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from gns3server.modules.project import Project
 import os
+from gns3server.modules.project import Project
+
 
 def test_affect_uuid():
     p = Project()
@@ -26,16 +27,19 @@ def test_affect_uuid():
     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))
+    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, 'files'))
 
+
 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}

tests/modules/test_project_manager.py

@@ -25,8 +25,8 @@ def test_create_project():
     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')
-

tests/modules/vpcs/test_vpcs_device.py

@@ -19,7 +19,7 @@ import pytest
 import asyncio
 from tests.utils import asyncio_patch
 
-#TODO: Move loop to util
+# TODO: Move loop to util
 from tests.api.base import loop, project
 from asyncio.subprocess import Process
 from unittest.mock import patch, MagicMock
@@ -42,6 +42,7 @@ def test_vm(manager):
     assert vm.name == "test"
     assert vm.uuid == "00010203-0405-0607-0809-0a0b0c0d0e0f"
 
+
 @patch("subprocess.check_output", return_value="Welcome to Virtual PC Simulator, version 0.1".encode("utf-8"))
 def test_vm_invalid_vpcs_version(project, manager):
     with pytest.raises(VPCSError):
@@ -49,20 +50,23 @@ def test_vm_invalid_vpcs_version(project, manager):
         assert vm.name == "test"
         assert vm.uuid == "00010203-0405-0607-0809-0a0b0c0d0e0f"
 
-@patch("gns3server.config.Config.get_section_config", return_value = {"path": "/bin/test_fake"})
+
+@patch("gns3server.config.Config.get_section_config", return_value={"path": "/bin/test_fake"})
 def test_vm_invalid_vpcs_path(project, manager):
     with pytest.raises(VPCSError):
         vm = VPCSDevice("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
         assert vm.name == "test"
         assert vm.uuid == "00010203-0405-0607-0809-0a0b0c0d0e0f"
 
+
 def test_start(project, loop, manager):
     with asyncio_patch("asyncio.create_subprocess_exec", return_value=MagicMock()):
         vm = VPCSDevice("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
         nio = vm.port_add_nio_binding(0, {"type": "nio_udp", "lport": 4242, "rport": 4243, "rhost": "127.0.0.1"})
 
         loop.run_until_complete(asyncio.async(vm.start()))
-        assert vm.is_running() == True
+        assert vm.is_running()
+
 
 def test_stop(project, loop, manager):
     process = MagicMock()
@@ -71,22 +75,25 @@ def test_stop(project, loop, manager):
         nio = vm.port_add_nio_binding(0, {"type": "nio_udp", "lport": 4242, "rport": 4243, "rhost": "127.0.0.1"})
 
         loop.run_until_complete(asyncio.async(vm.start()))
-        assert vm.is_running() == True
+        assert vm.is_running()
         loop.run_until_complete(asyncio.async(vm.stop()))
-        assert vm.is_running() == False
+        assert vm.is_running() is False
         process.terminate.assert_called_with()
 
+
 def test_add_nio_binding_udp(manager):
     vm = VPCSDevice("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
     nio = vm.port_add_nio_binding(0, {"type": "nio_udp", "lport": 4242, "rport": 4243, "rhost": "127.0.0.1"})
     assert nio.lport == 4242
 
+
 def test_add_nio_binding_tap(project, manager):
     vm = VPCSDevice("test", 42, project, manager)
     with patch("gns3server.modules.vpcs.vpcs_device.has_privileged_access", return_value=True):
         nio = vm.port_add_nio_binding(0, {"type": "nio_tap", "tap_device": "test"})
         assert nio.tap_device == "test"
 
+
 def test_add_nio_binding_tap_no_privileged_access(manager):
     vm = VPCSDevice("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
     with patch("gns3server.modules.vpcs.vpcs_device.has_privileged_access", return_value=False):
@@ -94,6 +101,7 @@ def test_add_nio_binding_tap_no_privileged_access(manager):
             vm.port_add_nio_binding(0, {"type": "nio_tap", "tap_device": "test"})
     assert vm._ethernet_adapter.ports[0] is None
 
+
 def test_port_remove_nio_binding(manager):
     vm = VPCSDevice("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
     nio = vm.port_add_nio_binding(0, {"type": "nio_udp", "lport": 4242, "rport": 4243, "rhost": "127.0.0.1"})