async def test_update_filters(project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.emit_notification = MagicMock() project.dump = AsyncioMagicMock() await link.add_node(node1, 0, 4) node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] await link.add_node(node2, 0, 4) link.update = AsyncioMagicMock() assert link._created await link.update_filters({ "packet_loss": [10], "delay": [50, 10], "frequency_drop": [0], "bpf": [" \n "] }) assert link.filters == { "packet_loss": [10], "delay": [50, 10] } assert link.update.called
async def test_add_node(project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.emit_notification = MagicMock() project.dump = AsyncioMagicMock() await link.add_node(node1, 0, 4) assert link._nodes == [ { "node": node1, "port": node1._ports[0], "adapter_number": 0, "port_number": 4, 'label': { 'text': '0/4', 'style': 'font-family: TypeWriter;font-size: 10.0;font-weight: bold;fill: #000000;fill-opacity: 1.0;' } } ] assert project.dump.called assert not link._project.emit_notification.called assert not link.create.called # We call link.created only when both side are created node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] await link.add_node(node2, 0, 4) assert link.create.called link._project.emit_notification.assert_called_with("link.created", link.__json__()) assert link in node2.links
def link(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 1, 3)] link = Link(project) link.create = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 1, 3)) return link
def test_json_serial_link(async_run, project, compute, link): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [SerialPort("S0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [SerialPort("S0", 0, 1, 3)] link = Link(project) link.create = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 1, 3)) assert link.__json__()["link_type"] == "serial"
def test_default_capture_file_name(project, compute, async_run): node1 = Node(project, compute, "Hello@", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "w0.rld", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 1, 3)] link = Link(project) link.create = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 1, 3)) assert link.default_capture_file_name() == "Hello_0-4_to_w0rld_1-3.pcap"
def test_add_node_cloud(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="cloud") node2._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 0, 4))
async def test_add_node_cloud(project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="cloud") node2._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.emit_notification = MagicMock() await link.add_node(node1, 0, 4) await link.add_node(node2, 0, 4)
def test_add_node_same_node(async_run, project, compute): """ Connection to the same node is not allowed """ node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4), EthernetPort("E1", 0, 0, 5)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() async_run(link.add_node(node1, 0, 4)) with pytest.raises(aiohttp.web.HTTPConflict): async_run(link.add_node(node1, 0, 5))
def test_add_node_serial_to_ethernet(async_run, project, compute): """ Serial to ethernet connection is not allowed """ node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [SerialPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() async_run(link.add_node(node1, 0, 4)) with pytest.raises(aiohttp.web.HTTPConflict): async_run(link.add_node(node2, 0, 4))
def test_add_node_cloud_to_cloud(async_run, project, compute): """ Cloud to cloud connection is not allowed """ node1 = Node(project, compute, "node1", node_type="cloud") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="cloud") node2._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.emit_notification = MagicMock() async_run(link.add_node(node1, 0, 4)) with pytest.raises(aiohttp.web.HTTPConflict): async_run(link.add_node(node2, 0, 4))
def test_stop_capture(http_controller, tmpdir, project, compute, async_run): link = Link(project) project._links = {link.id: link} with asyncio_patch("gns3server.controller.link.Link.stop_capture") as mock: response = http_controller.post("/projects/{}/links/{}/stop_capture".format(project.id, link.id), example=True) assert mock.called assert response.status == 201
async def test_delete(project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.emit_notification = MagicMock() project.dump = AsyncioMagicMock() await link.add_node(node1, 0, 4) node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] await link.add_node(node2, 0, 4) assert link in node2.links await link.delete() assert link not in node2.links
def test_list_filters(http_controller, tmpdir, project, async_run): link = Link(project) project._links = {link.id: link} with patch("gns3server.controller.link.Link.available_filters", return_value=FILTERS) as mock: response = http_controller.get("/projects/{}/links/{}/available_filters".format(project.id, link.id), example=True) assert mock.called assert response.status == 200 assert response.json == FILTERS
def test_delete(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() project.dump = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] async_run(link.add_node(node2, 0, 4)) assert link in node2.link async_run(link.delete()) assert link not in node2.link
def test_pcap(http_controller, tmpdir, project, compute, loop): @asyncio.coroutine def go(future): response = yield from aiohttp.request("GET", http_controller.get_url("/projects/{}/links/{}/pcap".format(project.id, link.id))) response.body = yield from response.content.read(5) response.close() future.set_result(response) link = Link(project) link._capture_file_name = "test" link._capturing = True with open(link.capture_file_path, "w+") as f: f.write("hello") project._links = {link.id: link} future = asyncio.Future() asyncio.async(go(future)) response = loop.run_until_complete(future) assert response.status == 200 assert b'hello' == response.body
def test_update_filters(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.emit_notification = MagicMock() project.dump = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] async_run(link.add_node(node2, 0, 4)) link.update = AsyncioMagicMock() assert link._created async_run(link.update_filters({ "packet_loss": [10], "delay": [50, 10], "frequency_drop": [0], "bpf": [" \n "] })) assert link.filters == { "packet_loss": [10], "delay": [50, 10] } assert link.update.called
def test_available_filters(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="ethernet_switch") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() assert link.available_filters() == [] # Ethernet switch is not supported should return 0 filters async_run(link.add_node(node1, 0, 4)) assert link.available_filters() == [] node2 = Node(project, compute, "node2", node_type="vpcs") node2._ports = [EthernetPort("E0", 0, 0, 4)] async_run(link.add_node(node2, 0, 4)) assert len(link.available_filters()) > 0
def test_add_node(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() project.dump = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) assert link._nodes == [ { "node": node1, "port": node1._ports[0], "adapter_number": 0, "port_number": 4, 'label': { 'y': -10, 'text': '0/4', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } } ] assert project.dump.called assert not link._project.controller.notification.emit.called assert not link.create.called # We call link.created only when both side are created node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] async_run(link.add_node(node2, 0, 4)) assert link.create.called link._project.controller.notification.emit.assert_called_with("link.created", link.__json__()) assert link in node2.link
def test_add_node_already_connected(async_run, project, compute): """ Raise an error if we try to use an already connected port """ project.dump = AsyncioMagicMock() node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() async_run(link.add_node(node1, 0, 4)) node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] async_run(link.add_node(node2, 0, 4)) assert link.create.called link2 = Link(project) link2.create = AsyncioMagicMock() with pytest.raises(aiohttp.web.HTTPConflict): async_run(link2.add_node(node1, 0, 4))
def test_add_node(async_run, project, compute): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] link = Link(project) link.create = AsyncioMagicMock() link._project.controller.notification.emit = MagicMock() project.dump = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) assert link._nodes == [{ "node": node1, "port": node1._ports[0], "adapter_number": 0, "port_number": 4, 'label': { 'y': -10, 'text': '0/4', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }] assert project.dump.called assert not link._project.controller.notification.emit.called assert not link.create.called # We call link.created only when both side are created node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 0, 4)] async_run(link.add_node(node2, 0, 4)) assert link.create.called link._project.controller.notification.emit.assert_called_with( "link.created", link.__json__()) assert link in node2.link
def test_eq(project, link, controller): assert link == Link(project, link_id=link.id) assert link != "a" assert link != Link(project)
def test_eq(project, link): assert link == Link(project, link_id=link.id) assert link != "a" assert link != Link(project)
def test_json(async_run, project, compute, link): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 1, 3)] link = Link(project) link.create = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 1, 3)) assert link.__json__() == { "link_id": link.id, "project_id": project.id, "nodes": [{ "node_id": node1.id, "adapter_number": 0, "port_number": 4, 'label': { 'y': -10, 'text': '0/4', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }, { "node_id": node2.id, "adapter_number": 1, "port_number": 3, 'label': { 'y': -10, 'text': '1/3', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }], "link_type": "ethernet", "capturing": False, "capture_file_name": None, "capture_file_path": None } assert link.__json__(topology_dump=True) == { "link_id": link.id, "nodes": [{ "node_id": node1.id, "adapter_number": 0, "port_number": 4, 'label': { 'y': -10, 'text': '0/4', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }, { "node_id": node2.id, "adapter_number": 1, "port_number": 3, 'label': { 'y': -10, 'text': '1/3', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }] }
def test_json(async_run, project, compute, link): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 1, 3)] link = Link(project) link.create = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 1, 3)) assert link.__json__() == { "link_id": link.id, "project_id": project.id, "nodes": [ { "node_id": node1.id, "adapter_number": 0, "port_number": 4, 'label': { 'text': '0/4', 'style': 'font-family: TypeWriter;font-size: 10.0;font-weight: bold;fill: #000000;fill-opacity: 1.0;' } }, { "node_id": node2.id, "adapter_number": 1, "port_number": 3, 'label': { 'text': '1/3', 'style': 'font-family: TypeWriter;font-size: 10.0;font-weight: bold;fill: #000000;fill-opacity: 1.0;' } } ], "filters": {}, "suspend": False, "link_type": "ethernet", "capturing": False, "capture_file_name": None, "capture_file_path": None, "capture_compute_id": None } assert link.__json__(topology_dump=True) == { "link_id": link.id, "nodes": [ { "node_id": node1.id, "adapter_number": 0, "port_number": 4, 'label': { 'text': '0/4', 'style': 'font-family: TypeWriter;font-size: 10.0;font-weight: bold;fill: #000000;fill-opacity: 1.0;' } }, { "node_id": node2.id, "adapter_number": 1, "port_number": 3, 'label': { 'text': '1/3', 'style': 'font-family: TypeWriter;font-size: 10.0;font-weight: bold;fill: #000000;fill-opacity: 1.0;' } } ], "filters": {}, "suspend": False }
def test_json(async_run, project, compute, link): node1 = Node(project, compute, "node1", node_type="qemu") node1._ports = [EthernetPort("E0", 0, 0, 4)] node2 = Node(project, compute, "node2", node_type="qemu") node2._ports = [EthernetPort("E0", 0, 1, 3)] link = Link(project) link.create = AsyncioMagicMock() async_run(link.add_node(node1, 0, 4)) async_run(link.add_node(node2, 1, 3)) assert link.__json__() == { "link_id": link.id, "project_id": project.id, "nodes": [ { "node_id": node1.id, "adapter_number": 0, "port_number": 4, 'label': { 'y': -10, 'text': '0/4', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }, { "node_id": node2.id, "adapter_number": 1, "port_number": 3, 'label': { 'y': -10, 'text': '1/3', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } } ], "link_type": "ethernet", "capturing": False, "capture_file_name": None, "capture_file_path": None } assert link.__json__(topology_dump=True) == { "link_id": link.id, "nodes": [ { "node_id": node1.id, "adapter_number": 0, "port_number": 4, 'label': { 'y': -10, 'text': '0/4', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } }, { "node_id": node2.id, "adapter_number": 1, "port_number": 3, 'label': { 'y': -10, 'text': '1/3', 'x': -10, 'rotation': 0, 'style': 'font-size: 10; font-style: Verdana' } } ] }