2 * Copyright (c) 2015 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.netconf.test.tool;
10 import static com.google.common.base.Preconditions.checkState;
12 import com.google.common.base.Function;
13 import com.google.common.collect.ArrayListMultimap;
14 import com.google.common.collect.Collections2;
15 import com.google.common.collect.Maps;
16 import java.util.ArrayList;
17 import java.util.Collections;
18 import java.util.HashSet;
19 import java.util.List;
20 import java.util.Optional;
22 import org.opendaylight.netconf.api.capability.Capability;
23 import org.opendaylight.netconf.api.monitoring.NetconfManagementSession;
24 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
25 import org.opendaylight.netconf.api.monitoring.SessionEvent;
26 import org.opendaylight.netconf.api.monitoring.SessionListener;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SessionsBuilder;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location;
37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location.Enumeration;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaBuilder;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaKey;
41 public class DummyMonitoringService implements NetconfMonitoringService {
43 private static final Sessions EMPTY_SESSIONS = new SessionsBuilder().build();
44 private static final Function<Capability, Uri> CAPABILITY_URI_FUNCTION =
45 capability -> new Uri(capability.getCapabilityUri());
47 private static final Function<Capability, Schema> CAPABILITY_SCHEMA_FUNCTION = capability -> new SchemaBuilder()
48 .setIdentifier(capability.getModuleName().get())
49 .setNamespace(new Uri(capability.getModuleNamespace().get()))
50 .setFormat(Yang.class)
51 .setVersion(capability.getRevision().orElse(""))
52 .setLocation(Collections.singletonList(new Location(Enumeration.NETCONF)))
53 .withKey(new SchemaKey(Yang.class, capability.getModuleName().get(),
54 capability.getRevision().orElse("")))
57 private final Capabilities capabilities;
58 private final ArrayListMultimap<String, Capability> capabilityMultiMap;
59 private final Schemas schemas;
61 public DummyMonitoringService(final Set<Capability> capabilities) {
63 this.capabilities = new CapabilitiesBuilder().setCapability(
64 new ArrayList<>(Collections2.transform(capabilities, CAPABILITY_URI_FUNCTION))).build();
66 Set<Capability> moduleCapabilities = new HashSet<>();
67 this.capabilityMultiMap = ArrayListMultimap.create();
68 for (Capability cap : capabilities) {
69 if (cap.getModuleName().isPresent()) {
70 capabilityMultiMap.put(cap.getModuleName().get(), cap);
71 moduleCapabilities.add(cap);
75 this.schemas = new SchemasBuilder()
76 .setSchema(Maps.uniqueIndex(Collections2.transform(moduleCapabilities, CAPABILITY_SCHEMA_FUNCTION),
82 public Sessions getSessions() {
83 return EMPTY_SESSIONS;
87 public SessionListener getSessionListener() {
88 return new SessionListener() {
90 public void onSessionUp(final NetconfManagementSession session) {
95 public void onSessionDown(final NetconfManagementSession session) {
100 public void onSessionEvent(final SessionEvent event) {
107 public Schemas getSchemas() {
112 public String getSchemaForCapability(final String moduleName, final Optional<String> revision) {
113 final List<Capability> capabilityList = capabilityMultiMap.get(moduleName);
114 if (revision.isPresent()) {
115 for (Capability capability : capabilityList) {
116 if (capability.getRevision().orElseThrow().equals(revision.get())) {
117 return capability.getCapabilitySchema().orElseThrow();
121 checkState(capabilityList.size() == 1,
122 "Expected 1 capability for module %s, available revisions : %s", moduleName, capabilityList);
123 //Only one revision is present, so return it
124 return capabilityList.iterator().next().getCapabilitySchema().orElseThrow();
127 throw new IllegalArgumentException(
128 "Module with name: " + moduleName + " and revision: " + revision + " does not exist");
132 public Capabilities getCapabilities() {
137 public AutoCloseable registerCapabilitiesListener(final CapabilitiesListener listener) {
142 public AutoCloseable registerSessionsListener(final SessionsListener listener) {