data/yang-data-api/src/test/java/org/opendaylight/yangtools/yang/data/api/schema/stream/YT1433Test.java

   1 /*
   2  * Copyright (c) 2022 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.yangtools.yang.data.api.schema.stream;
   9
  10 import static org.junit.jupiter.api.Assertions.assertEquals;
  11
  12 import java.io.IOException;
  13 import org.junit.jupiter.api.BeforeAll;
  14 import org.junit.jupiter.api.Test;
  15 import org.opendaylight.yangtools.yang.common.QName;
  16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  17 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  18 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
  19
  20 public class YT1433Test {
  21     private static EffectiveModelContext CONTEXT;
  22
  23     @BeforeAll
  24     public static void beforeAll() {
  25         CONTEXT = YangParserTestUtils.parseYangResourceDirectory("/YT1433");
  26     }
  27
  28     @Test
  29     public void testContainerAugmentContainer() throws IOException {
  30         final FormattingNormalizedNodeStreamWriter streamWriter = new FormattingNormalizedNodeStreamWriter();
  31
  32         final QName bar = QName.create("foo", "bar");
  33         final QName baz = QName.create("foo", "baz");
  34
  35         final YangInstanceIdentifier path = YangInstanceIdentifier.builder()
  36             .node(QName.create("foo", "foo"))
  37             .node(bar)
  38             .node(baz)
  39             .build();
  40
  41         try (var iidWriter = YangInstanceIdentifierWriter.open(streamWriter, CONTEXT, path)) {
  42             try (var nnWriter = new NormalizedNodeWriter(streamWriter)) {
  43                 // No-op
  44             }
  45         }
  46
  47         assertEquals("""
  48             (foo)foo(container)
  49               (foo)bar(container)
  50                 (foo)baz(container)
  51                 (end)
  52               (end)
  53             (end)
  54             """, streamWriter.result());
  55     }
  56
  57     @Test
  58     public void testChoiceAugmentCointainer() throws IOException {
  59         final FormattingNormalizedNodeStreamWriter streamWriter = new FormattingNormalizedNodeStreamWriter();
  60
  61         final QName bar = QName.create("bar", "bar");
  62         final QName baz = QName.create("bar", "baz");
  63
  64         final YangInstanceIdentifier path = YangInstanceIdentifier.builder()
  65             .node(QName.create("bar", "foo"))
  66             .node(bar)
  67             .node(baz)
  68             .build();
  69
  70         try (var iidWriter = YangInstanceIdentifierWriter.open(streamWriter, CONTEXT, path)) {
  71             try (var nnWriter = new NormalizedNodeWriter(streamWriter)) {
  72                 // No-op
  73             }
  74         }
  75
  76         assertEquals("""
  77             (bar)foo(choice)
  78               (bar)bar(container)
  79                 (bar)baz(container)
  80                 (end)
  81               (end)
  82             (end)
  83             """, streamWriter.result());
  84     }
  85 }