2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.yang.data.impl.leafref.context;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertTrue;
14 import org.apache.log4j.BasicConfigurator;
15 import org.junit.AfterClass;
16 import org.junit.BeforeClass;
17 import org.junit.Test;
18 import org.opendaylight.yangtools.yang.common.QName;
19 import org.opendaylight.yangtools.yang.common.QNameModule;
20 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
23 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
24 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
25 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
26 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
27 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTree;
28 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
29 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeConfiguration;
30 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
31 import org.opendaylight.yangtools.yang.data.impl.leafref.LeafRefContext;
32 import org.opendaylight.yangtools.yang.data.impl.leafref.LeafRefDataValidationFailedException;
33 import org.opendaylight.yangtools.yang.data.impl.leafref.LeafRefValidation;
34 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
35 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
36 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.CollectionNodeBuilder;
37 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
38 import org.opendaylight.yangtools.yang.data.impl.schema.tree.InMemoryDataTreeFactory;
39 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
40 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
41 import org.opendaylight.yangtools.yang.model.api.Module;
42 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
43 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
44 import org.slf4j.Logger;
45 import org.slf4j.LoggerFactory;
47 public class DataTreeCandidateValidatorTest2 {
49 private static SchemaContext context;
50 private static Module mainModule;
51 private static QNameModule rootModuleQname;
52 private static LeafRefContext rootLeafRefContext;
53 public static DataTree inMemoryDataTree;
55 private static QName chips;
56 private static QName chip;
57 private static QName devType;
58 private static QName chipDesc;
60 private static QName devices;
61 private static QName device;
62 private static QName typeChoice;
63 private static QName typeText;
64 private static QName devDesc;
65 private static QName sn;
66 private static QName defaultIp;
68 private static QName deviceTypeStr;
69 private static QName deviceType;
70 private static QName type;
71 private static QName desc;
73 private static final Logger LOG = LoggerFactory.getLogger(DataTreeCandidateValidatorTest2.class);
74 private static final String NEW_LINE = System.getProperty("line.separator");
77 BasicConfigurator.configure();
81 public static void init() {
82 context = YangParserTestUtils.parseYangResourceDirectory("/leafref-validation");
84 final Set<Module> modules = context.getModules();
85 for (final Module module : modules) {
86 if (module.getName().equals("leafref-validation2")) {
91 rootModuleQname = mainModule.getQNameModule();
92 rootLeafRefContext = LeafRefContext.create(context);
94 chips = QName.create(rootModuleQname, "chips");
95 chip = QName.create(rootModuleQname, "chip");
96 devType = QName.create(rootModuleQname, "dev_type");
97 chipDesc = QName.create(rootModuleQname, "chip_desc");
99 devices = QName.create(rootModuleQname, "devices");
100 device = QName.create(rootModuleQname, "device");
101 typeText = QName.create(rootModuleQname, "type_text");
102 devDesc = QName.create(rootModuleQname, "dev_desc");
103 sn = QName.create(rootModuleQname, "sn");
104 defaultIp = QName.create(rootModuleQname, "default_ip");
106 deviceTypeStr = QName.create(rootModuleQname, "device_types");
107 deviceType = QName.create(rootModuleQname, "device_type");
108 type = QName.create(rootModuleQname, "type");
109 desc = QName.create(rootModuleQname, "desc");
111 inMemoryDataTree = new InMemoryDataTreeFactory().create(DataTreeConfiguration.DEFAULT_OPERATIONAL, context);
112 final DataTreeModification initialDataTreeModification = inMemoryDataTree.takeSnapshot().newModification();
113 final ContainerSchemaNode chipsListContSchemaNode = (ContainerSchemaNode) mainModule.findDataChildByName(chips)
115 final ContainerNode chipsContainer = createChipsContainer(chipsListContSchemaNode);
116 final YangInstanceIdentifier path1 = YangInstanceIdentifier.of(chips);
117 initialDataTreeModification.write(path1, chipsContainer);
119 final ContainerSchemaNode devTypesListContSchemaNode = (ContainerSchemaNode) mainModule
120 .findDataChildByName(deviceTypeStr).get();
121 final ContainerNode deviceTypesContainer = createDevTypeStrContainer(devTypesListContSchemaNode);
122 final YangInstanceIdentifier path2 = YangInstanceIdentifier.of(deviceTypeStr);
123 initialDataTreeModification.write(path2, deviceTypesContainer);
125 initialDataTreeModification.ready();
126 final DataTreeCandidate writeChipsCandidate = inMemoryDataTree.prepare(initialDataTreeModification);
128 inMemoryDataTree.commit(writeChipsCandidate);
129 LOG.debug("{}", inMemoryDataTree);
133 public static void cleanup() {
134 inMemoryDataTree = null;
135 rootLeafRefContext = null;
141 public void dataTreeCanditateValidationTest2() {
146 private static void writeDevices() {
148 final ContainerSchemaNode devicesContSchemaNode = (ContainerSchemaNode) mainModule.findDataChildByName(devices)
151 final ContainerNode devicesContainer = createDevicesContainer(devicesContSchemaNode);
153 final YangInstanceIdentifier devicesPath = YangInstanceIdentifier.of(devices);
154 final DataTreeModification writeModification = inMemoryDataTree.takeSnapshot().newModification();
155 writeModification.write(devicesPath, devicesContainer);
157 writeModification.ready();
158 final DataTreeCandidate writeDevicesCandidate = inMemoryDataTree.prepare(writeModification);
160 LOG.debug("*************************");
161 LOG.debug("Before writeDevices: ");
162 LOG.debug("*************************");
163 LOG.debug("{}", inMemoryDataTree);
165 boolean exception = false;
167 LeafRefValidation.validate(writeDevicesCandidate, rootLeafRefContext);
168 } catch (final LeafRefDataValidationFailedException e) {
169 LOG.debug("All validation errors:{}{}", NEW_LINE, e.getMessage());
171 assertEquals(4, e.getValidationsErrorsCount());
175 assertTrue(exception);
177 inMemoryDataTree.commit(writeDevicesCandidate);
179 LOG.debug("*************************");
180 LOG.debug("After write: ");
181 LOG.debug("*************************");
182 LOG.debug("{}", inMemoryDataTree);
185 private static ContainerNode createDevTypeStrContainer(final ContainerSchemaNode container) {
187 final ListSchemaNode devTypeListSchemaNode = (ListSchemaNode) container.findDataChildByName(deviceType).get();
189 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> devTypeContainerBldr = Builders
190 .containerBuilder(container);
192 final MapNode devTypeMap = createDevTypeList(devTypeListSchemaNode);
193 devTypeContainerBldr.addChild(devTypeMap);
195 return devTypeContainerBldr.build();
198 private static MapNode createDevTypeList(final ListSchemaNode devTypeListSchemaNode) {
200 final CollectionNodeBuilder<MapEntryNode, MapNode> devTypeMapBldr = Builders.mapBuilder(devTypeListSchemaNode);
202 devTypeMapBldr.addChild(createDevTypeListEntry("dev_type_1", "typedesc1", devTypeListSchemaNode));
203 devTypeMapBldr.addChild(createDevTypeListEntry("dev_type_2", "typedesc2", devTypeListSchemaNode));
204 devTypeMapBldr.addChild(createDevTypeListEntry("dev_type_3", "typedesc3", devTypeListSchemaNode));
206 return devTypeMapBldr.build();
209 private static MapEntryNode createDevTypeListEntry(final String typeVal, final String descVal,
210 final ListSchemaNode devTypeListSchemaNode) {
212 final LeafNode<String> typeLeaf = ImmutableNodes.leafNode(type, typeVal);
213 final LeafNode<String> descLeaf = ImmutableNodes.leafNode(desc, descVal);
215 final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> devTypeMapEntryBldr = Builders
216 .mapEntryBuilder(devTypeListSchemaNode);
218 devTypeMapEntryBldr.addChild(typeLeaf);
219 devTypeMapEntryBldr.addChild(descLeaf);
221 return devTypeMapEntryBldr.build();
224 private static ContainerNode createChipsContainer(final ContainerSchemaNode container) {
226 final ListSchemaNode chipsListSchemaNode = (ListSchemaNode) container.findDataChildByName(chip).get();
228 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> chipsContainerBldr = Builders
229 .containerBuilder(container);
231 final MapNode chipsMap = createChipsList(chipsListSchemaNode);
232 chipsContainerBldr.addChild(chipsMap);
234 return chipsContainerBldr.build();
237 private static MapNode createChipsList(final ListSchemaNode chipsListSchemaNode) {
239 final CollectionNodeBuilder<MapEntryNode, MapNode> chipsMapBldr = Builders.mapBuilder(chipsListSchemaNode);
241 chipsMapBldr.addChild(createChipsListEntry("dev_type_1", "desc1", chipsListSchemaNode));
242 chipsMapBldr.addChild(createChipsListEntry("dev_type_2", "desc2", chipsListSchemaNode));
244 return chipsMapBldr.build();
247 private static MapEntryNode createChipsListEntry(final String devTypeVal, final String chipDescVal,
248 final ListSchemaNode chipsListSchemaNode) {
250 final LeafNode<String> devTypeLeaf = ImmutableNodes.leafNode(devType, devTypeVal);
251 final LeafNode<String> chipDescLeaf = ImmutableNodes.leafNode(chipDesc, chipDescVal);
253 final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> chipsMapEntryBldr = Builders
254 .mapEntryBuilder(chipsListSchemaNode);
256 chipsMapEntryBldr.addChild(devTypeLeaf);
257 chipsMapEntryBldr.addChild(chipDescLeaf);
259 return chipsMapEntryBldr.build();
262 private static ContainerNode createDevicesContainer(final ContainerSchemaNode container) {
264 final ListSchemaNode devicesListSchemaNode = (ListSchemaNode) container.findDataChildByName(device).get();
266 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> devicesContainerBldr = Builders
267 .containerBuilder(container);
269 final MapNode devicesMap = createDeviceList(devicesListSchemaNode);
270 devicesContainerBldr.addChild(devicesMap);
272 return devicesContainerBldr.build();
275 private static MapNode createDeviceList(final ListSchemaNode deviceListSchemaNode) {
277 final CollectionNodeBuilder<MapEntryNode, MapNode> devicesMapBldr = Builders.mapBuilder(deviceListSchemaNode);
279 devicesMapBldr.addChild(createDeviceListEntry("dev_type_1", "typedesc1", 123456, "192.168.0.1",
280 deviceListSchemaNode));
281 devicesMapBldr.addChild(createDeviceListEntry("dev_type_2", "typedesc2", 123457, "192.168.0.1",
282 deviceListSchemaNode));
283 devicesMapBldr.addChild(createDeviceListEntry("dev_type_2", "typedesc3", 123457, "192.168.0.1",
284 deviceListSchemaNode));
285 devicesMapBldr.addChild(createDeviceListEntry("dev_type_1", "typedesc2", 123458, "192.168.0.1",
286 deviceListSchemaNode));
288 .addChild(createDeviceListEntry("unknown", "unknown", 123457, "192.168.0.1", deviceListSchemaNode));
290 return devicesMapBldr.build();
293 private static MapEntryNode createDeviceListEntry(final String typeTextVal, final String descVal, final int snVal,
294 final String defaultIpVal, final ListSchemaNode devicesListSchemaNode) {
296 final LeafNode<String> typeTextLeaf = ImmutableNodes.leafNode(typeText, typeTextVal);
297 final LeafNode<String> descLeaf = ImmutableNodes.leafNode(devDesc, descVal);
298 final LeafNode<Integer> snValLeaf = ImmutableNodes.leafNode(sn, snVal);
299 final LeafNode<String> defaultIpLeaf = ImmutableNodes.leafNode(defaultIp, defaultIpVal);
301 final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> devicesMapEntryBldr = Builders
302 .mapEntryBuilder(devicesListSchemaNode);
304 devicesMapEntryBldr.addChild(typeTextLeaf);
305 devicesMapEntryBldr.addChild(descLeaf);
306 devicesMapEntryBldr.addChild(snValLeaf);
307 devicesMapEntryBldr.addChild(defaultIpLeaf);
309 return devicesMapEntryBldr.build();