Package org.springframework.aop.interceptor

Source Code of org.springframework.aop.interceptor.DebugInterceptorTests$StubDebugInterceptor

/*
* Copyright 2002-2013 the original author or 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 org.springframework.aop.interceptor;

import org.aopalliance.intercept.MethodInvocation;
import org.apache.commons.logging.Log;
import org.junit.Test;

import static org.junit.Assert.*;
import static org.mockito.BDDMockito.*;

/**
* Unit tests for the {@link DebugInterceptor} class.
*
* @author Rick Evans
* @author Chris Beams
*/
public final class DebugInterceptorTests {

  @Test
  public void testSunnyDayPathLogsCorrectly() throws Throwable {

    MethodInvocation methodInvocation = mock(MethodInvocation.class);

    Log log = mock(Log.class);
    given(log.isTraceEnabled()).willReturn(true);

    DebugInterceptor interceptor = new StubDebugInterceptor(log);
    interceptor.invoke(methodInvocation);
    checkCallCountTotal(interceptor);

    verify(log, times(2)).trace(anyString());
  }

  @Test
  public void testExceptionPathStillLogsCorrectly() throws Throwable {

    MethodInvocation methodInvocation = mock(MethodInvocation.class);

    IllegalArgumentException exception = new IllegalArgumentException();
    given(methodInvocation.proceed()).willThrow(exception);

    Log log = mock(Log.class);
    given(log.isTraceEnabled()).willReturn(true);

    DebugInterceptor interceptor = new StubDebugInterceptor(log);
    try {
      interceptor.invoke(methodInvocation);
      fail("Must have propagated the IllegalArgumentException.");
    } catch (IllegalArgumentException expected) {
    }
    checkCallCountTotal(interceptor);

    verify(log).trace(anyString());
    verify(log).trace(anyString(), eq(exception));
  }

  private void checkCallCountTotal(DebugInterceptor interceptor) {
    assertEquals("Intercepted call count not being incremented correctly", 1, interceptor.getCount());
  }


  @SuppressWarnings("serial")
  private static final class StubDebugInterceptor extends DebugInterceptor {

    private final Log log;


    public StubDebugInterceptor(Log log) {
      super(true);
      this.log = log;
    }

    @Override
    protected Log getLoggerForInvocation(MethodInvocation invocation) {
      return log;
    }

  }

}
TOP

Related Classes of org.springframework.aop.interceptor.DebugInterceptorTests$StubDebugInterceptor

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.