summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--test/test_flowprobe.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/test/test_flowprobe.py b/test/test_flowprobe.py
index 757226b0db8..5ffb7ce8efc 100644
--- a/test/test_flowprobe.py
+++ b/test/test_flowprobe.py
@@ -40,9 +40,11 @@ class VppCFLOW(VppObject):
def add_vpp_config(self):
self.enable_exporter()
- self._test.vapi.ppcli("flowprobe params record %s active %s "
- "passive %s" % (self._collect, self._active,
- self._passive))
+ self._test.vapi.flowprobe_params(
+ record_l2=1 if 'l2' in self._collect.lower() else 0,
+ record_l3=1 if 'l3' in self._collect.lower() else 0,
+ record_l4=1 if 'l4' in self._collect.lower() else 0,
+ active_timer=self._active, passive_timer=self._passive)
self.enable_flowprobe_feature()
self._test.vapi.cli("ipfix flush")
self._configured = True