X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Ftest_cli.py;h=5005bf4c43a13d95db2516f556338f8894480f9c;hb=91144fbf54764feb005a33419d0129febd2365d4;hp=a1ffbac81d81bace6c86ab97fab5df4cc667670e;hpb=e2ccdf0316243a1486109743e3de532cae5f3254;p=vpp.git

diff --git a/test/test_cli.py b/test/test_cli.py
index a1ffbac81d8..5005bf4c43a 100644
--- a/test/test_cli.py
+++ b/test/test_cli.py
@@ -5,7 +5,7 @@ import datetime
 import time
 import unittest
 
-from vpp_papi import vpp_transport_shmem
+from vpp_papi import VPPIOError
 
 from framework import VppTestCase, VppTestRunner
 
@@ -39,9 +39,8 @@ class TestCLI(VppTestCase):
         self.assertEqual(rv.retval, 0)
 
     def test_long_cli_delay(self):
-        """ Test that VppApiClient raises VppTransportShmemIOError if timeout."""  # noqa
-        with self.assertRaises(
-                vpp_transport_shmem.VppTransportShmemIOError) as ctx:
+        """ Test that VppApiClient raises VppIOError if timeout."""  # noqa
+        with self.assertRaises(VPPIOError) as ctx:
             rv = self.vapi.papi.cli_inband(cmd='wait 10')
 
     def test_long_cli_delay_override(self):