Fix uncompressed size for blob store and add test

This commit is contained in:
Joseph Schorr 2016-03-29 14:16:56 -04:00
parent 685dd1a925
commit 0dffdb87c9
2 changed files with 60 additions and 2 deletions

View file

@ -35,13 +35,17 @@ def store_blob_record_and_temp_link(namespace, repo_name, blob_digest, location_
try:
storage = ImageStorage.get(content_checksum=blob_digest)
storage.image_size = byte_count
storage.uncompressed_size = uncompressed_byte_count
if uncompressed_byte_count is not None:
storage.uncompressed_size = uncompressed_byte_count
storage.save()
ImageStoragePlacement.get(storage=storage, location=location_obj)
except ImageStorage.DoesNotExist:
storage = ImageStorage.create(content_checksum=blob_digest, uploading=False,
image_size=byte_count)
image_size=byte_count,
uncompressed_size=uncompressed_byte_count)
ImageStoragePlacement.create(storage=storage, location=location_obj)
except ImageStoragePlacement.DoesNotExist:
ImageStoragePlacement.create(storage=storage, location=location_obj)