def do_ncopy(self, protocol, url, filename, target):
try:
SwitchTftpParser(self.switch_configuration).parse(url, filename, ConfigCommandProcessor)
self.write_line("done")
except Exception as e:
self.logger.warning("tftp parsing went wrong : %s" % str(e))
self.write_line("%s: Download to %s failed - Session timed out" % (protocol.upper(), target))
def test_basic_parsing(self, tftp_reader_mock):
tftp_reader_mock.return_value = """
!
vlan 1000
name VLAN_1_0_0_0
!
!
vlan 2000
name VLAN_2_0_0_0
!
"""
config = SwitchConfiguration("127.0.0.1",
name="my_switch",
ports=[Port("GigabitEthernet0/1")])
parser = SwitchTftpParser(config)
config_processor = ConfigCommandProcessor(
config_vlan=ConfigVlanCommandProcessor(),
config_vrf=None,
config_interface=None)
parser.parse("hostname", "filename", config_processor)
tftp_reader_mock.assert_called_with("hostname", "filename")
vlan1000 = config.get_vlan(1000)
assert_that(vlan1000.name, equal_to("VLAN_1_0_0_0"))
vlan2000 = config.get_vlan(2000)
assert_that(vlan2000.name, equal_to("VLAN_2_0_0_0"))
def continue_validate_copy(self, source_url, _):
self.write_line("Accessing %s..." % source_url)
try:
url, filename = re.match('tftp://([^/]*)/(.*)', source_url).group(1, 2)
SwitchTftpParser(self.switch_configuration).parse(url, filename, ConfigCommandProcessor)
self.write_line("Done (or some official message...)")
except Exception as e:
self.logger.warning("tftp parsing went wrong : %s" % str(e))
self.write_line("Error opening %s (Timed out)" % source_url)
def test_longer_parsing(self, tftp_reader_mock):
tftp_reader_mock.return_value = """
!
vlan 1000
name VLAN_1_0_0_0
!
!
interface GigabitEthernet0/1
no switchport access vlan 1
switchport access vlan 1000
!
!
interface GigabitEthernet0/1
description "Gigabit Ethernet 1 desc"
switchport
load-interval 30
switchport mode access
switchport access vlan 1000
switchport nonegotiate
spanning-tree portfast
spanning-tree bpdufilter enable
spanning-tree bpduguard enable
no loopback
no keepalive
!
"""
config = SwitchConfiguration("127.0.0.1",
name="my_switch",
ports=[Port("GigabitEthernet0/1")])
parser = SwitchTftpParser(config)
config_processor = ConfigCommandProcessor(
config_vlan=ConfigVlanCommandProcessor(),
config_vrf=None,
config_interface=ConfigInterfaceCommandProcessor())
parser.parse("hostname", "filename", config_processor)
vlan1000 = config.get_vlan(1000)
assert_that(vlan1000.name, equal_to("VLAN_1_0_0_0"))
eth01 = config.get_port("GigabitEthernet0/1")
assert_that(eth01.description, equal_to("Gigabit Ethernet 1 desc"))
assert_that(eth01.mode, equal_to("access"))
assert_that(eth01.access_vlan, equal_to(1000))