diff options
2 files changed, 90 insertions, 41 deletions
diff --git a/infra/translate-utils/src/main/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizer.java b/infra/translate-utils/src/main/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizer.java index b6430a355..e058964a0 100644 --- a/infra/translate-utils/src/main/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizer.java +++ b/infra/translate-utils/src/main/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizer.java @@ -16,24 +16,26 @@ package io.fd.honeycomb.translate.util.read; +import static com.google.common.base.Preconditions.checkArgument; +import static com.google.common.base.Preconditions.checkNotNull; +import static com.google.common.base.Preconditions.checkState; + import com.google.common.base.Optional; import io.fd.honeycomb.translate.read.ReadContext; import io.fd.honeycomb.translate.read.ReadFailedException; import io.fd.honeycomb.translate.spi.read.ListReaderCustomizer; import io.fd.honeycomb.translate.util.ReflectionUtils; +import java.lang.reflect.InvocationTargetException; +import java.lang.reflect.Method; +import java.util.Collections; +import java.util.List; +import javax.annotation.Nonnull; import org.opendaylight.yangtools.concepts.Builder; import org.opendaylight.yangtools.yang.binding.DataObject; import org.opendaylight.yangtools.yang.binding.Identifiable; import org.opendaylight.yangtools.yang.binding.Identifier; import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; - -import javax.annotation.Nonnull; -import java.lang.reflect.InvocationTargetException; -import java.lang.reflect.Method; -import java.util.Collections; -import java.util.List; - -import static com.google.common.base.Preconditions.*; +import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier; /** * Might be slow. @@ -43,12 +45,29 @@ public class ReflexiveListReaderCustomizer<C extends DataObject & Identifiable<K implements ListReaderCustomizer<C, K, B> { private final List<K> staticKeys; + private final Class<? extends Identifier> keyType; public ReflexiveListReaderCustomizer(@Nonnull final Class<C> typeClass, @Nonnull final Class<B> builderClass, @Nonnull final List<K> staticKeys) { super(typeClass, builderClass); this.staticKeys = checkNotNull(staticKeys, "Static keys cannot be null"); checkState(!this.staticKeys.isEmpty(), "No static keys provided"); + keyType = staticKeys.get(0).getClass(); + } + + @Override + public void readCurrentAttributes(final InstanceIdentifier<C> id, final B builder, final ReadContext context) + throws ReadFailedException { + final Optional<Method> method = + ReflectionUtils.findMethodReflex(builder.getClass(), "setKey", + Collections.singletonList(keyType), builder.getClass()); + checkArgument(method.isPresent(), "Unable to setKey to %s", builder); + + try { + method.get().invoke(builder, ((KeyedInstanceIdentifier)id).getKey()); + } catch (IllegalAccessException | InvocationTargetException e) { + throw new IllegalArgumentException("Unable to setKey to " + builder, e); + } } @Override diff --git a/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizerTest.java b/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizerTest.java index fe12ff699..2cc934546 100644 --- a/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizerTest.java +++ b/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/read/ReflexiveListReaderCustomizerTest.java @@ -15,23 +15,71 @@ */ package io.fd.honeycomb.translate.util.read; +import static org.hamcrest.MatcherAssert.assertThat; +import static org.hamcrest.Matchers.contains; +import static org.hamcrest.Matchers.hasSize; +import static org.junit.Assert.assertEquals; + import io.fd.honeycomb.translate.read.ReadContext; import io.fd.honeycomb.translate.read.ReadFailedException; +import java.util.Arrays; +import java.util.Collections; +import java.util.List; import org.junit.Before; import org.junit.Test; import org.mockito.Mock; import org.mockito.MockitoAnnotations; import org.opendaylight.yangtools.concepts.Builder; -import org.opendaylight.yangtools.yang.binding.*; +import org.opendaylight.yangtools.yang.binding.DataContainer; +import org.opendaylight.yangtools.yang.binding.DataObject; +import org.opendaylight.yangtools.yang.binding.Identifiable; +import org.opendaylight.yangtools.yang.binding.Identifier; +import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; -import java.util.Arrays; -import java.util.List; +public class ReflexiveListReaderCustomizerTest { -import static org.hamcrest.MatcherAssert.assertThat; -import static org.hamcrest.Matchers.contains; -import static org.hamcrest.Matchers.hasSize; + @Mock + private ReadContext readContext; -public class ReflexiveListReaderCustomizerTest { + @Before + public void init() { + MockitoAnnotations.initMocks(this); + } + + @Test + @SuppressWarnings("unchecked") + public void testReadCurrentAttributes() throws ReadFailedException { + final TestingListObject.TestingListKey keyOne = new TestingListObject.TestingListKey("1"); + final TestingListObject.TestingListKey keyTwo = new TestingListObject.TestingListKey("2"); + final List<TestingListObject.TestingListKey> staticKeys = Arrays.asList(keyOne, keyTwo); + + final ReflexiveListReaderCustomizer<TestingListObject, TestingListObject.TestingListKey, TestingListObjectBuilder> customizer + = new ReflexiveListReaderCustomizer<>(TestingListObject.class, TestingListObjectBuilder.class, staticKeys); + + final TestingListObjectBuilder builder = new TestingListObjectBuilder(); + final InstanceIdentifier<TestingListObject> id = + (InstanceIdentifier<TestingListObject>) InstanceIdentifier.create( + Collections.singletonList(new InstanceIdentifier.IdentifiableItem<>(TestingListObject.class, keyOne))); + customizer.readCurrentAttributes(id, builder, readContext); + + assertEquals(keyOne, builder.getKey()); + } + + @Test + public void testGetAllIds() throws ReadFailedException { + final TestingListObject.TestingListKey keyOne = new TestingListObject.TestingListKey("1"); + final TestingListObject.TestingListKey keyTwo = new TestingListObject.TestingListKey("2"); + final List<TestingListObject.TestingListKey> staticKeys = Arrays.asList(keyOne, keyTwo); + + final ReflexiveListReaderCustomizer<TestingListObject, TestingListObject.TestingListKey, TestingListObjectBuilder> customizer + = new ReflexiveListReaderCustomizer<>(TestingListObject.class, TestingListObjectBuilder.class, staticKeys); + + final List<TestingListObject.TestingListKey> allIds = + customizer.getAllIds(InstanceIdentifier.create(TestingListObject.class), readContext); + + assertThat(allIds, hasSize(2)); + assertThat(allIds, contains(keyOne, keyTwo)); + } static class TestingListObject implements DataObject, Identifiable<TestingListObject.TestingListKey> { @@ -78,38 +126,20 @@ public class ReflexiveListReaderCustomizerTest { static class TestingListObjectBuilder implements Builder<TestingListObject> { - private final TestingListObject.TestingListKey key; - - TestingListObjectBuilder(final TestingListObject.TestingListKey key) { - this.key = key; - } + private TestingListObject.TestingListKey key; @Override public TestingListObject build() { return new TestingListObject(key); } - } - - @Mock - private ReadContext readContext; - @Before - public void init() { - MockitoAnnotations.initMocks(this); - } - - @Test - public void testStaticKeys() throws ReadFailedException { - final TestingListObject.TestingListKey keyOne = new TestingListObject.TestingListKey("1"); - final TestingListObject.TestingListKey keyTwo = new TestingListObject.TestingListKey("2"); - final List<TestingListObject.TestingListKey> staticKeys = Arrays.asList(keyOne, keyTwo); - - final ReflexiveListReaderCustomizer<TestingListObject, TestingListObject.TestingListKey, TestingListObjectBuilder> customizer - = new ReflexiveListReaderCustomizer<>(TestingListObject.class, TestingListObjectBuilder.class, staticKeys); - - final List<TestingListObject.TestingListKey> allIds = customizer.getAllIds(InstanceIdentifier.create(TestingListObject.class), readContext); + public TestingListObjectBuilder setKey(final TestingListObject.TestingListKey key) { + this.key = key; + return this; + } - assertThat(allIds, hasSize(2)); - assertThat(allIds, contains(keyOne, keyTwo)); + public TestingListObject.TestingListKey getKey() { + return key; + } } } |