AsecTests: Add a test to validate container size

Signed-off-by: San Mehat <san@google.com>

1 files changed, 11 insertions, 12 deletions

diff --git a/tests/AndroidTests/src/com/android/unit_tests/AsecTests.java b/tests/AndroidTests/src/com/android/unit_tests/AsecTests.java
index 5aaf13b..c7bacd4 100755
--- a/tests/AndroidTests/src/com/android/unit_tests/AsecTests.java
+++ b/tests/AndroidTests/src/com/android/unit_tests/AsecTests.java
@@ -346,21 +346,20 @@ public class AsecTests extends AndroidTestCase {
         }
     }
 
-    public void testIsContainerMountedAfterRename() {
+    public void testContainerSize() {
+        IMountService ms = getMs();
         try {
             Assert.assertEquals(StorageResultCode.OperationSucceeded,
-                    createContainer("testRenameContainer.1", 4, "none"));
-
-            Assert.assertEquals(StorageResultCode.OperationSucceeded,
-                    unmountContainer("testRenameContainer.1", false));
-
-            Assert.assertEquals(StorageResultCode.OperationSucceeded,
-                    renameContainer("testRenameContainer.1", "testRenameContainer.2"));
+                    createContainer("testContainerSize", 1, "none"));
+            String path = ms.getSecureContainerPath("com.android.unittests.AsecTests.testUnmountBusyContainer");
 
-            Assert.assertEquals(false, containerExists("testRenameContainer.1"));
-            Assert.assertEquals(true, containerExists("testRenameContainer.2"));
-            // Check if isContainerMounted returns valid value
-            Assert.assertEquals(true, isContainerMounted("testRenameContainer.2"));
+            byte[] buf = new byte[4096];
+            File f = new File(path, "reference");
+            FileOutputStream fos = new FileOutputStream(f);
+            for (int i = 0; i < (1024 * 1024); i+= buf.length) {
+                fos.write(buf);
+            }
+            fos.close();
         } catch (Exception e) {
             failStr(e);
         }