1 files changed, 63 insertions, 33 deletions

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