import java.nio.charset.StandardCharsets;
import java.util.Collection;
import java.util.Optional;
+import java.util.Set;
import javax.ws.rs.core.MediaType;
import org.junit.BeforeClass;
import org.junit.Test;
import org.opendaylight.yangtools.yang.common.Revision;
import org.opendaylight.yangtools.yang.common.XMLNamespace;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
final Module augmentModule = schemaContext.findModules(XMLNamespace.of("augment:module")).iterator().next();
final QName augmentChoice1QName = QName.create(augmentModule.getQNameModule(), "augment-choice1");
final QName augmentChoice2QName = QName.create(augmentChoice1QName, "augment-choice2");
- final QName containerQName = QName.create(augmentChoice1QName, "case-choice-case-container1");
- final YangInstanceIdentifier.AugmentationIdentifier augChoice1II =
- new YangInstanceIdentifier.AugmentationIdentifier(Sets.newHashSet(augmentChoice1QName));
- final YangInstanceIdentifier.AugmentationIdentifier augChoice2II =
- new YangInstanceIdentifier.AugmentationIdentifier(Sets.newHashSet(augmentChoice2QName));
- final YangInstanceIdentifier dataII = YangInstanceIdentifier.of(dataSchemaNode.getQName()).node(augChoice1II)
- .node(augmentChoice1QName).node(augChoice2II).node(augmentChoice2QName).node(containerQName);
+ final YangInstanceIdentifier dataII = YangInstanceIdentifier.of(dataSchemaNode.getQName())
+ .node(new AugmentationIdentifier(Set.of(augmentChoice1QName)))
+ .node(augmentChoice1QName)
+ // FIXME: DataSchemaContextTree bug? case children seem to ignore augments
+ // .node(new AugmentationIdentifier(Set.of(augmentChoice2QName)))
+ .node(augmentChoice2QName)
+ .node(QName.create(augmentChoice1QName, "case-choice-case-container1"));
final String uri = "instance-identifier-module:cont";
mockBodyReader(uri, xmlBodyReader, true);
final NormalizedNodePayload payload = xmlBodyReader.readFrom(null, null, null, mediaType, null,