restconf/restconf-nb/src/test/java/org/opendaylight/restconf/nb/rfc8040/jersey/providers/XmlNormalizedNodeBodyWriterTest.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.restconf.nb.rfc8040.jersey.providers;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.mockito.Mockito.mock;
  12
  13 import java.io.ByteArrayOutputStream;
  14 import java.io.IOException;
  15 import java.nio.charset.StandardCharsets;
  16 import javax.ws.rs.core.MediaType;
  17 import org.junit.Test;
  18 import org.junit.runner.RunWith;
  19 import org.mockito.junit.MockitoJUnitRunner;
  20 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
  21 import org.opendaylight.restconf.nb.rfc8040.TestRestconfUtils;
  22 import org.opendaylight.restconf.nb.rfc8040.legacy.NormalizedNodePayload;
  23 import org.opendaylight.yangtools.yang.common.QName;
  24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  25 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  26 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  27 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  28
  29 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  30 public class XmlNormalizedNodeBodyWriterTest {
  31     @Test
  32     public void testWriteEmptyRootContainer() throws IOException {
  33         final EffectiveModelContext schemaContext = mock(EffectiveModelContext.class);
  34
  35         final NormalizedNodePayload nodePayload = NormalizedNodePayload.of(
  36             InstanceIdentifierContext.ofLocalRoot(schemaContext),
  37             Builders.containerBuilder().withNodeIdentifier(new NodeIdentifier(SchemaContext.NAME)).build());
  38
  39         final ByteArrayOutputStream output = new ByteArrayOutputStream();
  40         final XmlNormalizedNodeBodyWriter xmlWriter = new XmlNormalizedNodeBodyWriter();
  41         xmlWriter.writeTo(nodePayload, null, null, null, MediaType.APPLICATION_XML_TYPE, null, output);
  42
  43         assertEquals("<data xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\"></data>",
  44             output.toString(StandardCharsets.UTF_8));
  45     }
  46
  47     @Test
  48     public void testRootContainerWrite() throws IOException {
  49         final EffectiveModelContext schemaContext =
  50                 TestRestconfUtils.loadSchemaContext("/instanceidentifier/yang", null);
  51
  52         final NormalizedNodePayload nodePayload = NormalizedNodePayload.of(
  53             InstanceIdentifierContext.ofLocalRoot(schemaContext),
  54             Builders.containerBuilder()
  55                 .withNodeIdentifier(new NodeIdentifier(SchemaContext.NAME))
  56                 .withChild(Builders.containerBuilder()
  57                     .withNodeIdentifier(new NodeIdentifier(
  58                         QName.create("foo:module", "2016-09-29", "foo-bar-container")))
  59                     .build())
  60                 .withChild(Builders.containerBuilder()
  61                     .withNodeIdentifier(new NodeIdentifier(
  62                         QName.create("bar:module", "2016-09-29", "foo-bar-container")))
  63                     .build())
  64                 .build());
  65
  66         final ByteArrayOutputStream output = new ByteArrayOutputStream();
  67         final XmlNormalizedNodeBodyWriter xmlWriter = new XmlNormalizedNodeBodyWriter();
  68         xmlWriter.writeTo(nodePayload, null, null, null, MediaType.APPLICATION_XML_TYPE, null, output);
  69
  70         assertEquals("<data xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">"
  71             + "<foo-bar-container xmlns=\"bar:module\"></foo-bar-container>"
  72             + "<foo-bar-container xmlns=\"foo:module\"></foo-bar-container>"
  73             + "</data>", output.toString(StandardCharsets.UTF_8));
  74     }
  75 }