ExternalJarClassInitializerTest.java

package com.epam.wilma.stubconfig.initializer.support;

/*==========================================================================
Copyright 2013-2017 EPAM Systems

This file is part of Wilma.

Wilma is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Wilma is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Wilma.  If not, see <http://www.gnu.org/licenses/>.
===========================================================================*/

import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.testng.AssertJUnit.assertNotNull;

import java.io.File;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Collection;

import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.internal.util.reflection.Whitebox;
import org.slf4j.Logger;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.epam.wilma.common.helper.FileFactory;
import com.epam.wilma.common.helper.FileUtils;
import com.epam.wilma.domain.stubconfig.sequence.SequenceHandler;
import com.epam.wilma.domain.stubconfig.sequencehandler.DummySequenceHandler;
import com.epam.wilma.domain.stubconfig.exception.DescriptorValidationFailedException;
import com.epam.wilma.stubconfig.initializer.condition.helper.ClassPathExtender;
import com.epam.wilma.stubconfig.initializer.support.helper.BeanRegistryService;
import com.epam.wilma.stubconfig.initializer.support.helper.ClassInstantiator;
import com.epam.wilma.stubconfig.initializer.support.helper.PackageBasedClassFinder;

/**
 * Unit test for {@link ExternalJarClassInitializer}.
 * @author Adam_Csaba_Kiraly
 *
 */
public class ExternalJarClassInitializerTest {

    private static final String PACKAGE_NAME = "com.test.something";
    private static final String JAR_FOLDER_PATH = "config/jar";
    private static final Class<SequenceHandler> INTERFACE_TO_CAST = SequenceHandler.class;
    private static final String BEAN_NAME = PACKAGE_NAME + INTERFACE_TO_CAST.getSimpleName();

    @Mock
    private ClassPathExtender classPathExtender;
    @Mock
    private PackageBasedClassFinder packageBasedClassFinder;
    @Mock
    private FileUtils fileUtils;
    @Mock
    private ClassInstantiator classInstantiator;
    @Mock
    private FileFactory fileFactory;
    @Mock
    private BeanRegistryService beanRegistryService;

    @InjectMocks
    private ExternalJarClassInitializer underTest;

    @Mock
    private File folder;
    @Mock
    private Logger logger;
    @Mock
    private DummySequenceHandler object;

    @BeforeMethod
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        Whitebox.setInternalState(underTest, "logger", logger);
        given(fileFactory.createFile(JAR_FOLDER_PATH)).willReturn(folder);
        doReturn(DummySequenceHandler.class).when(packageBasedClassFinder).findFirstOf(INTERFACE_TO_CAST, PACKAGE_NAME);
        given(beanRegistryService.getBean(BEAN_NAME, INTERFACE_TO_CAST)).willReturn(object);
    }

    @Test
    public void testLoadExternalClassShouldGetClassAsBeanFirst() throws InstantiationException, IllegalAccessException, InvocationTargetException {
        //GIVEN in setup
        //WHEN
        Object result = underTest.loadExternalClass(PACKAGE_NAME, JAR_FOLDER_PATH, INTERFACE_TO_CAST);
        //THEN
        verify(classInstantiator, never()).createClassInstanceOf(INTERFACE_TO_CAST);
        verify(logger).info(Mockito.anyString(), Mockito.anyObject(), Mockito.anyObject(), Mockito.anyObject());
        assertNotNull(result);
    }

    @Test
    public void testLoadExternalClassShouldAddJarFilesInFolderPathToClassPathWhenBeanWasNotFound() {
        //GIVEN
        given(beanRegistryService.getBean(BEAN_NAME, INTERFACE_TO_CAST)).willThrow(new NoSuchBeanDefinitionException("error"));
        Collection<File> jarFiles = new ArrayList<>();
        File one = new File("one");
        File two = new File("two");
        jarFiles.add(one);
        jarFiles.add(two);
        given(fileUtils.listFiles(folder, "jar")).willReturn(jarFiles);
        //WHEN
        underTest.loadExternalClass(PACKAGE_NAME, JAR_FOLDER_PATH, INTERFACE_TO_CAST);
        //THEN
        verify(classPathExtender).addFile(one.getPath());
        verify(classPathExtender).addFile(two.getPath());
    }

    @Test
    public void testLoadExternalClassShouldInstantiateClassWhenBeanWasNotFound() throws InstantiationException, IllegalAccessException,
        InvocationTargetException {
        //GIVEN
        given(beanRegistryService.getBean(BEAN_NAME, INTERFACE_TO_CAST)).willThrow(new NoSuchBeanDefinitionException("error"));
        Collection<File> jarFiles = new ArrayList<>();
        given(fileUtils.listFiles(folder, "jar")).willReturn(jarFiles);
        DummySequenceHandler externalClass = new DummySequenceHandler();
        given(classInstantiator.createClassInstanceOf(DummySequenceHandler.class)).willReturn(externalClass);
        //WHEN
        Object result = underTest.loadExternalClass(PACKAGE_NAME, JAR_FOLDER_PATH, INTERFACE_TO_CAST);
        //THEN
        verify(logger).info(Mockito.anyString(), Mockito.anyObject(), Mockito.anyObject(), Mockito.anyObject());
        assertNotNull(result);
    }

    @Test(expectedExceptions = DescriptorValidationFailedException.class)
    public void testLoadExternalClassShouldThrowDescriptorValidationFailedExceptionWhenNeitherBeanNorClassWasFound() throws ClassNotFoundException {
        //GIVEN
        given(beanRegistryService.getBean(BEAN_NAME, INTERFACE_TO_CAST)).willThrow(new NoSuchBeanDefinitionException("error"));
        Collection<File> jarFiles = new ArrayList<>();
        given(fileUtils.listFiles(folder, "jar")).willReturn(jarFiles);
        given(packageBasedClassFinder.findFirstOf(INTERFACE_TO_CAST, PACKAGE_NAME)).willThrow(new ClassNotFoundException());
        //WHEN
        underTest.loadExternalClass(PACKAGE_NAME, JAR_FOLDER_PATH, INTERFACE_TO_CAST);
        //THEN exception is thrown
    }

    @Test(expectedExceptions = DescriptorValidationFailedException.class)
    public void testLoadExternalClassShouldThrowDescriptorValidationFailedExceptionWhenBeanWasNotFoundButClassIsAbstract()
        throws InstantiationException, IllegalAccessException, InvocationTargetException {
        //GIVEN
        given(beanRegistryService.getBean(BEAN_NAME, INTERFACE_TO_CAST)).willThrow(new NoSuchBeanDefinitionException("error"));
        Collection<File> jarFiles = new ArrayList<>();
        given(fileUtils.listFiles(folder, "jar")).willReturn(jarFiles);
        given(classInstantiator.createClassInstanceOf(DummySequenceHandler.class)).willThrow(new InstantiationException());
        //WHEN
        underTest.loadExternalClass(PACKAGE_NAME, JAR_FOLDER_PATH, INTERFACE_TO_CAST);
        //THEN exception is thrown
    }
}