commit 5e74f9b3315384a59b628580af7b17693e5d2880
author Bastiaan Van Eeckhoudt <bastiaan@google.com> Thu Jul 24 14:07:51 2014 +0200
committer Erich Douglass <erich.douglass@gmail.com> Sun Aug 17 16:21:38 2014 -0700
tree b481e6558f5e00f1030c6ef28ea270b6bcf57161
parent ea68dee08262d527344629c1c87557f3fc8170cd

Add support for Context.MODE_APPEND in Context.openFileOutput.

Closes #1226.

src/main/java/org/robolectric/shadows/ShadowContext.java

diff --git a/src/main/java/org/robolectric/shadows/ShadowContext.java b/src/main/java/org/robolectric/shadows/ShadowContext.java
index 253b069..38ad569 100644
--- a/src/main/java/org/robolectric/shadows/ShadowContext.java
+++ b/src/main/java/org/robolectric/shadows/ShadowContext.java
@@ -115,6 +115,9 @@
 
   @Implementation
   public FileOutputStream openFileOutput(String path, int mode) throws FileNotFoundException {
+    if ((mode & Context.MODE_APPEND) == Context.MODE_APPEND) {
+      return new FileOutputStream(getFileStreamPath(path), true);
+    }
     return new FileOutputStream(getFileStreamPath(path));
   }
 

src/test/java/org/robolectric/shadows/ContextTest.java

diff --git a/src/test/java/org/robolectric/shadows/ContextTest.java b/src/test/java/org/robolectric/shadows/ContextTest.java
index 731946a..6254a4e 100644
--- a/src/test/java/org/robolectric/shadows/ContextTest.java
+++ b/src/test/java/org/robolectric/shadows/ContextTest.java
@@ -264,6 +264,71 @@
   }
 
   @Test
+  public void openFileOutput_shouldAppendData() throws Exception {
+    File file = new File("__test__");
+    String initialFileContents = "foo";
+    String appendedFileContents = "bar";
+    String finalFileContents = initialFileContents + appendedFileContents;
+    FileOutputStream fileOutputStream = null;
+    try {
+      fileOutputStream = context.openFileOutput("__test__", Context.MODE_APPEND);
+      fileOutputStream.write(initialFileContents.getBytes());
+    } finally {
+      if (fileOutputStream != null)
+        fileOutputStream.close();
+    }
+    try {
+      fileOutputStream = context.openFileOutput("__test__", Context.MODE_APPEND);
+      fileOutputStream.write(appendedFileContents.getBytes());
+    } finally {
+      if (fileOutputStream != null)
+        fileOutputStream.close();
+    }
+    FileInputStream fileInputStream = null;
+    try {
+      fileInputStream = new FileInputStream(new File(context.getFilesDir(), file.getName()));
+      byte[] readBuffer = new byte[finalFileContents.length()];
+      fileInputStream.read(readBuffer);
+      assertThat(new String(readBuffer)).isEqualTo(finalFileContents);
+    } finally {
+      if (fileInputStream != null)
+        fileInputStream.close();
+    }
+  }
+
+  @Test
+  public void openFileOutput_shouldOverwriteData() throws Exception {
+    File file = new File("__test__");
+    String initialFileContents = "foo";
+    String newFileContents = "bar";
+    FileOutputStream fileOutputStream = null;
+    try {
+      fileOutputStream = context.openFileOutput("__test__", 0);
+      fileOutputStream.write(initialFileContents.getBytes());
+    } finally {
+      if (fileOutputStream != null)
+        fileOutputStream.close();
+    }
+    try {
+      fileOutputStream = context.openFileOutput("__test__", 0);
+      fileOutputStream.write(newFileContents.getBytes());
+    } finally {
+      if (fileOutputStream != null)
+        fileOutputStream.close();
+    }
+    FileInputStream fileInputStream = null;
+    try {
+      fileInputStream = new FileInputStream(new File(context.getFilesDir(), file.getName()));
+      byte[] readBuffer = new byte[newFileContents.length()];
+      fileInputStream.read(readBuffer);
+      assertThat(new String(readBuffer)).isEqualTo(newFileContents);
+    } finally {
+      if (fileInputStream != null)
+        fileInputStream.close();
+    }
+  }
+
+  @Test
   public void deleteFile_shouldReturnTrue() throws IOException {
     File filesDir = context.getFilesDir();
     File file = new File(filesDir, "test.txt");