From 6f91d90b135979bb32904312ede46a789bd5db8c Mon Sep 17 00:00:00 2001
From: Lluis Gifre <lluis.gifre@cttc.es>
Date: Fri, 29 Jul 2022 13:05:44 +0200
Subject: [PATCH] Device component - OpenConfigDriver:
- updated to new protobuf definitions for config rules
---
src/device/tests/test_unitary_openconfig.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/device/tests/test_unitary_openconfig.py b/src/device/tests/test_unitary_openconfig.py
index 47bea7ef6..32fb5709a 100644
--- a/src/device/tests/test_unitary_openconfig.py
+++ b/src/device/tests/test_unitary_openconfig.py
@@ -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'
]
--
GitLab