Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit a2c2087f authored by Sudheer Shanka's avatar Sudheer Shanka
Browse files

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

Fixes: 157204810
Test: atest --test-mapping apex/blobstore
Change-Id: Ic587da72c4018b15902b2b65307b112edc1c119e
parent c8f4a8a5

tests/BlobStoreTestUtils/src/com/android/utils/blob/DummyBlobData.java

+8 −1
Original line number Diff line number Diff line
@@ -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);

        }

    }

tests/BlobStoreTestUtils/src/com/android/utils/blob/Utils.java

+3 −3
Original line number Diff line number Diff line
@@ -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);

        }

    }