Fix enabld/disable rsyslog rate limitation. (#3279)

This PR is to fix test failure of test_disable_rsyslog_rate_limit and test_recover_rsyslog_rate_limit in pretest and posttest. The state of feature may be always_enabled or always_disabled. Therefore, we have to update the logic to check if a feature is enabled. Signed-off-by: bingwang <[email protected]>
sonic-net · Apr 6, 2021 · 18dce2d · 18dce2d
1 parent 0f45e23
commit 18dce2d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_posttest.py b/tests/test_posttest.py
@@ -51,7 +51,7 @@ def test_recover_rsyslog_rate_limit(duthosts, enum_dut_hostname):
     cmd_enable_rate_limit = r"docker exec -i {} sed -i 's/^#\$SystemLogRateLimit/\$SystemLogRateLimit/g' /etc/rsyslog.conf"
     cmd_reload = r"docker exec -i {} supervisorctl restart rsyslogd"
     for feature_name, state in features_dict.items():
-        if state == "disabled":
+        if 'enabled' not in state:
             continue
         cmds = []
         cmds.append(cmd_enable_rate_limit.format(feature_name))

diff --git a/tests/test_pretest.py b/tests/test_pretest.py
@@ -85,7 +85,7 @@ def test_disable_rsyslog_rate_limit(duthosts, enum_dut_hostname):
     cmd_disable_rate_limit = r"docker exec -i {} sed -i 's/^\$SystemLogRateLimit/#\$SystemLogRateLimit/g' /etc/rsyslog.conf"
     cmd_reload = r"docker exec -i {} supervisorctl restart rsyslogd"
     for feature_name, state in features_dict.items():
-        if state == "disabled":
+        if 'enabled' not in state:
             continue
         cmds = []
         cmds.append(cmd_disable_rate_limit.format(feature_name))