/*
 * Copyright (C) 2008 The Guava Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.common.io;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Arrays;

/**
 * Unit tests for {@link FileBackedOutputStream}.
 *
 * @author Chris Nokleberg
 */
public class FileBackedOutputStreamTest extends IoTestCase {

  public void testThreshold() throws Exception {
    testThreshold(0, 100, true, false);
    testThreshold(10, 100, true, false);
    testThreshold(100, 100, true, false);
    testThreshold(1000, 100, true, false);
    testThreshold(0, 100, false, false);
    testThreshold(10, 100, false, false);
    testThreshold(100, 100, false, false);
    testThreshold(1000, 100, false, false);
  }

  // TODO(user): Add an @VisibleForTesting invokeFinalize method inside
  // FileBackedOutputStream and use that to test that the file was actually deleted
  // on finalize

  public void testThreshold_resetOnFinalize() throws Exception {
    testThreshold(0, 100, true, true);
    testThreshold(10, 100, true, true);
    testThreshold(100, 100, true, true);
    testThreshold(1000, 100, true, true);
    testThreshold(0, 100, false, true);
    testThreshold(10, 100, false, true);
    testThreshold(100, 100, false, true);
    testThreshold(1000, 100, false, true);
  }

  private void testThreshold(int fileThreshold, int dataSize, boolean singleByte,
      boolean resetOnFinalize) throws IOException {
    byte[] data = newPreFilledByteArray(dataSize);
    FileBackedOutputStream out = new FileBackedOutputStream(fileThreshold, resetOnFinalize);
    InputSupplier<InputStream> supplier = out.getSupplier();
    int chunk1 = Math.min(dataSize, fileThreshold);
    int chunk2 = dataSize - chunk1;

    // Write just enough to not trip the threshold
    if (chunk1 > 0) {
      write(out, data, 0, chunk1, singleByte);
      assertTrue(ByteStreams.equal(
          ByteStreams.newInputStreamSupplier(data, 0, chunk1), supplier));
    }
    File file = out.getFile();
    assertNull(file);

    // Write data to go over the threshold
    if (chunk2 > 0) {
      write(out, data, chunk1, chunk2, singleByte);
      file = out.getFile();
      assertEquals(dataSize, file.length());
      assertTrue(file.exists());
    }
    out.close();

    // Check that supplier returns the right data
    assertTrue(Arrays.equals(data, ByteStreams.toByteArray(supplier)));

    // Make sure that reset deleted the file
    out.reset();
    if (file != null) {
      assertFalse(file.exists());
    }
  }

  private static void write(
      OutputStream out, byte[] b, int off, int len, boolean singleByte)
      throws IOException {
    if (singleByte) {
      for (int i = off; i < off + len; i++) {
        out.write(b[i]);
      }
    } else {
      out.write(b, off, len);
    }
    out.flush(); // for coverage
  }

  // TODO(chrisn): only works if we ensure we have crossed file threshold

  public void testWriteErrorAfterClose() throws Exception {
    byte[] data = newPreFilledByteArray(100);
    FileBackedOutputStream out = new FileBackedOutputStream(50);
    InputSupplier<InputStream> supplier = out.getSupplier();

    out.write(data);
    assertTrue(Arrays.equals(data, ByteStreams.toByteArray(supplier)));

    out.close();
    try {
      out.write(42);
      fail("expected exception");
    } catch (IOException expected) {
      // expected
    }

    // Verify that write had no effect
    assertTrue(Arrays.equals(data, ByteStreams.toByteArray(supplier)));
    out.reset();
  }

  public void testReset() throws Exception {
    byte[] data = newPreFilledByteArray(100);
    FileBackedOutputStream out = new FileBackedOutputStream(Integer.MAX_VALUE);
    InputSupplier<InputStream> supplier = out.getSupplier();

    out.write(data);
    assertTrue(Arrays.equals(data, ByteStreams.toByteArray(supplier)));

    out.reset();
    assertTrue(Arrays.equals(new byte[0], ByteStreams.toByteArray(supplier)));

    out.write(data);
    assertTrue(Arrays.equals(data, ByteStreams.toByteArray(supplier)));

    out.close();
  }
}

@@ -626,16 +626,66 @@
 on.io;%0A%0A
+import com.google.common.testing.GcFinalization;%0A%0A
 import j
@@ -697,16 +697,16 @@
 o.File;%0A
-
 import j
@@ -1354,82 +1354,114 @@
 %0A%0A  
-// TODO(user): Add an @VisibleForTesting invokeFinalize method inside%0A  //
+public void testFinalizeDeletesFile() throws Exception %7B%0A    byte%5B%5D data = newPreFilledByteArray(100);%0A   
  Fil
@@ -1484,80 +1484,505 @@
 eam 
-and use that to test that the file was actually deleted%0A  // on finalize
+out = new FileBackedOutputStream(0, true);%0A%0A    write(out, data, 0, 100, true);%0A    final File file = out.getFile();%0A    assertEquals(100, file.length());%0A    assertTrue(file.exists());%0A    out.close();%0A%0A    // Make sure that finalize deletes the file%0A    out = null;%0A%0A    // times out and throws RuntimeException on failure%0A    GcFinalization.awaitDone(new GcFinalization.FinalizationPredicate() %7B%0A      @Override%0A      public boolean isDone() %7B%0A        return !file.exists();%0A      %7D%0A    %7D);%0A  %7D
 %0A%0A  

Test that finalize deletes the backing file. This required refactoring the model code to make finalize accessible from tests. ------------- Created by MOE: http://code.google.com/p/moe-java MOE_MIGRATED_REVID=25515639