core/src/test/java/org/jclouds/logging/config/BindLoggersAnnotatedWithResourceTest.java

/**
 *
 * Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
 *
 * ====================================================================
 * 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.jclouds.logging.config;

import static com.google.inject.matcher.Matchers.any;
import static org.easymock.classextension.EasyMock.createMock;
import static org.testng.Assert.assertEquals;

import java.lang.reflect.Field;
import java.util.Set;

import javax.annotation.Resource;
import javax.inject.Inject;
import javax.inject.Named;

import org.jclouds.logging.Logger;
import org.jclouds.logging.config.BindLoggersAnnotatedWithResource.AssignLoggerToField;
import org.jclouds.logging.config.BindLoggersAnnotatedWithResource.LoggerFieldsAnnotatedWithResource;
import org.jclouds.logging.jdk.JDKLogger;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.google.inject.AbstractModule;
import com.google.inject.Guice;
import com.google.inject.Injector;

@Test
public class BindLoggersAnnotatedWithResourceTest {

   private BindLoggersAnnotatedWithResource blawr;

   public static class A {
      @Resource
      private Logger logger = Logger.NULL;
   }

   private static class B {
      @Resource
      private Logger logger = Logger.NULL;

      @Resource
      @Named("blogger")
      private Logger blogger = Logger.NULL;
   }

   @BeforeMethod
   void createBlawr() {
      blawr = new BindLoggersAnnotatedWithResource(new JDKLogger.JDKLoggerFactory());
   }

   @Test
   void testHear() {
      Injector i = Guice.createInjector(new AbstractModule() {

         @Override
         protected void configure() {
            bindListener(any(), blawr);
         }

      });
      assertEquals(i.getInstance(A.class).logger.getCategory(), getClass().getName() + "$A");
      assertEquals(i.getInstance(B.class).logger.getCategory(), getClass().getName() + "$B");
      assertEquals(i.getInstance(B.class).blogger.getCategory(), "blogger");
   }

   @Test
   public void testAssignLoggerToField() throws SecurityException, NoSuchFieldException,
            IllegalArgumentException, IllegalAccessException {
      Logger logger = createMock(Logger.class);
      A a = new A();
      Field field = A.class.getDeclaredField("logger");
      AssignLoggerToField<A> assigner = new AssignLoggerToField<A>(logger, field);
      assigner.afterInjection(a);
      assert field.get(a).equals(logger);
   }

   @Test
   public void testLoggerFieldsAnnotatedWithResource() throws SecurityException,
            NoSuchFieldException {
      LoggerFieldsAnnotatedWithResource predicate = new LoggerFieldsAnnotatedWithResource();
      assert predicate.apply(A.class.getDeclaredField("logger"));
   }

   public static class C {
      @SuppressWarnings("unused")
      @Inject
      private Logger logger = Logger.NULL;
   }

   @Test
   public void testLoggerFieldsAnnotatedWithInjectReturnsNull() throws SecurityException,
            NoSuchFieldException {
      LoggerFieldsAnnotatedWithResource predicate = new LoggerFieldsAnnotatedWithResource();
      assert !predicate.apply(C.class.getDeclaredField("logger"));
   }

   public static class D {
      @SuppressWarnings("unused")
      @Resource
      private Logger logger = Logger.NULL;

      @SuppressWarnings("unused")
      @Resource
      private Logger blogger;

   }

   @Test
   public void testGetLoggerFieldsAnnotatedWithResourceNoLogger() {
      Set<Field> fields = blawr.getLoggerFieldsAnnotatedWithResource(this.getClass());
      assertEquals(fields.size(), 0);
   }

   @Test
   public void testGetLoggerFieldsAnnotatedWithResourceOneLogger() {
      Set<Field> fields = blawr.getLoggerFieldsAnnotatedWithResource(A.class);
      assertEquals(fields.size(), 1);
   }

   @Test
   public void testGetLoggerFieldsAnnotatedWithResourceTwoLoggers() {
      Set<Field> fields = blawr.getLoggerFieldsAnnotatedWithResource(D.class);
      assertEquals(fields.size(), 2);
   }

}