src/content/public/android/javatests/src/org/chromium/content/browser/ImportantFileWriterAndroidTest.java

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 package org.chromium.content.browser;
   6
   7 import android.test.InstrumentationTestCase;
   8 import android.test.suitebuilder.annotation.SmallTest;
   9
  10 import org.chromium.base.ImportantFileWriterAndroid;
  11 import org.chromium.base.library_loader.LibraryLoader;
  12 import org.chromium.base.library_loader.ProcessInitException;
  13 import org.chromium.base.test.util.Feature;
  14 import org.chromium.content_shell_apk.ContentShellApplication;
  15
  16 import java.io.DataInputStream;
  17 import java.io.File;
  18 import java.io.FileInputStream;
  19 import java.io.IOException;
  20
  21
  22 /**
  23  * Tests for {@Link ImportantFileWriterAndroid}
  24  *
  25  * Note that this assumes that the underlying native atomic write functions
  26  * work, so is not attempting to test that writes are atomic. Instead it is just
  27  * testing that the Java code is calling the native code correctly.
  28  */
  29 public class ImportantFileWriterAndroidTest extends InstrumentationTestCase {
  30
  31     void loadJni() {
  32         getInstrumentation().runOnMainSync(new Runnable() {
  33             @Override
  34             public void run() {
  35                 ContentShellApplication.initializeApplicationParameters();
  36                 try {
  37                     LibraryLoader.ensureInitialized();
  38                 } catch (ProcessInitException e) {
  39                     throw new Error(e);
  40                 }
  41             }
  42         });
  43     }
  44
  45     private void checkFile(File testFile, byte[] data) {
  46         assertTrue(testFile.exists());
  47         try {
  48             byte[] fileData = new byte[(int) testFile.length()];
  49             DataInputStream dis =
  50                     new DataInputStream(new FileInputStream(testFile));
  51             dis.readFully(fileData);
  52             dis.close();
  53             assertEquals("Data length wrong", data.length, fileData.length);
  54             for (int i = 0; i < data.length; i++) {
  55                 assertEquals("Data byte wrong", data[i], fileData[i]);
  56             }
  57         } catch (IOException e) {
  58             fail("Failed to read file");
  59         }
  60
  61     }
  62
  63     @SmallTest
  64     @Feature({"Android-AppBase"})
  65     public void testAtomicWrite() {
  66         // Try writing a file that can't be created.
  67         byte[] data1 = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
  68         assertFalse("Writing bad file succeded",
  69                 ImportantFileWriterAndroid.writeFileAtomically(
  70                         "/junk/junk", data1));
  71         File dir = getInstrumentation().getTargetContext().getFilesDir();
  72         File testFile = new File(dir, "ImportantFileTest");
  73
  74         // Make sure the file doesn't already exist
  75         if (testFile.exists()) {
  76             assertTrue(testFile.delete());
  77         }
  78
  79         // Write a new file
  80         assertTrue("Writing new file failed",
  81                 ImportantFileWriterAndroid.writeFileAtomically(
  82                         testFile.getAbsolutePath(), data1));
  83         checkFile(testFile, data1);
  84         byte[] data2 = {10, 20, 30, 40, 50, 60, 70, 80};
  85
  86         // Overwrite an existing file
  87         assertTrue("Writing exiting file failed",
  88                 ImportantFileWriterAndroid.writeFileAtomically(
  89                         testFile.getAbsolutePath(), data2));
  90         checkFile(testFile, data2);
  91
  92         // Done, tidy up
  93         assertTrue(testFile.delete());
  94     }
  95
  96     @Override
  97     public void setUp() throws Exception {
  98         loadJni();
  99     }
 100 }