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