mirror of
https://github.com/GNS3/gns3-gui.git
synced 2026-05-17 00:46:01 +03:00
144 lines
4.4 KiB
Python
144 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2013 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 uuid
|
|
|
|
|
|
from unittest.mock import patch, Mock, MagicMock
|
|
from gns3.modules.vpcs.vpcs_node import VPCSNode
|
|
from gns3.node import Node
|
|
from gns3.ports.port import Port
|
|
from gns3.ports.ethernet_port import EthernetPort
|
|
from gns3.ports.serial_port import SerialPort
|
|
|
|
|
|
def test_setupVMCallback(vpcs_device):
|
|
node_id = str(uuid.uuid4())
|
|
vpcs_device._createCallback = MagicMock()
|
|
vpcs_device.createNodeCallback({
|
|
"name": "PC 1",
|
|
"node_id": node_id,
|
|
"properties": {
|
|
"startup_script": "echo TEST"
|
|
}
|
|
})
|
|
assert vpcs_device._node_id == node_id
|
|
assert vpcs_device._settings["startup_script"] == "echo TEST"
|
|
vpcs_device._createCallback.assert_called_with({
|
|
"name": "PC 1",
|
|
"node_id": node_id,
|
|
"startup_script": "echo TEST"
|
|
}
|
|
)
|
|
|
|
|
|
def test_vpcs_device_start(vpcs_device):
|
|
|
|
with patch('gns3.base_node.BaseNode.controllerHttpPost') as mock:
|
|
vpcs_device.start()
|
|
assert mock.called
|
|
args, kwargs = mock.call_args
|
|
assert args[0] == "/nodes/{node_id}/start".format(node_id=vpcs_device.node_id())
|
|
|
|
|
|
def test_vpcs_device_stop(vpcs_device):
|
|
|
|
with patch('gns3.base_node.BaseNode.controllerHttpPost') as mock:
|
|
vpcs_device.setStatus(Node.started)
|
|
vpcs_device.stop()
|
|
assert mock.called
|
|
args, kwargs = mock.call_args
|
|
assert args[0] == "/nodes/{node_id}/stop".format(node_id=vpcs_device.node_id())
|
|
|
|
|
|
def test_vpcs_device_reload(vpcs_device):
|
|
|
|
with patch('gns3.base_node.BaseNode.controllerHttpPost') as mock:
|
|
vpcs_device.reload()
|
|
assert mock.called
|
|
args, kwargs = mock.call_args
|
|
assert args[0] == "/nodes/{node_id}/reload".format(node_id=vpcs_device.node_id())
|
|
|
|
|
|
def test_updatePorts(vpcs_device):
|
|
vpcs_device._updatePorts([
|
|
{
|
|
"name": "Ethernet0/0",
|
|
"short_name": "e0/0",
|
|
"data_link_types": {"Ethernet": "DLT_EN10MB"},
|
|
"port_number": 0,
|
|
"adapter_number": 0,
|
|
"link_type": "ethernet"
|
|
}
|
|
])
|
|
assert len(vpcs_device._ports) == 1
|
|
port = vpcs_device._ports[0]
|
|
assert port.name() == "Ethernet0/0"
|
|
assert port.shortName() == "e0/0"
|
|
assert port.portNumber() == 0
|
|
assert port.adapterNumber() == 0
|
|
assert port.dataLinkTypes() == {"Ethernet": "DLT_EN10MB"}
|
|
assert port.status() == Port.stopped
|
|
assert isinstance(port, EthernetPort)
|
|
|
|
vpcs_device.setStatus(Node.started)
|
|
vpcs_device._updatePorts([
|
|
{
|
|
"name": "Serial0/0",
|
|
"short_name": "s0/0",
|
|
"data_link_types": {},
|
|
"port_number": 0,
|
|
"adapter_number": 0,
|
|
"link_type": "serial"
|
|
}
|
|
])
|
|
assert len(vpcs_device._ports) == 1
|
|
port = vpcs_device._ports[0]
|
|
assert port.status() == Port.started
|
|
assert isinstance(port, SerialPort)
|
|
|
|
|
|
def test_updatePorts_PortChange(vpcs_device):
|
|
"""
|
|
If the same port we do not recreate it but just update his informations
|
|
"""
|
|
vpcs_device._updatePorts([
|
|
{
|
|
"name": "Ethernet0/0",
|
|
"short_name": "e0/0",
|
|
"data_link_types": {"Ethernet": "DLT_EN10MB"},
|
|
"port_number": 0,
|
|
"adapter_number": 0,
|
|
"link_type": "ethernet"
|
|
}
|
|
])
|
|
port = vpcs_device._ports[0]
|
|
|
|
vpcs_device.setStatus(Node.started)
|
|
vpcs_device._updatePorts([
|
|
{
|
|
"name": "Ethernet0/0",
|
|
"short_name": "e0/0",
|
|
"data_link_types": {"Ethernet": "DLT_EN10MB"},
|
|
"port_number": 0,
|
|
"adapter_number": 0,
|
|
"link_type": "ethernet"
|
|
}
|
|
])
|
|
assert port == vpcs_device._ports[0]
|
|
assert port.status() == Port.started
|