tests/honeynode/2.2.1/honeynode-plugin-impl/src/main/java/io/fd/honeycomb/transportpce/device/read/DummyMonitoringService.java

   1 /*
   2  * Copyright (c) 2018 Orange and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package io.fd.honeycomb.transportpce.device.read;
  17
  18 import java.util.Collections;
  19 import java.util.Set;
  20
  21 import javax.annotation.Nonnull;
  22 import javax.annotation.Nullable;
  23
  24 import org.opendaylight.netconf.api.capability.Capability;
  25 import org.opendaylight.netconf.api.monitoring.NetconfManagementSession;
  26 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
  27 import org.opendaylight.netconf.api.monitoring.SessionEvent;
  28 import org.opendaylight.netconf.api.monitoring.SessionListener;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SessionsBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location.Enumeration;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaKey;
  42
  43 import com.google.common.base.Function;
  44 import com.google.common.collect.ArrayListMultimap;
  45 import com.google.common.collect.Collections2;
  46 import com.google.common.collect.Lists;
  47 import com.google.common.collect.Sets;
  48
  49 public class DummyMonitoringService implements NetconfMonitoringService {
  50
  51     private static final Sessions EMPTY_SESSIONS = new SessionsBuilder().setSession(Collections.emptyList()).build();
  52     private static final Function<Capability, Uri> CAPABILITY_URI_FUNCTION =
  53         capability -> new Uri(capability.getCapabilityUri());
  54
  55     private static final Function<Capability, Schema> CAPABILITY_SCHEMA_FUNCTION = new Function<Capability, Schema>() {
  56         @Nullable
  57         @Override
  58         public Schema apply(@Nonnull final Capability capability) {
  59             return new SchemaBuilder()
  60                     .setIdentifier(capability.getModuleName().get())
  61                     .setNamespace(new Uri(capability.getModuleNamespace().get()))
  62                     .setFormat(Yang.class)
  63                     .setVersion(capability.getRevision().get())
  64                     .setLocation(Collections.singletonList(new Location(Enumeration.NETCONF)))
  65                     .withKey(new SchemaKey(Yang.class, capability.getModuleName().get(),
  66                         capability.getRevision().get()))
  67                     .build();
  68         }
  69     };
  70
  71     private final Capabilities capabilities;
  72     private final ArrayListMultimap<String, Capability> capabilityMultiMap;
  73     private final Schemas schemas;
  74
  75     public DummyMonitoringService(final Set<Capability> capabilities) {
  76
  77         this.capabilities = new CapabilitiesBuilder().setCapability(
  78                 Lists.newArrayList(Collections2.transform(capabilities, CAPABILITY_URI_FUNCTION))).build();
  79
  80         Set<Capability> moduleCapabilities = Sets.newHashSet();
  81         this.capabilityMultiMap = ArrayListMultimap.create();
  82         for (Capability cap : capabilities) {
  83             if (cap.getModuleName().isPresent()) {
  84                 capabilityMultiMap.put(cap.getModuleName().get(), cap);
  85                 moduleCapabilities.add(cap);
  86             }
  87         }
  88
  89         this.schemas = new SchemasBuilder().setSchema(
  90             Lists.newArrayList(Collections2.transform(moduleCapabilities, CAPABILITY_SCHEMA_FUNCTION))).build();
  91     }
  92
  93     @Override
  94     public Sessions getSessions() {
  95         return EMPTY_SESSIONS;
  96     }
  97
  98     @Override
  99     public SessionListener getSessionListener() {
 100         return new SessionListener() {
 101             @Override
 102             public void onSessionUp(final NetconfManagementSession session) {
 103                 //no op
 104             }
 105
 106             @Override
 107             public void onSessionDown(final NetconfManagementSession session) {
 108                 //no op
 109             }
 110
 111             @Override
 112             public void onSessionEvent(final SessionEvent event) {
 113                 //no op
 114             }
 115         };
 116     }
 117
 118     @Override
 119     public Schemas getSchemas() {
 120         return schemas;
 121     }
 122
 123
 124     @Override
 125     public Capabilities getCapabilities() {
 126         return capabilities;
 127     }
 128
 129     @Override
 130     public AutoCloseable registerCapabilitiesListener(final CapabilitiesListener listener) {
 131         return null;
 132     }
 133
 134     @Override
 135     public AutoCloseable registerSessionsListener(final SessionsListener listener) {
 136         return null;
 137     }
 138
 139 @Override
 140 public String getSchemaForCapability(String moduleName, java.util.Optional<String> revision) {
 141    for (Capability capability : capabilityMultiMap.get(moduleName)) {
 142             if (capability.getRevision().get().equals(revision.get())) {
 143                 return capability.getCapabilitySchema().get();
 144             }
 145         }
 146         throw new IllegalArgumentException(
 147             "Module with name: " + moduleName + " and revision: " + revision + " does not exist");
 148 }
 149
 150 }