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_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_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_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"
}
