mockit.internal.injection.full.TestDataSource Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation
Show all versions of jmockit Show documentation
JMockit is a Java toolkit for automated developer testing.
It contains mocking/faking APIs and a code coverage tool, supporting both JUnit and TestNG.
The mocking APIs allow all kinds of Java code, without testability restrictions, to be tested
in isolation from selected dependencies.
/*
* Copyright (c) 2006 JMockit developers
* This file is subject to the terms of the MIT license (see LICENSE.txt).
*/
package mockit.internal.injection.full;
import java.beans.*;
import java.lang.annotation.*;
import java.lang.reflect.*;
import javax.annotation.*;
import javax.annotation.sql.*;
import javax.sql.*;
import mockit.internal.injection.*;
final class TestDataSource
{
@Nullable private final String dsName;
private Class dsClass;
private CommonDataSource ds;
TestDataSource(@Nonnull InjectionPoint injectionPoint) { dsName = injectionPoint.name; }
@Nullable
CommonDataSource createIfDataSourceDefinitionAvailable(@Nonnull TestedClass testedClass) {
TestedClass testedClassWithDataSource = testedClass.parent;
if (testedClassWithDataSource == null || dsName == null) {
return null;
}
Class testClass = testedClassWithDataSource.testClass;
if (testClass != null) {
createFromTestedClassOrASuperclass(testClass);
}
if (ds != null) {
return ds;
}
TestedClass testedClassToBeSearched = testedClassWithDataSource;
do {
createFromTestedClassOrASuperclass(testedClassToBeSearched.targetClass);
if (ds != null) {
return ds;
}
testedClassToBeSearched = testedClassToBeSearched.parent;
}
while (testedClassToBeSearched != null);
throw new IllegalStateException(
"Missing @DataSourceDefinition of name \"" + dsName + "\" on " + testedClassWithDataSource.nameOfTestedClass +
" or on a super/parent class");
}
private void createFromTestedClassOrASuperclass(@Nonnull Class targetClass) {
do {
createDataSource(targetClass);
if (ds != null) {
return;
}
targetClass = targetClass.getSuperclass();
}
while (targetClass != null && targetClass != Object.class);
}
private void createDataSource(@Nonnull Class targetClass) {
for (Annotation annotation : targetClass.getDeclaredAnnotations()) {
String annotationName = annotation.annotationType().getName();
if ("javax.annotation.sql.DataSourceDefinitions".equals(annotationName)) {
createDataSource((DataSourceDefinitions) annotation);
}
else if ("javax.annotation.sql.DataSourceDefinition".equals(annotationName)) {
createDataSource((DataSourceDefinition) annotation);
}
if (ds != null) {
return;
}
}
}
private void createDataSource(@Nonnull DataSourceDefinitions dsDefs) {
for (DataSourceDefinition dsDef : dsDefs.value()) {
createDataSource(dsDef);
if (ds != null) {
return;
}
}
}
private void createDataSource(@Nonnull DataSourceDefinition dsDef) {
String configuredDataSourceName = InjectionPoint.getNameFromJNDILookup(dsDef.name());
if (configuredDataSourceName.equals(dsName)) {
instantiateConfiguredDataSourceClass(dsDef);
setDataSourcePropertiesFromConfiguredValues(dsDef);
}
}
private void instantiateConfiguredDataSourceClass(@Nonnull DataSourceDefinition dsDef) {
String className = dsDef.className();
try {
//noinspection unchecked
dsClass = (Class) Class.forName(className);
//noinspection ClassNewInstance
ds = dsClass.newInstance();
}
catch (ClassNotFoundException | IllegalAccessException e) { throw new RuntimeException(e); }
catch (InstantiationException e) { throw new RuntimeException(e.getCause()); }
}
private void setDataSourcePropertiesFromConfiguredValues(@Nonnull DataSourceDefinition dsDef) {
try {
BeanInfo beanInfo = Introspector.getBeanInfo(dsClass, Object.class);
PropertyDescriptor[] properties = beanInfo.getPropertyDescriptors();
setProperty(properties, "url", dsDef.url());
setProperty(properties, "user", dsDef.user());
setProperty(properties, "password", dsDef.password());
}
catch (IntrospectionException | IllegalAccessException | InvocationTargetException e) { throw new RuntimeException(e); }
}
private void setProperty(
@Nonnull PropertyDescriptor[] properties, @Nonnull String name, @Nonnull String value
) throws InvocationTargetException, IllegalAccessException {
for (PropertyDescriptor property : properties) {
if (property.getName().equals(name)) {
Method writeMethod = property.getWriteMethod();
if (writeMethod != null) {
writeMethod.invoke(ds, value);
}
return;
}
}
}
}