policer counter tests

sonic-net · Dec 18, 2024 · 119d11d · 119d11d
1 parent 68893a4
commit 119d11d
Show file tree

Hide file tree

Showing 8 changed files with 470 additions and 257 deletions.
diff --git a/counterpoll/main.py b/counterpoll/main.py
@@ -259,38 +259,38 @@ def disable():
 
 # Policer counter commands
 @cli.group()
-def policer():
+@click.pass_context
+def policer(ctx):
     """ Policer counter commands """
+    ctx.obj = ConfigDBConnector()
+    ctx.obj.connect()
 
 @policer.command()
+@click.pass_context
 @click.argument('poll_interval')
-def interval(poll_interval):
+def interval(ctx, poll_interval):
     """ Set policer counter query interval """
-    configdb = ConfigDBConnector()
-    configdb.connect()
     policer_info = {}
     if poll_interval is not None:
         policer_info['POLL_INTERVAL'] = poll_interval
-    configdb.mod_entry("FLEX_COUNTER_TABLE", POLICER, policer_info)
+    ctx.obj.mod_entry("FLEX_COUNTER_TABLE", POLICER, policer_info)
 
 
 @policer.command()
-def enable():
+@click.pass_context
+def enable(ctx):
     """ Enable policer counter query """
-    configdb = ConfigDBConnector()
-    configdb.connect()
     policer_info = {}
     policer_info['FLEX_COUNTER_STATUS'] = ENABLE
-    configdb.mod_entry("FLEX_COUNTER_TABLE", POLICER, policer_info)
+    ctx.obj.mod_entry("FLEX_COUNTER_TABLE", POLICER, policer_info)
 
 @policer.command()
-def disable():
+@click.pass_context
+def disable(ctx):
     """ Disable policer counter query """
-    configdb = ConfigDBConnector()
-    configdb.connect()
     policer_info = {}
     policer_info['FLEX_COUNTER_STATUS'] = DISABLE
-    configdb.mod_entry("FLEX_COUNTER_TABLE", POLICER, policer_info)
+    ctx.obj.mod_entry("FLEX_COUNTER_TABLE", POLICER, policer_info)
 
 # ACL counter commands
 @cli.group()