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