25 lines
708 B
Python
25 lines
708 B
Python
|
import json
|
||
|
|
|
||
|
|
from image.docker.interfaces import ContentRetriever
|
||
|
|
|
||
|
|
class ContentRetrieverForTesting(ContentRetriever):
|
||
|
|
def __init__(self, digests=None):
|
||
|
|
self.digests = digests or {}
|
||
|
|
|
||
|
|
def add_digest(self, digest, content):
|
||
|
|
self.digests[digest] = content
|
||
|
|
|
||
|
|
def get_manifest_bytes_with_digest(self, digest):
|
||
|
|
return self.digests.get(digest)
|
||
|
|
|
||
|
|
def get_blob_bytes_with_digest(self, digest):
|
||
|
|
return self.digests.get(digest)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def for_config(cls, config_obj, digest, size):
|
||
|
|
config_str = json.dumps(config_obj)
|
||
|
|
padded_string = config_str + ' ' * (size - len(config_str))
|
||
|
|
digests = {}
|
||
|
|
digests[digest] = padded_string
|
||
|
|
return ContentRetrieverForTesting(digests)
|