src/test/java/com/googlesource/gerrit/plugins/its/base/workflow/LogEventTest.java

// Copyright (C) 2013 The Android Open Source Project
//
// 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.googlesource.gerrit.plugins.its.base.workflow;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import com.google.common.collect.ImmutableMap;
import com.google.gerrit.extensions.config.FactoryModule;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.googlesource.gerrit.plugins.its.base.its.ItsFacade;
import com.googlesource.gerrit.plugins.its.base.testutil.LoggingMockingTestCase;
import java.io.IOException;
import java.util.Map;
import org.apache.log4j.Level;

public class LogEventTest extends LoggingMockingTestCase {
  private Injector injector;
  private ItsFacade its;

  public void testNull() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn(null);

    LogEvent logEvent = createLogEvent();
    logEvent.execute(null, "4711", actionRequest, ImmutableMap.of());
  }

  public void testEmpty() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("");

    LogEvent logEvent = createLogEvent();
    logEvent.execute(null, "4711", actionRequest, ImmutableMap.of());
  }

  public void testLevelDefault() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("");

    Map<String, String> properties = ImmutableMap.of("KeyA", "ValueA");

    LogEvent logEvent = createLogEvent();
    logEvent.execute(its, "4711", actionRequest, properties);

    assertLogMessageContains("KeyA = ValueA", Level.INFO);
  }

  public void testLevelError() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("error");

    Map<String, String> properties = ImmutableMap.of("KeyA", "ValueA");

    LogEvent logEvent = createLogEvent();
    logEvent.execute(its, "4711", actionRequest, properties);

    assertLogMessageContains("KeyA = ValueA", Level.ERROR);
  }

  public void testLevelWarn() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("warn");

    Map<String, String> properties = ImmutableMap.of("KeyA", "ValueA");

    LogEvent logEvent = createLogEvent();
    logEvent.execute(its, "4711", actionRequest, properties);

    assertLogMessageContains("KeyA = ValueA", Level.WARN);
  }

  public void testLevelInfo() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("info");

    Map<String, String> properties = ImmutableMap.of("KeyA", "ValueA");

    LogEvent logEvent = createLogEvent();
    logEvent.execute(its, "4711", actionRequest, properties);

    assertLogMessageContains("KeyA = ValueA", Level.INFO);
  }

  public void testLevelDebug() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("debug");

    Map<String, String> properties = ImmutableMap.of("KeyA", "ValueA");

    LogEvent logEvent = createLogEvent();
    logEvent.execute(its, "4711", actionRequest, properties);

    assertLogMessageContains("KeyA = ValueA", Level.DEBUG);
  }

  public void testMultipleProperties() throws IOException {
    ActionRequest actionRequest = mock(ActionRequest.class);
    when(actionRequest.getParameter(1)).thenReturn("info");

    Map<String, String> properties =
        ImmutableMap.<String, String>builder()
            .put("KeyA", "ValueA")
            .put("KeyB", "ValueB")
            .put("KeyC", "ValueC")
            .build();

    LogEvent logEvent = createLogEvent();
    logEvent.execute(its, "4711", actionRequest, properties);

    assertLogMessageContains("KeyA = ValueA", Level.INFO);
    assertLogMessageContains("KeyB = ValueB", Level.INFO);
    assertLogMessageContains("KeyC = ValueC", Level.INFO);
  }

  private LogEvent createLogEvent() {
    return injector.getInstance(LogEvent.class);
  }

  @Override
  public void setUp() throws Exception {
    super.setUp();
    injector = Guice.createInjector(new TestModule());
  }

  private class TestModule extends FactoryModule {
    @Override
    protected void configure() {
      its = mock(ItsFacade.class);
    }
  }
}