Add simulated discard-changes to testtool default datastore
[netconf.git] / netconf / tools / netconf-testtool / src / main / java / org / opendaylight / netconf / test / tool / SimulatedOperationProvider.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;
10
11 import com.google.common.base.Optional;
12 import com.google.common.collect.Sets;
13 import java.io.File;
14 import java.util.Collections;
15 import java.util.Set;
16 import org.opendaylight.controller.config.util.capability.Capability;
17 import org.opendaylight.netconf.api.monitoring.CapabilityListener;
18 import org.opendaylight.netconf.impl.SessionIdProvider;
19 import org.opendaylight.netconf.mapping.api.NetconfOperation;
20 import org.opendaylight.netconf.mapping.api.NetconfOperationService;
21 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
22 import org.opendaylight.netconf.test.tool.rpc.DataList;
23 import org.opendaylight.netconf.test.tool.rpc.SimulatedCommit;
24 import org.opendaylight.netconf.test.tool.rpc.SimulatedCreateSubscription;
25 import org.opendaylight.netconf.test.tool.rpc.SimulatedDiscardChanges;
26 import org.opendaylight.netconf.test.tool.rpc.SimulatedEditConfig;
27 import org.opendaylight.netconf.test.tool.rpc.SimulatedGet;
28 import org.opendaylight.netconf.test.tool.rpc.SimulatedGetConfig;
29 import org.opendaylight.netconf.test.tool.rpc.SimulatedLock;
30 import org.opendaylight.netconf.test.tool.rpc.SimulatedUnLock;
31
32 class SimulatedOperationProvider implements NetconfOperationServiceFactory {
33     private final Set<Capability> caps;
34     private final SimulatedOperationService simulatedOperationService;
35
36     public SimulatedOperationProvider(final SessionIdProvider idProvider,
37                                       final Set<Capability> caps,
38                                       final Optional<File> notificationsFile,
39                                       final Optional<File> initialConfigXMLFile) {
40         this.caps = caps;
41         simulatedOperationService = new SimulatedOperationService(idProvider.getCurrentSessionId(), notificationsFile, initialConfigXMLFile);
42     }
43
44     @Override
45     public Set<Capability> getCapabilities() {
46         return caps;
47     }
48
49     @Override
50     public AutoCloseable registerCapabilityListener(
51             final CapabilityListener listener) {
52         listener.onCapabilitiesChanged(caps, Collections.<Capability>emptySet());
53         return new AutoCloseable() {
54             @Override
55             public void close() throws Exception {
56             }
57         };
58     }
59
60     @Override
61     public NetconfOperationService createService(
62             final String netconfSessionIdForReporting) {
63         return simulatedOperationService;
64     }
65
66     static class SimulatedOperationService implements NetconfOperationService {
67         private final long currentSessionId;
68         private final Optional<File> notificationsFile;
69         private final Optional<File> initialConfigXMLFile;
70
71         public SimulatedOperationService(final long currentSessionId, final Optional<File> notificationsFile, final Optional<File> initialConfigXMLFile) {
72             this.currentSessionId = currentSessionId;
73             this.notificationsFile = notificationsFile;
74             this.initialConfigXMLFile = initialConfigXMLFile;
75         }
76
77         @Override
78         public Set<NetconfOperation> getNetconfOperations() {
79             final DataList storage = new DataList();
80             final SimulatedGet sGet = new SimulatedGet(String.valueOf(currentSessionId), storage);
81             final SimulatedEditConfig sEditConfig = new SimulatedEditConfig(String.valueOf(currentSessionId), storage);
82             final SimulatedGetConfig sGetConfig = new SimulatedGetConfig(String.valueOf(currentSessionId), storage, initialConfigXMLFile );
83             final SimulatedCommit sCommit = new SimulatedCommit(String.valueOf(currentSessionId));
84             final SimulatedLock sLock = new SimulatedLock(String.valueOf(currentSessionId));
85             final SimulatedUnLock sUnlock = new SimulatedUnLock(String.valueOf(currentSessionId));
86             final SimulatedCreateSubscription sCreateSubs = new SimulatedCreateSubscription(
87                     String.valueOf(currentSessionId), notificationsFile);
88             final SimulatedDiscardChanges sDiscardChanges = new SimulatedDiscardChanges(String.valueOf(currentSessionId));
89             return Sets.<NetconfOperation>newHashSet(sGet, sGetConfig, sEditConfig, sCommit, sLock, sUnlock, sCreateSubs, sDiscardChanges);
90         }
91
92         @Override
93         public void close() {
94         }
95
96     }
97 }