yang/yang-common/src/test/java/org/opendaylight/yangtools/yang/common/BiMapYangNamespaceContextTest.java

   1 /*
   2  * Copyright (c) 2019 Pantheon Technologies, s.r.o.  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.yangtools.yang.common;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertFalse;
  12 import static org.junit.Assert.assertTrue;
  13
  14 import com.google.common.collect.ImmutableBiMap;
  15 import java.io.ByteArrayInputStream;
  16 import java.io.ByteArrayOutputStream;
  17 import java.io.DataInputStream;
  18 import java.io.DataOutputStream;
  19 import java.io.IOException;
  20 import java.net.URI;
  21 import java.util.Optional;
  22 import org.junit.Test;
  23
  24 public class BiMapYangNamespaceContextTest {
  25     private static final QNameModule FOO = QNameModule.create(URI.create("foo"));
  26     private static final QNameModule BAR = QNameModule.create(URI.create("bar"));
  27     private static final QNameModule BAZ = QNameModule.create(URI.create("baz"));
  28
  29     private final BiMapYangNamespaceContext context = new BiMapYangNamespaceContext(
  30         ImmutableBiMap.of("foo", FOO, "bar", BAR));
  31
  32     @Test
  33     public void testEquals() {
  34         assertTrue(context.equals(context));
  35         assertTrue(context.equals(new BiMapYangNamespaceContext(ImmutableBiMap.of("foo", FOO, "bar", BAR))));
  36         assertFalse(context.equals(null));
  37         assertFalse(context.equals(new BiMapYangNamespaceContext(ImmutableBiMap.of("foo", FOO))));
  38         assertFalse(context.equals(new BiMapYangNamespaceContext(ImmutableBiMap.of("bar", BAR))));
  39     }
  40
  41     @Test
  42     public void testPrefixForNamespace() {
  43         assertEquals(Optional.of("foo"), context.findPrefixForNamespace(FOO));
  44         assertEquals(Optional.of("bar"), context.findPrefixForNamespace(BAR));
  45         assertEquals(Optional.empty(), context.findPrefixForNamespace(BAZ));
  46     }
  47
  48     @Test
  49     public void testNamespaceForPrefix() {
  50         assertEquals(Optional.of(FOO), context.findNamespaceForPrefix("foo"));
  51         assertEquals(Optional.of(BAR), context.findNamespaceForPrefix("bar"));
  52         assertEquals(Optional.empty(), context.findNamespaceForPrefix("baz"));
  53     }
  54
  55     @Test
  56     public void testReadWrite() throws IOException {
  57         final byte[] bytes;
  58         try (ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
  59             try (DataOutputStream dos = new DataOutputStream(bos)) {
  60                 context.writeTo(dos);
  61             }
  62             bytes = bos.toByteArray();
  63         }
  64
  65         final BiMapYangNamespaceContext other;
  66         try (DataInputStream dis = new DataInputStream(new ByteArrayInputStream(bytes))) {
  67             other = BiMapYangNamespaceContext.readFrom(dis);
  68         }
  69
  70         assertEquals(context, other);
  71     }
  72
  73     @Test
  74     public void testCreateQName() {
  75         assertEquals(QName.create(FOO, "some"), context.createQName("foo", "some"));
  76     }
  77 }