AgentBuilderFallbackStrategyByThrowableTypeTest.java

package net.bytebuddy.agent.builder;

import net.bytebuddy.test.utility.ObjectPropertyAssertion;
import org.junit.Test;

import java.util.Arrays;
import java.util.Iterator;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

public class AgentBuilderFallbackStrategyByThrowableTypeTest {

    @Test
    @SuppressWarnings("unchecked") // In absence of @SafeVarargs for Java 6
    public void testIsFallback() throws Exception {
        assertThat(new AgentBuilder.FallbackStrategy.ByThrowableType(Exception.class).isFallback(Object.class, new Exception()), is(true));
    }

    @Test
    @SuppressWarnings("unchecked") // In absence of @SafeVarargs for Java 6
    public void testIsFallbackInherited() throws Exception {
        assertThat(new AgentBuilder.FallbackStrategy.ByThrowableType(Exception.class).isFallback(Object.class, new RuntimeException()), is(true));
    }

    @Test
    @SuppressWarnings("unchecked") // In absence of @SafeVarargs for Java 6
    public void testIsNoFallback() throws Exception {
        assertThat(new AgentBuilder.FallbackStrategy.ByThrowableType(RuntimeException.class).isFallback(Object.class, new Exception()), is(false));
    }

    @Test
    public void testObjectProperties() throws Exception {
        final Iterator<Class<?>> iterator = Arrays.<Class<?>>asList(Object.class, String.class).iterator();
        ObjectPropertyAssertion.of(AgentBuilder.FallbackStrategy.ByThrowableType.class).create(new ObjectPropertyAssertion.Creator<Class<?>>() {
            @Override
            public Class<?> create() {
                return iterator.next();
            }
        }).apply();
    }
}