X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Ftest_flowprobe.py;h=f1f8597219d1fae71eec20f54adcd1b51f211bcd;hb=8800f732f868bf54da8adba05e38bd2477895ca5;hp=234cb3b55c7b3aa05ca7b4ef1f23c5174c3336c6;hpb=af5684bf18077acf1f448c6f2a62ef1af9f9be05;p=vpp.git

diff --git a/test/test_flowprobe.py b/test/test_flowprobe.py
index 234cb3b55c7..f1f8597219d 100644
--- a/test/test_flowprobe.py
+++ b/test/test_flowprobe.py
@@ -5,7 +5,6 @@ import random
 import socket
 import unittest
 import time
-import re
 
 from scapy.packet import Raw
 from scapy.layers.l2 import Ether
@@ -14,12 +13,17 @@ from scapy.layers.inet6 import IPv6
 from scapy.contrib.lacp import SlowProtocol, LACP
 
 from config import config
-from framework import tag_fixme_vpp_workers, tag_fixme_ubuntu2204, tag_fixme_debian11
-from framework import is_distro_ubuntu2204, is_distro_debian11
-from framework import VppTestCase, VppTestRunner
-from framework import tag_run_solo
+from framework import VppTestCase
+from asfframework import (
+    tag_fixme_vpp_workers,
+    tag_fixme_ubuntu2204,
+    tag_fixme_debian11,
+    tag_run_solo,
+    is_distro_ubuntu2204,
+    is_distro_debian11,
+    VppTestRunner,
+)
 from vpp_object import VppObject
-from vpp_pg_interface import CaptureTimeoutError
 from util import ppp
 from ipfix import IPFIX, Set, Template, Data, IPFIXDecoder
 from vpp_ip_route import VppIpRoute, VppRoutePath
@@ -1395,6 +1399,10 @@ class DisableFP(MethodHolder):
         self.sleep(1, "wait before verifying no packets sent")
         self.collector.assert_nothing_captured()
 
+        # enable FPP feature so the remove_vpp_config() doesn't fail
+        # due to missing feature on interface.
+        ipfix.enable_flowprobe_feature()
+
         ipfix.remove_vpp_config()
         self.logger.info("FFP_TEST_FINISH_0001")