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