Commit defad037 authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Device component: 

- fixed parsing of custom rules in unitary tests
parent d09c0f7a

src/device/tests/test_unitary_openconfig.py

+5 −3
Original line number Diff line number Diff line
@@ -88,8 +88,9 @@ def test_device_openconfig_add_correct( 


    device_data = context_client.GetDevice(DeviceId(**DEVICE_OC_ID))

    config_rules = [

        (ConfigActionEnum.Name(config_rule.action), config_rule.resource_key, config_rule.resource_value)

        (ConfigActionEnum.Name(config_rule.action), config_rule.custom.resource_key, config_rule.custom.resource_value)

        for config_rule in device_data.device_config.config_rules

        if config_rule.WhichOneof('config_rule') == 'custom'

    ]

    LOGGER.info('device_data.device_config.config_rules = \n{:s}'.format(

        '\n'.join(['{:s} {:s} = {:s}'.format(*config_rule) for config_rule in config_rules])))
@@ -135,8 +136,9 @@ def test_device_openconfig_configure( 


    device_data = context_client.GetDevice(DeviceId(**DEVICE_OC_ID))

    config_rules = [

        (ConfigActionEnum.Name(config_rule.action), config_rule.resource_key, config_rule.resource_value)

        (ConfigActionEnum.Name(config_rule.action), config_rule.custom.resource_key, config_rule.custom.resource_value)

        for config_rule in device_data.device_config.config_rules

        if config_rule.WhichOneof('config_rule') == 'custom'

    ]

    LOGGER.info('device_data.device_config.config_rules = \n{:s}'.format(

        '\n'.join(['{:s} {:s} = {:s}'.format(*config_rule) for config_rule in config_rules])))
@@ -276,7 +278,7 @@ def test_device_openconfig_deconfigure( 


    device_data = context_client.GetDevice(DeviceId(**DEVICE_OC_ID))

    config_rules = [

        (ConfigActionEnum.Name(config_rule.action), config_rule.resource_key, config_rule.resource_value)

        (ConfigActionEnum.Name(config_rule.action), config_rule.custom.resource_key, config_rule.custom.resource_value)

        for config_rule in device_data.device_config.config_rules

        if config_rule.WhichOneof('config_rule') == 'custom'

    ]