LoggingStreamingContentTest.java

/*
 * Copyright (c) 2011 Google Inc.
 *
 * 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.api.client.util;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import com.google.api.client.testing.util.LogRecordingHandler;
import java.io.ByteArrayOutputStream;
import java.util.Arrays;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

/**
 * Tests {@link LoggingStreamingContent}.
 *
 * @author Yaniv Inbar
 */
@RunWith(JUnit4.class)
public class LoggingStreamingContentTest {

  static final Logger LOGGER = Logger.getLogger(LoggingStreamingContentTest.class.getName());

  private static final byte[] SAMPLE_UTF8 =
      new byte[] {49, 50, 51, -41, -103, -41, -96, -41, -103, -41, -111};
  private static final String SAMPLE = "123\u05D9\u05e0\u05D9\u05D1";

  /** Test method for {@link LoggingStreamingContent#writeTo(java.io.OutputStream)}. */
  @Test
  public void testWriteTo() throws Exception {
    LoggingStreamingContent logContent =
        new LoggingStreamingContent(
            new ByteArrayStreamingContent(SAMPLE_UTF8), LOGGER, Level.CONFIG, Integer.MAX_VALUE);
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    LOGGER.setLevel(Level.CONFIG);
    LogRecordingHandler recorder = new LogRecordingHandler();
    LOGGER.addHandler(recorder);
    logContent.writeTo(out);
    out.close();
    assertEquals(Arrays.asList("Total: 11 bytes", SAMPLE), recorder.messages());
  }

  @Test
  public void testContentLoggingLimit() throws Exception {
    LOGGER.setLevel(Level.CONFIG);

    // Set the content logging limit to be equal to the length of the content.
    LogRecordingHandler recorder = new LogRecordingHandler();
    LOGGER.addHandler(recorder);
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    LoggingStreamingContent logContent =
        new LoggingStreamingContent(
            new ByteArrayStreamingContent(SAMPLE_UTF8), LOGGER, Level.CONFIG, SAMPLE_UTF8.length);
    logContent.writeTo(out);
    assertEquals(Arrays.asList("Total: 11 bytes", SAMPLE), recorder.messages());

    // Set the content logging limit to be less than the length of the content.
    recorder = new LogRecordingHandler();
    LOGGER.addHandler(recorder);
    logContent =
        new LoggingStreamingContent(
            new ByteArrayStreamingContent(SAMPLE_UTF8),
            LOGGER,
            Level.CONFIG,
            SAMPLE_UTF8.length - 1);
    logContent.writeTo(new ByteArrayOutputStream());
    assertEquals(
        Arrays.asList("Total: 11 bytes (logging first 10 bytes)", "123\u05D9\u05e0\u05D9\ufffd"),
        recorder.messages());

    // Set the content logging limit to 0 to disable content logging.
    recorder = new LogRecordingHandler();
    LOGGER.addHandler(recorder);
    logContent =
        new LoggingStreamingContent(
            new ByteArrayStreamingContent(SAMPLE_UTF8), LOGGER, Level.CONFIG, 0);
    logContent.writeTo(new ByteArrayOutputStream());
    assertEquals(Arrays.asList("Total: 11 bytes"), recorder.messages());

    // writeTo should behave as expected even if content length is specified to be -1.
    recorder = new LogRecordingHandler();
    LOGGER.addHandler(recorder);
    logContent =
        new LoggingStreamingContent(
            new ByteArrayStreamingContent(SAMPLE_UTF8), LOGGER, Level.CONFIG, SAMPLE_UTF8.length);
    logContent.writeTo(new ByteArrayOutputStream());
    assertEquals(Arrays.asList("Total: 11 bytes", SAMPLE), recorder.messages());

    // Assert that an exception is thrown if content logging limit < 0.
    try {
      logContent =
          new LoggingStreamingContent(
              new ByteArrayStreamingContent(SAMPLE_UTF8), LOGGER, Level.CONFIG, -1);
      logContent.writeTo(new ByteArrayOutputStream());
      fail("Expected: " + IllegalArgumentException.class);
    } catch (IllegalArgumentException e) {
      // Expected.
    }
  }
}