def testBlobsAreCorrectlyMigrated(self):
mem_bs = memory_stream_bs.MemoryStreamBlobStore()
db_bs = db_blob_store.DbBlobStore()
blob_contents_1 = b"A" * 1024
blob_hash_1 = mem_bs.WriteBlobWithUnknownHash(blob_contents_1)
blob_contents_2 = b"B" * 1024
blob_hash_2 = mem_bs.WriteBlobWithUnknownHash(blob_contents_2)
data_migration.BlobsMigrator().Execute(2)
contents = db_bs.ReadBlob(blob_hash_1)
self.assertEqual(contents, blob_contents_1)
contents = db_bs.ReadBlob(blob_hash_2)
self.assertEqual(contents, blob_contents_2)
def CreateBlobStore(self):
db, db_cleanup_fn = self.CreateDatabase()
return (db_blob_store.DbBlobStore(db), db_cleanup_fn)
def CreateBlobStore(self):
return (db_blob_store.DbBlobStore(), lambda: None)
def __init__(self):
super(TestBlobStore, self).__init__()
self.new = db_blob_store.DbBlobStore()
self.legacy = memory_stream_bs.MemoryStreamBlobStore()
def __init__(self):
super().__init__()
self.new = db_blob_store.DbBlobStore()
def Wrapper(*args, **kwargs):
blobstore = db_blob_store.DbBlobStore()
with mock.patch.object(data_store, "BLOBS", blobstore):
func(*(args + (blobstore, )), **kwargs)
