def test_image_in_project(tmpdir): project_dir = tmpdir / "project" / "project-files" os.makedirs(str(project_dir / "images")) project = Project() project.setFilesDir(str(tmpdir / "project")) topology = Topology() topology.project = project with open(str(project_dir / "images" / "1.jpg"), "w+") as f: f.write("AA") image1 = PixmapImageItem(None, "images/1.jpg") topology.addImage(image1) assert image1 in topology._images image2 = PixmapImageItem(None, "images/1.jpg") topology.addImage(image2) assert image1 in topology._images assert image2 in topology._images topology.removeImage(image2) assert os.path.exists(str(project_dir / "images" / "1.jpg")) # If not image use this file delete it topology.removeImage(image1) assert not os.path.exists(str(project_dir / "images" / "1.jpg"))
def test_dump_http_auth(vpcs_device, project, remote_server): remote_server.setUser("hello") remote_server.setPassword("world") from gns3.modules.vpcs.vpcs_device import VPCSDevice from gns3.modules.vpcs import VPCS vpcs_device = VPCSDevice(VPCS(), remote_server, project) vpcs_device._vpcs_device_id = str(uuid.uuid4()) vpcs_device._settings = {"name": "VPCS 1", "script_file": "", "console": None, "startup_script": None} vpcs_device.setInitialized(True) topology = Topology() topology.project = project topology.addNode(vpcs_device) dump = topology.dump(include_gui_data=False) assert dict(dump) == { "project_id": project.id(), "auto_start": False, "name": project.name(), "version": __version__, "revision": 4, "topology": { "nodes": [ { "description": "VPCS device", "id": vpcs_device.id(), "ports": [ { "id": vpcs_device.ports()[0].id(), "name": "Ethernet0", "port_number": 0, "adapter_number": 0 } ], "properties": { "name": vpcs_device.name() }, "server_id": remote_server.id(), "type": "VPCSDevice", "vm_id": None } ], "servers": [ { "vm": False, "host": "127.0.0.1", "id": remote_server.id(), "local": False, "port": 8001, "protocol": "http", "ram_limit": 0, "user": "******" } ] }, "type": "topology" }
def test_randomize_id(project, tmpdir): project.setTopologyFile(str(tmpdir / "test.gns3")) project_uuid = str(uuid.uuid4()) vm_uuid1 = str(uuid.uuid4()) os.makedirs(str(tmpdir / "project-files" / "vpcs" / vm_uuid1)) open(str(tmpdir / "project-files" / "vpcs" / vm_uuid1 / "test.log"), "w+").close() vm_uuid2 = str(uuid.uuid4()) orig_topology = { "project_id": project_uuid, "topology": { "nodes": [ { "vm_id": vm_uuid1, }, { "vm_id": vm_uuid2 } ] } } topology = Topology() topology.project = project top = topology._randomize_id(orig_topology) assert top["project_id"] != project_uuid assert top["topology"]["nodes"][0]["vm_id"] != vm_uuid1 assert not os.path.exists(str(tmpdir / "project-files" / "vpcs" / vm_uuid1 / "test.log")) assert os.path.exists(str(tmpdir / "project-files" / "vpcs" / top["topology"]["nodes"][0]["vm_id"] / "test.log")) assert top["topology"]["nodes"][1]["vm_id"] != vm_uuid2 assert top["topology"]["nodes"][0]["vm_id"] != top["topology"]["nodes"][1]["vm_id"]
def test_randomize_id(project, tmpdir): project.setTopologyFile(str(tmpdir / "test.gns3")) project_uuid = str(uuid.uuid4()) vm_uuid1 = str(uuid.uuid4()) os.makedirs(str(tmpdir / "project-files" / "vpcs" / vm_uuid1)) open(str(tmpdir / "project-files" / "vpcs" / vm_uuid1 / "test.log"), "w+").close() vm_uuid2 = str(uuid.uuid4()) orig_topology = { "project_id": project_uuid, "topology": { "nodes": [{ "vm_id": vm_uuid1, }, { "vm_id": vm_uuid2 }] } } topology = Topology() topology.project = project top = topology._randomize_id(orig_topology) assert top["project_id"] != project_uuid assert top["topology"]["nodes"][0]["vm_id"] != vm_uuid1 assert not os.path.exists( str(tmpdir / "project-files" / "vpcs" / vm_uuid1 / "test.log")) assert os.path.exists( str(tmpdir / "project-files" / "vpcs" / top["topology"]["nodes"][0]["vm_id"] / "test.log")) assert top["topology"]["nodes"][1]["vm_id"] != vm_uuid2 assert top["topology"]["nodes"][0]["vm_id"] != top["topology"]["nodes"][1][ "vm_id"]
def test_dump_random_id(vpcs_device, project, local_server): topology = Topology() topology.project = project topology.addNode(vpcs_device) fake_uuid = str(uuid.uuid4()) with patch("uuid.uuid4", return_value=fake_uuid): dump = topology.dump(include_gui_data=False, random_id=True) assert dict(dump) == { "project_id": fake_uuid, "auto_start": False, "name": project.name(), "version": __version__, "revision": 4, "topology": { "nodes": [ { "id": vpcs_device.id(), "description": "VPCS device", "ports": [ { "id": vpcs_device.ports()[0].id(), "name": "Ethernet0", "port_number": 0, "adapter_number": 0 } ], "properties": { "name": vpcs_device.name() }, "server_id": local_server.id(), "type": "VPCSDevice", "vm_id": fake_uuid} ], "servers": [ { "host": "127.0.0.1", "id": local_server.id(), "local": True, "port": 8000, "protocol": "http", "ram_limit": 0, "vm": False } ] }, "type": "topology" }
def test_dump_random_id(vpcs_device, project, local_server): topology = Topology() topology.project = project topology.addNode(vpcs_device) fake_uuid = str(uuid.uuid4()) with patch("uuid.uuid4", return_value=fake_uuid): dump = topology.dump(include_gui_data=False, random_id=True) assert dict(dump) == { "project_id": fake_uuid, "auto_start": False, "name": project.name(), "version": __version__, "revision": 4, "topology": { "nodes": [{ "id": vpcs_device.id(), "description": "VPCS device", "ports": [{ "id": vpcs_device.ports()[0].id(), "name": "Ethernet0", "port_number": 0, "adapter_number": 0 }], "properties": { "name": vpcs_device.name() }, "server_id": local_server.id(), "type": "VPCSDevice", "vm_id": fake_uuid }], "servers": [{ "host": "127.0.0.1", "id": local_server.id(), "local": True, "port": 8000, "protocol": "http", "ram_limit": 0, "vm": False }] }, "type": "topology" }
def test_dump(vpcs_device, project, local_server): topology = Topology() topology.project = project topology.addNode(vpcs_device) dump = topology.dump(include_gui_data=False) assert dict(dump) == { "project_id": project.id(), "auto_start": False, "name": project.name(), "version": __version__, "revision": 4, "topology": { "nodes": [ { "description": "VPCS device", "id": vpcs_device.id(), "ports": [ { "id": vpcs_device.ports()[0].id(), "name": "Ethernet0", "port_number": 0, "adapter_number": 0 } ], "properties": { "name": vpcs_device.name() }, "server_id": local_server.id(), "type": "VPCSDevice", "vm_id": None } ], "servers": [ { "vm": False, "host": "127.0.0.1", "id": local_server.id(), "local": True, "port": 8000, "protocol": "http", "ram_limit": 0 } ] }, "type": "topology" }
def test_dump(vpcs_device, project, local_server): topology = Topology() topology.project = project topology.addNode(vpcs_device) dump = topology.dump(include_gui_data=False) assert dict(dump) == { "project_id": project.id(), "auto_start": False, "name": project.name(), "version": __version__, "revision": 4, "topology": { "nodes": [{ "description": "VPCS device", "id": vpcs_device.id(), "ports": [{ "id": vpcs_device.ports()[0].id(), "name": "Ethernet0", "port_number": 0, "adapter_number": 0 }], "properties": { "name": vpcs_device.name() }, "server_id": local_server.id(), "type": "VPCSDevice", "vm_id": None }], "servers": [{ "vm": False, "host": "127.0.0.1", "id": local_server.id(), "local": True, "port": 8000, "protocol": "http", "ram_limit": 0 }] }, "type": "topology" }
def test_image_outside_project(tmpdir): """ By security we do not delete image outside project. This should not append but if someone reuse the image items for something else. """ project_img_dir = tmpdir / "project" / "project-files" / "images" os.makedirs(str(project_img_dir)) project = Project() project.setFilesDir(str(tmpdir / "project")) topology = Topology() topology.project = project with open(str(tmpdir / "1.jpg"), "w+") as f: f.write("AA") image1 = PixmapImageItem(None, str(tmpdir / "1.jpg")) topology.addImage(image1) assert image1 in topology._images topology.removeImage(image1) assert os.path.exists(str(tmpdir / "1.jpg"))
def test_load_1_2_topology(project, monkeypatch, main_window, tmpdir): topo = { "auto_start": False, "name": "twovpcs", "topology": { "links": [ { "description": "Link from VPCS 1 port Ethernet0 to VPCS 2 port Ethernet0", "destination_node_id": 2, "destination_port_id": 2, "id": 1, "source_node_id": 1, "source_port_id": 1 } ], "nodes": [ { "description": "VPCS device", "id": 1, "label": { "color": "#000000", "font": "TypeWriter,10,-1,5,75,0,0,0,0,0", "text": "VPCS 1", "x": 10.75, "y": -25.0 }, "ports": [ { "description": "connected to VPCS 2 on port Ethernet0", "id": 1, "link_id": 1, "name": "Ethernet0", "nio": "NIO_UDP", "port_number": 0 } ], "properties": { "console": 4501, "name": "VPCS 1", "script_file": "startup.vpc" }, "server_id": 1, "type": "VPCSDevice", "vpcs_id": 1, "x": -349.5, "y": -206.5 }, { "description": "VPCS device", "id": 2, "label": { "color": "#000000", "font": "TypeWriter,10,-1,5,75,0,0,0,0,0", "text": "VPCS 2", "x": 10.75, "y": -25.0 }, "ports": [ { "description": "connected to VPCS 1 on port Ethernet0", "id": 2, "link_id": 1, "name": "Ethernet0", "nio": "NIO_UDP", "port_number": 0 } ], "properties": { "console": 4502, "name": "VPCS 2", "script_file": "startup.vpc" }, "server_id": 1, "type": "VPCSDevice", "vpcs_id": 2, "x": 69.5, "y": -190.5 } ], "servers": [ { "cloud": False, "host": "127.0.0.1", "id": 1, "local": True, "port": 8000 } ] }, "type": "topology", "version": "1.2.3" } monkeypatch.setattr('gns3.main_window.MainWindow.instance', lambda: main_window) project_call = 0 # We return an uuid for each HTTP post def http_loader(self, method, path, callback, body={}, **kwargs): if path == "/projects": callback({"project_id": uuid.uuid4(), "path": str(tmpdir)}, error=False, server=local_server) project_call += 1 assert project_call < 2 elif path[-14:] == "/notifications": pass else: callback({"vm_id": uuid.uuid4()}) monkeypatch.setattr("gns3.http_client.HTTPClient.createHTTPQuery", http_loader) monkeypatch.setattr("gns3.http_client.HTTPClient.connected", lambda self: True) topology = Topology() topology.project = project topology._project.setName("unsaved") topology._project.setTopologyFile(str(tmpdir)) topology.dump = MagicMock() topology._load(topo) assert topology._project.id() is not None assert len(topology.nodes()) == 2 assert len(topology._node_to_links_mapping) == 2 assert topology.getNode(1).initialized() assert topology.getNode(2).initialized() assert main_window.uiGraphicsView.addLink.called assert main_window.saveProject.called
def test_load(project, monkeypatch, main_window, tmpdir): topo = { "project_id": project.id(), "auto_start": False, "name": "twovpcs", "resources_type": "local", "topology": { "links": [ { "description": "Link from VPCS 1 port Ethernet0 to VPCS 2 port Ethernet0", "destination_node_id": 2, "destination_port_id": 2, "id": 1, "source_node_id": 1, "source_port_id": 1 } ], "nodes": [ { "description": "VPCS device", "id": 1, "label": { "color": "#000000", "font": "TypeWriter,10,-1,5,75,0,0,0,0,0", "text": "VPCS 1", "x": 10.75, "y": -25.0 }, "ports": [ { "description": "connected to VPCS 2 on port Ethernet0", "id": 1, "link_id": 1, "name": "Ethernet0", "nio": "NIO_UDP", "port_number": 0, "adapter_number": 0 } ], "properties": { "console": 4501, "name": "VPCS 1", "script_file": "startup.vpc" }, "server_id": 1, "type": "VPCSDevice", "vpcs_id": 1, "vm_id": "2b5476de-6e79-4eb5-b0eb-8c54c7821cb8", "x": -349.5, "y": -206.5 }, { "description": "VPCS device", "id": 2, "label": { "color": "#000000", "font": "TypeWriter,10,-1,5,75,0,0,0,0,0", "text": "VPCS 2", "x": 10.75, "y": -25.0 }, "ports": [ { "description": "connected to VPCS 1 on port Ethernet0", "id": 2, "link_id": 1, "name": "Ethernet0", "nio": "NIO_UDP", "port_number": 0 } ], "properties": { "console": 4502, "name": "VPCS 2", "script_file": "startup.vpc" }, "server_id": 1, "type": "VPCSDevice", "vm_id": "2b5476de-6e79-4eb5-b0eb-8c54c7821cba", "vpcs_id": 2, "x": 69.5, "y": -190.5 } ], "servers": [ { "cloud": False, "host": "127.0.0.1", "id": 1, "local": True, "port": 8000 } ] }, "type": "topology", "version": "1.3.0" } monkeypatch.setattr('gns3.main_window.MainWindow.instance', lambda: main_window) # We return an uuid for each HTTP post def http_loader(self, method, path, callback, body={}, **kwargs): if path == "/projects": callback({"project_id": uuid.uuid4(), "path": str(tmpdir)}) else: callback({"vm_id": uuid.uuid4()}) monkeypatch.setattr("gns3.http_client.HTTPClient.createHTTPQuery", http_loader) monkeypatch.setattr("gns3.http_client.HTTPClient.connected", lambda self: True) topology = Topology() topology.project = project topology._load(topo) assert topology._project.id() == project.id() assert len(topology.nodes()) == 2 assert len(topology._node_to_links_mapping) == 2 assert topology.getNode(1).initialized() assert topology.getNode(2).initialized() assert main_window.uiGraphicsView.addLink.called
def test_load_1_2_topology(project, monkeypatch, main_window, tmpdir): topo = { "auto_start": False, "name": "twovpcs", "resources_type": "cloud", "topology": { "links": [{ "description": "Link from VPCS 1 port Ethernet0 to VPCS 2 port Ethernet0", "destination_node_id": 2, "destination_port_id": 2, "id": 1, "source_node_id": 1, "source_port_id": 1 }], "nodes": [{ "description": "VPCS device", "id": 1, "label": { "color": "#000000", "font": "TypeWriter,10,-1,5,75,0,0,0,0,0", "text": "VPCS 1", "x": 10.75, "y": -25.0 }, "ports": [{ "description": "connected to VPCS 2 on port Ethernet0", "id": 1, "link_id": 1, "name": "Ethernet0", "nio": "NIO_UDP", "port_number": 0 }], "properties": { "console": 4501, "name": "VPCS 1", "script_file": "startup.vpc" }, "server_id": 1, "type": "VPCSDevice", "vpcs_id": 1, "x": -349.5, "y": -206.5 }, { "description": "VPCS device", "id": 2, "label": { "color": "#000000", "font": "TypeWriter,10,-1,5,75,0,0,0,0,0", "text": "VPCS 2", "x": 10.75, "y": -25.0 }, "ports": [{ "description": "connected to VPCS 1 on port Ethernet0", "id": 2, "link_id": 1, "name": "Ethernet0", "nio": "NIO_UDP", "port_number": 0 }], "properties": { "console": 4502, "name": "VPCS 2", "script_file": "startup.vpc" }, "server_id": 1, "type": "VPCSDevice", "vpcs_id": 2, "x": 69.5, "y": -190.5 }], "servers": [{ "cloud": False, "host": "127.0.0.1", "id": 1, "local": True, "port": 8000 }] }, "type": "topology", "version": "1.2.3" } monkeypatch.setattr('gns3.main_window.MainWindow.instance', lambda: main_window) # We return an uuid for each HTTP post def http_loader(self, method, path, callback, body={}, **kwargs): if path == "/projects": callback({ "project_id": uuid.uuid4(), "path": str(tmpdir) }, error=False, server=local_server) else: callback({"vm_id": uuid.uuid4()}) monkeypatch.setattr("gns3.http_client.HTTPClient.createHTTPQuery", http_loader) monkeypatch.setattr("gns3.http_client.HTTPClient.connected", lambda self: True) topology = Topology() topology.project = project topology._project.setName("unsaved") topology._project.setTopologyFile(str(tmpdir)) topology.dump = MagicMock() topology._load(topo) assert topology._project.id() is not None assert len(topology.nodes()) == 2 assert len(topology._node_to_links_mapping) == 2 assert topology.getNode(1).initialized() assert topology.getNode(2).initialized() assert main_window.uiGraphicsView.addLink.called assert main_window.saveProject.called
def test_dump_http_auth(vpcs_device, project, remote_server): remote_server.setUser("hello") remote_server.setPassword("world") from gns3.modules.vpcs.vpcs_device import VPCSDevice from gns3.modules.vpcs import VPCS vpcs_device = VPCSDevice(VPCS(), remote_server, project) vpcs_device._vpcs_device_id = str(uuid.uuid4()) vpcs_device._settings = { "name": "VPCS 1", "script_file": "", "console": None, "startup_script": None } vpcs_device.setInitialized(True) topology = Topology() topology.project = project topology.addNode(vpcs_device) dump = topology.dump(include_gui_data=False) assert dict(dump) == { "project_id": project.id(), "auto_start": False, "name": project.name(), "version": __version__, "revision": 4, "topology": { "nodes": [{ "description": "VPCS device", "id": vpcs_device.id(), "ports": [{ "id": vpcs_device.ports()[0].id(), "name": "Ethernet0", "port_number": 0, "adapter_number": 0 }], "properties": { "name": vpcs_device.name() }, "server_id": remote_server.id(), "type": "VPCSDevice", "vm_id": None }], "servers": [{ "vm": False, "host": "127.0.0.1", "id": remote_server.id(), "local": False, "port": 8001, "protocol": "http", "ram_limit": 0, "user": "******" }] }, "type": "topology" }