tests/test_cache_invocation.py: Adapt to new progress API.

1 files changed, 5 insertions, 4 deletions

diff --git a/tests/test_cache_invocation.py b/tests/test_cache_invocation.py
index ff86b102..4245eb50 100644
--- a/tests/test_cache_invocation.py
+++ b/tests/test_cache_invocation.py
@@ -2,6 +2,7 @@
 import unittest
 
 import apt_pkg
+import apt.progress.base
 
 
 class TestCache(unittest.TestCase):
@@ -13,21 +14,21 @@ class TestCache(unittest.TestCase):
 
     def test_wrong_invocation(self):
         """cache_invocation: Test wrong invocation."""
-        apt_cache = apt_pkg.Cache(apt_pkg.OpProgress())
+        apt_cache = apt_pkg.Cache(apt.progress.base.OpProgress())
         if apt_pkg._COMPAT_0_7:
             self.assertRaises(ValueError, apt_pkg.Cache, apt_cache)
             self.assertRaises(ValueError, apt_pkg.Cache,
-                              apt_pkg.AcquireProgress())
+                              apt.progress.base.AcquireProgress())
             self.assertRaises(ValueError, apt_pkg.Cache, 0)
         else:
             self.assertRaises(TypeError, apt_pkg.Cache, apt_cache)
             self.assertRaises(TypeError, apt_pkg.Cache,
-                              apt_pkg.AcquireProgress())
+                              apt.progress.base.AcquireProgress())
             self.assertRaises(TypeError, apt_pkg.Cache, 0)
 
     def test_proper_invocation(self):
         """cache_invocation: Test correct invocation."""
-        apt_cache = apt_pkg.Cache(apt_pkg.OpProgress())
+        apt_cache = apt_pkg.Cache(apt.progress.base.OpProgress())
         apt_depcache = apt_pkg.DepCache(apt_cache)
 
 if __name__ == "__main__":