Convert to using requireNonNull()
[netconf.git] / netconf / tools / netconf-testtool / src / main / java / org / opendaylight / netconf / test / tool / DummyMonitoringService.java
1 /*
2  * Copyright (c) 2015 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 package org.opendaylight.netconf.test.tool;
9
10 import com.google.common.base.Function;
11 import com.google.common.collect.ArrayListMultimap;
12 import com.google.common.collect.Collections2;
13 import java.util.ArrayList;
14 import java.util.Collections;
15 import java.util.HashSet;
16 import java.util.Optional;
17 import java.util.Set;
18 import org.opendaylight.netconf.api.capability.Capability;
19 import org.opendaylight.netconf.api.monitoring.NetconfManagementSession;
20 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
21 import org.opendaylight.netconf.api.monitoring.SessionEvent;
22 import org.opendaylight.netconf.api.monitoring.SessionListener;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SessionsBuilder;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location.Enumeration;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaBuilder;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaKey;
36
37 public class DummyMonitoringService implements NetconfMonitoringService {
38
39     private static final Sessions EMPTY_SESSIONS = new SessionsBuilder().setSession(Collections.emptyList()).build();
40     private static final Function<Capability, Uri> CAPABILITY_URI_FUNCTION =
41         capability -> new Uri(capability.getCapabilityUri());
42
43     private static final Function<Capability, Schema> CAPABILITY_SCHEMA_FUNCTION = capability -> new SchemaBuilder()
44             .setIdentifier(capability.getModuleName().get())
45             .setNamespace(new Uri(capability.getModuleNamespace().get()))
46             .setFormat(Yang.class)
47             .setVersion(capability.getRevision().orElse(""))
48             .setLocation(Collections.singletonList(new Location(Enumeration.NETCONF)))
49             .withKey(new SchemaKey(Yang.class, capability.getModuleName().get(),
50                 capability.getRevision().orElse("")))
51             .build();
52
53     private final Capabilities capabilities;
54     private final ArrayListMultimap<String, Capability> capabilityMultiMap;
55     private final Schemas schemas;
56
57     public DummyMonitoringService(final Set<Capability> capabilities) {
58
59         this.capabilities = new CapabilitiesBuilder().setCapability(
60                 new ArrayList<>(Collections2.transform(capabilities, CAPABILITY_URI_FUNCTION))).build();
61
62         Set<Capability> moduleCapabilities = new HashSet<>();
63         this.capabilityMultiMap = ArrayListMultimap.create();
64         for (Capability cap : capabilities) {
65             if (cap.getModuleName().isPresent()) {
66                 capabilityMultiMap.put(cap.getModuleName().get(), cap);
67                 moduleCapabilities.add(cap);
68             }
69         }
70
71         this.schemas = new SchemasBuilder().setSchema(
72             new ArrayList<>(Collections2.transform(moduleCapabilities, CAPABILITY_SCHEMA_FUNCTION))).build();
73     }
74
75     @Override
76     public Sessions getSessions() {
77         return EMPTY_SESSIONS;
78     }
79
80     @Override
81     public SessionListener getSessionListener() {
82         return new SessionListener() {
83             @Override
84             public void onSessionUp(final NetconfManagementSession session) {
85                 //no op
86             }
87
88             @Override
89             public void onSessionDown(final NetconfManagementSession session) {
90                 //no op
91             }
92
93             @Override
94             public void onSessionEvent(final SessionEvent event) {
95                 //no op
96             }
97         };
98     }
99
100     @Override
101     public Schemas getSchemas() {
102         return schemas;
103     }
104
105     @Override
106     public String getSchemaForCapability(final String moduleName, final Optional<String> revision) {
107
108         for (Capability capability : capabilityMultiMap.get(moduleName)) {
109             if (capability.getRevision().get().equals(revision.get())) {
110                 return capability.getCapabilitySchema().get();
111             }
112         }
113         throw new IllegalArgumentException(
114             "Module with name: " + moduleName + " and revision: " + revision + " does not exist");
115     }
116
117     @Override
118     public Capabilities getCapabilities() {
119         return capabilities;
120     }
121
122     @Override
123     public AutoCloseable registerCapabilitiesListener(final CapabilitiesListener listener) {
124         return null;
125     }
126
127     @Override
128     public AutoCloseable registerSessionsListener(final SessionsListener listener) {
129         return null;
130     }
131
132 }