tests: refactor asf framework code

[vpp.git] / test / asf / test_policer.py

diff --git a/test/asf/test_policer.py b/test/asf/test_policer.py
index c23ec00..9c01bf0 100644 (file)
--- a/test/asf/test_policer.py
+++ b/test/asf/test_policer.py
@@ -3,8 +3,8 @@
 
 import unittest
 
-from asfframework import VppTestCase, VppTestRunner
-from vpp_policer import VppPolicer, PolicerAction
+from asfframework import VppAsfTestCase, VppTestRunner
+from vpp_policer import VppPolicer
 
 # Default for the tests is 10s of "Green" packets at 8Mbps, ie. 10M bytes.
 # The policer helper CLI "sends" 500 byte packets, so default is 20000.
@@ -23,7 +23,7 @@ CBURST = 100000  # Committed burst in bytes
 EBURST = 200000  # Excess burst in bytes
 
 
-class TestPolicer(VppTestCase):
+class TestPolicer(VppAsfTestCase):
     """Policer Test Case"""
 
     def run_policer_test(