binding/binding-test-model/src/test/java/org/opendaylight/mdsal/binding/test/model/TestBindingMap.java

   1 /*
   2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.binding.test.model;
   9
  10 import static org.junit.jupiter.api.Assertions.assertEquals;
  11 import static org.junit.jupiter.api.Assertions.assertNotNull;
  12
  13 import java.util.Iterator;
  14 import java.util.List;
  15 import java.util.Map;
  16 import org.junit.jupiter.api.Test;
  17 import org.opendaylight.yang.gen.v1.lal.norev.Foo;
  18 import org.opendaylight.yang.gen.v1.lal.norev.FooBuilder;
  19 import org.opendaylight.yang.gen.v1.lal.norev.foo.Bar;
  20 import org.opendaylight.yang.gen.v1.lal.norev.foo.BarBuilder;
  21 import org.opendaylight.yang.gen.v1.lal.norev.foo.BarKey;
  22 import org.opendaylight.yangtools.binding.util.BindingMap;
  23
  24 public class TestBindingMap {
  25     private static final BarKey BAR_KEY_ONE = new BarKey(1);
  26     private static final BarKey BAR_KEY_TWO = new BarKey(2);
  27     private static final BarKey BAR_KEY_THREE = new BarKey(3);
  28     private static final BarKey BAR_KEY_FOUR = new BarKey(4);
  29     private static final Bar BAR_ONE = new BarBuilder().withKey(BAR_KEY_ONE).setName("one").build();
  30     private static final Bar BAR_TWO = new BarBuilder().withKey(BAR_KEY_TWO).setName("two").build();
  31     private static final Bar BAR_THREE = new BarBuilder().withKey(BAR_KEY_THREE).setName("three").build();
  32     private static final Bar BAR_FOUR = new BarBuilder().withKey(BAR_KEY_FOUR).setName("four").build();
  33     private static final List<Bar> BAR_LIST = List.of(BAR_TWO, BAR_ONE, BAR_FOUR, BAR_THREE);
  34
  35     @Test
  36     public void ofTest() {
  37         final Foo foo = new FooBuilder()
  38                 .setBar(BindingMap.of(BAR_ONE, BAR_TWO))
  39                 .build();
  40         final Map<BarKey, Bar> bar = foo.getBar();
  41         assertNotNull(bar);
  42         assertEquals(bar.get(BAR_KEY_ONE), BAR_ONE);
  43         assertEquals(bar.get(BAR_KEY_TWO), BAR_TWO);
  44     }
  45
  46     @Test
  47     public void builderTest() {
  48         final BindingMap.Builder<BarKey, Bar> builder = BindingMap.builder();
  49         for (Bar bar : BAR_LIST) {
  50             builder.add(bar);
  51         }
  52         final Foo foo = new FooBuilder()
  53                 .setBar(builder.build())
  54                 .build();
  55         final Map<BarKey, Bar> bar = foo.getBar();
  56         assertNotNull(bar);
  57         assertEquals(bar.get(BAR_KEY_ONE), BAR_ONE);
  58         assertEquals(bar.get(BAR_KEY_TWO), BAR_TWO);
  59         assertEquals(bar.get(BAR_KEY_THREE), BAR_THREE);
  60         assertEquals(bar.get(BAR_KEY_FOUR), BAR_FOUR);
  61     }
  62
  63     @Test
  64     public void orderedTest() {
  65         final Foo foo = new FooBuilder()
  66                 .setBar(BindingMap.ordered(BAR_TWO, BAR_ONE, BAR_FOUR, BAR_THREE))
  67                 .build();
  68         final Map<BarKey, Bar> bar = foo.getBar();
  69         assertNotNull(bar);
  70         checkOrderedMap(bar, BAR_LIST);
  71     }
  72
  73     @Test
  74     public void orderedBuilderTest() {
  75         final Foo foo = new FooBuilder()
  76                 .setBar(BindingMap.<BarKey, Bar>orderedBuilder()
  77                         .add(BAR_TWO)
  78                         .add(BAR_ONE)
  79                         .addAll(BAR_FOUR, BAR_THREE)
  80                         .build())
  81                 .build();
  82         final Map<BarKey, Bar> bar = foo.getBar();
  83         assertNotNull(bar);
  84         checkOrderedMap(bar, BAR_LIST);
  85     }
  86
  87     private static void checkOrderedMap(final Map<BarKey, Bar> barMap, final List<Bar> barList) {
  88         final Iterator<Map.Entry<BarKey, Bar>> mapIt = barMap.entrySet().iterator();
  89         final Iterator<Bar> listIt = barList.iterator();
  90         while (mapIt.hasNext() && listIt.hasNext()) {
  91             assertEquals(mapIt.next().getValue(), listIt.next());
  92         }
  93         assertEquals(mapIt.hasNext(), listIt.hasNext());
  94     }
  95 }