Fix checkstyle violations in yang-binding

[mdsal.git] / binding / yang-binding / src / test / java / org / opendaylight / yangtools / yang / binding / util / AugmentationFieldGetterTest.java

diff --git a/binding/yang-binding/src/test/java/org/opendaylight/yangtools/yang/binding/util/AugmentationFieldGetterTest.java b/binding/yang-binding/src/test/java/org/opendaylight/yangtools/yang/binding/util/AugmentationFieldGetterTest.java
index 43edb2c01d759e79fc3dd169795fd99772b92ee8..5e0e89454bdefcf590c68b860e03a0f412d47466 100644 (file)
--- a/binding/yang-binding/src/test/java/org/opendaylight/yangtools/yang/binding/util/AugmentationFieldGetterTest.java
+++ b/binding/yang-binding/src/test/java/org/opendaylight/yangtools/yang/binding/util/AugmentationFieldGetterTest.java
@@ -31,7 +31,7 @@ public class AugmentationFieldGetterTest {
         assertTrue(getGetter(TestAugmentationWrongTypeClass.class).getAugmentations(null).isEmpty());
 
         final AugmentationFieldGetter augmentationFieldGetter = getGetter(TestAugmentationClass.class);
-        final Augmentation augmentation = mock(Augmentation.class);
+        final Augmentation<?> augmentation = mock(Augmentation.class);
         final TestAugmentationClass testAugmentationClass = new TestAugmentationClass();
 
         testAugmentationClass.addAugmentation(augmentation, augmentation);
@@ -52,9 +52,9 @@ public class AugmentationFieldGetterTest {
     }
 
     private final class TestAugmentationClass {
-        private Map augmentation = new HashMap();
+        private final Map<Augmentation<?>, Augmentation<?>> augmentation = new HashMap<>();
 
-        void addAugmentation(Augmentation key, Augmentation value){
+        void addAugmentation(final Augmentation<?> key, final Augmentation<?> value) {
             augmentation.put(key, value);
         }
     }