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)