T5050: fix smoketest policy_route, which was failing after previos commit was merged

1 files changed, 14 insertions, 12 deletions

diff --git a/python/vyos/firewall.py b/python/vyos/firewall.py
index 137fd2c56..919032a41 100644
--- a/python/vyos/firewall.py
+++ b/python/vyos/firewall.py
@@ -223,21 +223,23 @@ def parse_rule(rule_conf, fw_name, rule_id, ip_name):
         action = rule_conf['action'] if 'action' in rule_conf else 'accept'
         output.append(f'log prefix "[{fw_name[:19]}-{rule_id}-{action[:1].upper()}]"')
 
-        if 'level' in rule_conf['log_options']:
-            log_level = rule_conf['log_options']['level']
-            output.append(f'log level {log_level}')
+        if 'log_options' in rule_conf:
 
-        if 'group' in rule_conf['log_options']:
-            log_group = rule_conf['log_options']['group']
-            output.append(f'log group {log_group}')
+            if 'level' in rule_conf['log_options']:
+                log_level = rule_conf['log_options']['level']
+                output.append(f'log level {log_level}')
 
-            if 'queue_threshold' in rule_conf['log_options']:
-                queue_threshold = rule_conf['log_options']['queue_threshold']
-                output.append(f'queue-threshold {queue_threshold}')
+            if 'group' in rule_conf['log_options']:
+                log_group = rule_conf['log_options']['group']
+                output.append(f'log group {log_group}')
 
-            if 'snapshot_length' in rule_conf['log_options']:
-                log_snaplen = rule_conf['log_options']['snapshot_length']
-                output.append(f'snaplen {log_snaplen}')
+                if 'queue_threshold' in rule_conf['log_options']:
+                    queue_threshold = rule_conf['log_options']['queue_threshold']
+                    output.append(f'queue-threshold {queue_threshold}')
+
+                if 'snapshot_length' in rule_conf['log_options']:
+                    log_snaplen = rule_conf['log_options']['snapshot_length']
+                    output.append(f'snaplen {log_snaplen}')
 
     if 'hop_limit' in rule_conf:
         operators = {'eq': '==', 'gt': '>', 'lt': '<'}