fd8c49838d26ae162deeeb23a507e9c52be33f74
[netconf.git] / netconf / tools / netconf-testtool / src / main / java / org / opendaylight / netconf / test / tool / rpc / SimulatedEditConfig.java
1 /*
2  * Copyright (c) 2014 Cisco Systems, Inc. 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
9 package org.opendaylight.netconf.test.tool.rpc;
10
11 import com.google.common.base.Optional;
12 import org.opendaylight.controller.config.util.xml.DocumentedException;
13 import org.opendaylight.controller.config.util.xml.XmlElement;
14 import org.opendaylight.controller.config.util.xml.XmlUtil;
15 import org.opendaylight.netconf.api.xml.XmlNetconfConstants;
16 import org.opendaylight.netconf.confignetconfconnector.operations.AbstractConfigNetconfOperation;
17 import org.opendaylight.netconf.confignetconfconnector.operations.editconfig.EditConfigXmlParser;
18 import org.w3c.dom.Attr;
19 import org.w3c.dom.Document;
20 import org.w3c.dom.Element;
21
22 public class SimulatedEditConfig extends AbstractConfigNetconfOperation {
23     private static final String DELETE_EDIT_CONFIG = "delete";
24     private static final String OPERATION = "operation";
25     private static final String REMOVE_EDIT_CONFIG = "remove";
26     private final DataList storage;
27
28     public SimulatedEditConfig(final String netconfSessionIdForReporting, final DataList storage) {
29         super(null, netconfSessionIdForReporting);
30         this.storage = storage;
31     }
32
33     @Override
34     protected Element handleWithNoSubsequentOperations(final Document document, final XmlElement operationElement)
35             throws DocumentedException {
36         final XmlElement configElementData = operationElement.getOnlyChildElement(XmlNetconfConstants.CONFIG_KEY);
37
38         containsDelete(configElementData);
39         if (containsDelete(configElementData)) {
40             storage.resetConfigList();
41         } else {
42             storage.setConfigList(configElementData.getChildElements());
43         }
44
45         return XmlUtil.createElement(document, XmlNetconfConstants.OK, Optional.<String>absent());
46     }
47
48     @Override
49     protected String getOperationName() {
50         return EditConfigXmlParser.EDIT_CONFIG;
51     }
52
53     private boolean containsDelete(final XmlElement element) {
54         for (final Attr o : element.getAttributes().values()) {
55             if (o.getLocalName().equals(OPERATION)
56                     && (o.getValue().equals(DELETE_EDIT_CONFIG) || o.getValue()
57                             .equals(REMOVE_EDIT_CONFIG))) {
58                 return true;
59             }
60
61         }
62
63         for (final XmlElement xmlElement : element.getChildElements()) {
64             if (containsDelete(xmlElement)) {
65                 return true;
66             }
67
68         }
69
70         return false;
71     }
72 }