[rsyslog] Disable rsyslog rate limit in pre_test and do a recover in post_test

tests/test_posttest.py

@@ -3,6 +3,7 @@
 import os.path
 import os
 import json
+import time
 
 logger = logging.getLogger(__name__)
 
@@ -54,3 +55,24 @@ def test_restore_container_autorestart(duthost):
     cmds_enable.append("config save -y")
     duthost.shell_cmds(cmds=cmds_enable)
     os.remove(state_file_name)
+    # Wait sometime for snmp reloading
+    SNMP_RELOADING_TIME = 30
+    time.sleep(SNMP_RELOADING_TIME)
+
+def test_recover_rsyslog_rate_limit(duthost):
+    features_dict, succeed = duthost.get_feature_status()
+    if not succeed:
+        # Something unexpected happened.
+        # We don't want to fail here because it's an util
+        logging.warn("Failed to retrieve feature status")
+        return
+    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":
+            continue
+        cmds = []
+        cmds.append(cmd_enable_rate_limit.format(feature_name))
+        cmds.append(cmd_reload.format(feature_name))
+        duthost.shell_cmds(cmds=cmds)
+

tests/test_pretest.py

@@ -1,12 +1,14 @@
 import pytest
 import logging
 import json
+import time
 
 logger = logging.getLogger(__name__)
 
 pytestmark = [
     pytest.mark.pretest,
-    pytest.mark.topology('util')
+    pytest.mark.topology('util'),
+    pytest.mark.disable_loganalyzer
 ]
 
 def test_cleanup_testbed(duthost, request, ptfhost):
@@ -45,5 +47,24 @@ def test_disable_container_autorestart(duthost):
     # Write into config_db
     cmds_disable.append("config save -y")
     duthost.shell_cmds(cmds=cmds_disable)
+    # Wait sometime for snmp reloading
+    SNMP_RELOADING_TIME = 30
+    time.sleep(SNMP_RELOADING_TIME)
 
+def test_disable_rsyslog_rate_limit(duthost):
+    features_dict, succeed = duthost.get_feature_status()
+    if not succeed:
+        # Something unexpected happened.
+        # We don't want to fail here because it's an util
+        logging.warn("Failed to retrieve feature status")
+        return
+    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":
+            continue
+        cmds = []
+        cmds.append(cmd_disable_rate_limit.format(feature_name))
+        cmds.append(cmd_reload.format(feature_name))
+        duthost.shell_cmds(cmds=cmds)