tests/testutils/artifactshare.py: use CASCache directlyabderrahim/artifact-cas

1 files changed, 2 insertions, 6 deletions

diff --git a/tests/testutils/artifactshare.py b/tests/testutils/artifactshare.py
index d929a7b91..03186862a 100644
--- a/tests/testutils/artifactshare.py
+++ b/tests/testutils/artifactshare.py
@@ -12,7 +12,7 @@ from multiprocessing import Process, Queue
 
 from buildstream import _yaml
 from buildstream._artifactcache.casserver import create_server
-from buildstream._context import Context
+from buildstream._artifactcache.cascache import CASCache
 from buildstream._exceptions import CASError
 
 
@@ -48,11 +48,7 @@ class ArtifactShare():
 
         os.makedirs(self.repodir)
 
-        context = Context()
-        context.artifactdir = self.repodir
-        context.set_message_handler(self._message_handler)
-
-        self.cas = context.artifactcache.cas
+        self.cas = CASCache(self.repodir)
 
         self.total_space = total_space
         self.free_space = free_space