FieldRegistryDefaultTest.java

package net.bytebuddy.dynamic.scaffold;

import net.bytebuddy.description.field.FieldDescription;
import net.bytebuddy.dynamic.Transformer;
import net.bytebuddy.implementation.attribute.AnnotationValueFilter;
import net.bytebuddy.implementation.attribute.FieldAttributeAppender;
import net.bytebuddy.matcher.ElementMatcher;
import net.bytebuddy.matcher.LatentMatcher;
import net.bytebuddy.test.utility.MockitoRule;
import net.bytebuddy.test.utility.ObjectPropertyAssertion;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TestRule;
import org.mockito.Mock;
import org.objectweb.asm.FieldVisitor;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class FieldRegistryDefaultTest {

    @Rule
    public TestRule mockitoRule = new MockitoRule(this);

    @Mock
    private InstrumentedType instrumentedType;

    @Mock
    private FieldAttributeAppender.Factory distinctFactory;

    @Mock
    private FieldAttributeAppender distinct;

    @Mock
    private FieldDescription knownField, unknownField, instrumentedField;

    @Mock
    private LatentMatcher<FieldDescription> latentMatcher;

    @Mock
    private ElementMatcher<FieldDescription> matcher;

    @Mock
    private Object defaultValue, otherDefaultValue;

    @Mock
    private Transformer<FieldDescription> transformer;

    @Before
    @SuppressWarnings("unchecked")
    public void setUp() throws Exception {
        when(distinctFactory.make(instrumentedType)).thenReturn(distinct);
        when(latentMatcher.resolve(instrumentedType)).thenReturn((ElementMatcher) matcher);
        when(matcher.matches(knownField)).thenReturn(true);
        when(transformer.transform(instrumentedType, knownField)).thenReturn(instrumentedField);
    }

    @Test(expected = IllegalStateException.class)
    public void testImplicitFieldCannotResolveDefaultValue() throws Exception {
        new FieldRegistry.Default()
                .compile(instrumentedType)
                .target(unknownField)
                .resolveDefault(defaultValue);
    }

    @Test(expected = IllegalStateException.class)
    public void testImplicitFieldCannotReceiveAppender() throws Exception {
        new FieldRegistry.Default()
                .compile(instrumentedType)
                .target(unknownField)
                .getFieldAppender();
    }

    @Test(expected = IllegalStateException.class)
    public void testImplicitFieldCannotRApplyPartially() throws Exception {
        new FieldRegistry.Default()
                .compile(instrumentedType)
                .target(unknownField)
                .apply(mock(FieldVisitor.class), mock(AnnotationValueFilter.Factory.class));
    }

    @Test
    public void testKnownFieldRegistered() throws Exception {
        TypeWriter.FieldPool fieldPool = new FieldRegistry.Default()
                .prepend(latentMatcher, distinctFactory, defaultValue, transformer)
                .compile(instrumentedType);
        assertThat(fieldPool.target(knownField).isImplicit(), is(false));
        assertThat(fieldPool.target(knownField).getField(), is(instrumentedField));
        assertThat(fieldPool.target(knownField).getFieldAppender(), is(distinct));
        assertThat(fieldPool.target(knownField).resolveDefault(otherDefaultValue), is(defaultValue));
        assertThat(fieldPool.target(unknownField).isImplicit(), is(true));
        assertThat(fieldPool.target(unknownField).getField(), is(unknownField));

    }

    @Test
    public void testObjectProperties() throws Exception {
        ObjectPropertyAssertion.of(FieldRegistry.Default.class).apply();
        ObjectPropertyAssertion.of(FieldRegistry.Default.Entry.class).apply();
        ObjectPropertyAssertion.of(FieldRegistry.Default.Compiled.class).apply();
        ObjectPropertyAssertion.of(FieldRegistry.Default.Compiled.Entry.class).apply();
    }
}