caliper/src/test/java/com/google/caliper/config/LoggingConfigLoaderTest.java

/*
 * Copyright (C) 2013 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.caliper.config;

import static com.google.common.base.Charsets.UTF_8;
import static java.util.logging.Level.INFO;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import com.google.caliper.model.Run;
import com.google.common.io.Files;

import org.joda.time.Instant;
import org.joda.time.format.ISODateTimeFormat;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import java.io.File;
import java.io.IOException;
import java.util.UUID;
import java.util.logging.FileHandler;
import java.util.logging.Handler;
import java.util.logging.LogManager;
import java.util.logging.LogRecord;
import java.util.logging.Logger;
import java.util.logging.SimpleFormatter;

/**
 * Tests {@link LoggingConfigLoader}.
 */

@RunWith(MockitoJUnitRunner.class)
public class LoggingConfigLoaderTest {
  @Rule public TemporaryFolder folder = new TemporaryFolder();

  @Mock LogManager logManager;
  @Mock Logger logger;
  @Captor ArgumentCaptor<Handler> handlerCaptor;

  private LoggingConfigLoader loader;
  private UUID runId = UUID.randomUUID();
  private Instant startTime = new Instant();
  private File caliperDirectory;

  @Before public void setUp() throws IOException {
    this.caliperDirectory = folder.newFolder();
    this.loader = new LoggingConfigLoader(caliperDirectory, logManager, new Run.Builder(runId)
        .label("fake run")
        .startTime(startTime)
        .build());
  }

  @Test public void testLoadDefaultLogConfiguration()
      throws SecurityException, IOException {
    when(logManager.getLogger("")).thenReturn(logger);
    loader.maybeLoadDefaultLogConfiguration(logManager);
    verify(logManager).reset();
    verify(logger).addHandler(handlerCaptor.capture());
    FileHandler fileHandler = (FileHandler) handlerCaptor.getValue();
    assertEquals(UTF_8.name(), fileHandler.getEncoding());
    assertTrue(fileHandler.getFormatter() instanceof SimpleFormatter);
    fileHandler.publish(new LogRecord(INFO, "some message"));
    File logFile = new File(new File(caliperDirectory, "log"),
        ISODateTimeFormat.basicDateTimeNoMillis().print(startTime) + "." + runId + ".log");
    assertTrue(logFile.isFile());
    assertTrue(Files.toString(logFile, UTF_8).contains("some message"));
  }
}