Correct the blobstore Session.openWrite() API usage in tests.

Fixes: 157204810
Test: atest --test-mapping apex/blobstore
Change-Id: Ic587da72c4018b15902b2b65307b112edc1c119e
This commit is contained in:
Sudheer Shanka
2020-06-21 22:31:26 -07:00
parent c8f4a8a5e9
commit a2c2087f94
2 changed files with 11 additions and 4 deletions

View File

@@ -153,7 +153,14 @@ public class DummyBlobData {
public void writeToSession(BlobStoreManager.Session session,
long offsetBytes, long lengthBytes) throws Exception {
try (FileInputStream in = new FileInputStream(mFile)) {
Utils.writeToSession(session, in, offsetBytes, lengthBytes);
Utils.writeToSession(session, in, offsetBytes, lengthBytes, lengthBytes);
}
}
public void writeToSession(BlobStoreManager.Session session,
long offsetBytes, long lengthBytes, long allocateBytes) throws Exception {
try (FileInputStream in = new FileInputStream(mFile)) {
Utils.writeToSession(session, in, offsetBytes, lengthBytes, allocateBytes);
}
}

View File

@@ -59,15 +59,15 @@ public class Utils {
public static void writeToSession(BlobStoreManager.Session session, ParcelFileDescriptor input,
long lengthBytes) throws IOException {
try (FileInputStream in = new ParcelFileDescriptor.AutoCloseInputStream(input)) {
writeToSession(session, in, 0, lengthBytes);
writeToSession(session, in, 0, lengthBytes, lengthBytes);
}
}
public static void writeToSession(BlobStoreManager.Session session, FileInputStream in,
long offsetBytes, long lengthBytes) throws IOException {
long offsetBytes, long lengthBytes, long allocateBytes) throws IOException {
in.getChannel().position(offsetBytes);
try (FileOutputStream out = new ParcelFileDescriptor.AutoCloseOutputStream(
session.openWrite(offsetBytes, lengthBytes))) {
session.openWrite(offsetBytes, allocateBytes))) {
copy(in, out, lengthBytes);
}
}