diff --git a/inventory_provider/juniper.py b/inventory_provider/juniper.py index 2c3b0b82768aeddc4aa81f03abfe1047e04f8f9c..07fd597773534794c89ed0c6a2479452ce038cfd 100644 --- a/inventory_provider/juniper.py +++ b/inventory_provider/juniper.py @@ -243,6 +243,36 @@ def list_interfaces(netconf_config): yield u +def asn_to_int(asn_string: str) -> int: + """ + Convert a possibly dotted ASN to an integer. + + Args: + asn_string (str): ASN to be converted, can be in dot notation or not. + + Returns: + int: ASN in integer format. + + Raises: + ValueError: If the ASN string is not in the expected format or exceeds valid range. + """ + + dotted_asn_pattern = re.compile(r'^(\d+)\.(\d+)$') + match = dotted_asn_pattern.match(asn_string) + + if match: + high_order, low_order = map(int, match.groups()) + + if high_order > 0xffff or low_order > 0xffff: + raise ValueError(f'Invalid ASN format: {asn_string}. Both components must be <= 0xffff.') + + return (high_order << 16) | low_order + elif asn_string.isdigit(): + return int(asn_string) + else: + raise ValueError(f'Unable to parse ASN string: {asn_string}. Expected either a pure integer or a dot notation.') + + def _system_bgp_peers(system_node): def _peering_params(neighbor_node): @@ -251,11 +281,11 @@ def _system_bgp_peers(system_node): peer_as = neighbor_node.find('peer-as') if peer_as is not None: # lxml usage warning: can't just test `if peer_as:` - info['remote-asn'] = int(peer_as.text) + info['remote-asn'] = asn_to_int(peer_as.text) local_as = neighbor_node.find('local-as') if local_as is not None: asn_value_node = local_as.find('as-number') - info['local-asn'] = int(asn_value_node.text) + info['local-asn'] = asn_to_int(asn_value_node.text) description = neighbor_node.find('description') if description is not None: # lxml usage warning: can't just test `if description:` diff --git a/test/test_juniper_data_global.py b/test/test_juniper_data_global.py index accb5d6f43bec6b39ef8b3bd1465c502d82cdb04..12a1a3762c20a860c517a8ac58b8267caf6da0ae 100644 --- a/test/test_juniper_data_global.py +++ b/test/test_juniper_data_global.py @@ -5,6 +5,7 @@ import ipaddress import pytest from inventory_provider import juniper +from inventory_provider.juniper import asn_to_int NETIFACES_TEST_DATA_STRING = """{ 'lo0': {{AF_INET}: [{'addr': '127.0.0.1', 'netmask': '255.0.0.0', 'peer': '127.0.0.1'}], @@ -70,3 +71,21 @@ def test_local_v6_interfaces(mocked_netifaces): assert len(addresses) == 4 for a in addresses: assert isinstance(a, ipaddress.IPv6Interface) + + +def test_asn_to_int_functionality(): + """Test that ASN to int conversion works with and without dot notation.""" + + # Test with dot notation + assert asn_to_int('1.1') == 65537 + assert asn_to_int('64512.2') == 4227858434 + assert asn_to_int('0.65535') == 65535 + + # Test without dot notation + assert asn_to_int('65537') == 65537 + assert asn_to_int('0') == 0 + assert asn_to_int('4227858434') == 4227858434 + + with pytest.raises(ValueError): + asn_to_int('66512.2') + asn_to_int('Test')