diff options
Diffstat (limited to 'tests/testutils/artifactshare.py')
| -rw-r--r-- | tests/testutils/artifactshare.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/tests/testutils/artifactshare.py b/tests/testutils/artifactshare.py index 05e87a499..e3f709b0a 100644 --- a/tests/testutils/artifactshare.py +++ b/tests/testutils/artifactshare.py @@ -15,6 +15,7 @@ from buildstream._artifactcache.cascache import CASCache from buildstream._artifactcache.casserver import create_server from buildstream._context import Context from buildstream._exceptions import ArtifactError +from buildstream._protos.build.bazel.remote.execution.v2 import remote_execution_pb2 # ArtifactShare() @@ -87,6 +88,23 @@ class ArtifactShare(): # Sleep until termination by signal signal.pause() + # has_object(): + # + # Checks whether the object is present in the share + # + # Args: + # digest (str): The object's digest + # + # Returns: + # (bool): True if the object exists in the share, otherwise false. + def has_object(self, digest): + + assert isinstance(digest, remote_execution_pb2.Digest) + + object_path = self.cas.objpath(digest) + + return os.path.exists(object_path) + # has_artifact(): # # Checks whether the artifact is present in the share |