Add blob upload to new registry data model

data/registry_model/test/test_pre_oci_model.py

@@ -525,3 +525,35 @@ def test_torrent_info(pre_oci_model):
   assert torrent_info is not None
   assert torrent_info.piece_length == 2
   assert torrent_info.pieces == 'foo'
+
+
+def test_blob_uploads(pre_oci_model):
+  repository_ref = pre_oci_model.lookup_repository('devtable', 'simple')
+
+  blob_upload = pre_oci_model.create_blob_upload(repository_ref, 'local_us', {'some': 'metadata'})
+  assert blob_upload
+  assert blob_upload.storage_metadata == {'some': 'metadata'}
+  assert blob_upload.location_name == 'local_us'
+
+  # Ensure we can find the blob upload.
+  assert pre_oci_model.lookup_blob_upload(repository_ref, blob_upload.upload_id) == blob_upload
+
+  # Update and ensure the changes are saved.
+  assert pre_oci_model.update_blob_upload(blob_upload, 1, 'the-pieces_hash',
+                                          blob_upload.piece_sha_state,
+                                          {'new': 'metadata'}, 2, 3,
+                                          blob_upload.sha_state)
+
+  updated = pre_oci_model.lookup_blob_upload(repository_ref, blob_upload.upload_id)
+  assert updated
+  assert updated.uncompressed_byte_count == 1
+  assert updated.piece_hashes == 'the-pieces_hash'
+  assert updated.storage_metadata == {'new': 'metadata'}
+  assert updated.byte_count == 2
+  assert updated.chunk_count == 3
+
+  # Delete the upload.
+  pre_oci_model.delete_blob_upload(blob_upload)
+
+  # Ensure it can no longer be found.
+  assert not pre_oci_model.lookup_blob_upload(repository_ref, blob_upload.upload_id)