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
}
}