*/
package org.opendaylight.netconf.test.tool;
+import static com.google.common.base.Preconditions.checkState;
+
import com.google.common.base.Function;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Collections2;
-import com.google.common.collect.Lists;
-import com.google.common.collect.Sets;
-import java.util.Collections;
+import com.google.common.collect.ImmutableSet;
+import com.google.common.collect.Maps;
+import java.util.HashSet;
+import java.util.List;
import java.util.Optional;
import java.util.Set;
-import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
import org.opendaylight.netconf.api.capability.Capability;
import org.opendaylight.netconf.api.monitoring.NetconfManagementSession;
import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
public class DummyMonitoringService implements NetconfMonitoringService {
- private static final Sessions EMPTY_SESSIONS = new SessionsBuilder().setSession(Collections.emptyList()).build();
+ private static final Sessions EMPTY_SESSIONS = new SessionsBuilder().build();
private static final Function<Capability, Uri> CAPABILITY_URI_FUNCTION =
capability -> new Uri(capability.getCapabilityUri());
- private static final Function<Capability, Schema> CAPABILITY_SCHEMA_FUNCTION = new Function<Capability, Schema>() {
- @Nullable
- @Override
- public Schema apply(@Nonnull final Capability capability) {
- return new SchemaBuilder()
- .setIdentifier(capability.getModuleName().get())
- .setNamespace(new Uri(capability.getModuleNamespace().get()))
- .setFormat(Yang.class)
- .setVersion(capability.getRevision().or(""))
- .setLocation(Collections.singletonList(new Location(Enumeration.NETCONF)))
- .withKey(new SchemaKey(Yang.class, capability.getModuleName().get(),
- capability.getRevision().or("")))
- .build();
- }
- };
+ private static final Function<Capability, Schema> CAPABILITY_SCHEMA_FUNCTION = capability -> new SchemaBuilder()
+ .setIdentifier(capability.getModuleName().get())
+ .setNamespace(new Uri(capability.getModuleNamespace().get()))
+ .setFormat(Yang.VALUE)
+ .setVersion(capability.getRevision().orElse(""))
+ .setLocation(Set.of(new Location(Enumeration.NETCONF)))
+ .withKey(new SchemaKey(Yang.VALUE, capability.getModuleName().get(),
+ capability.getRevision().orElse("")))
+ .build();
private final Capabilities capabilities;
private final ArrayListMultimap<String, Capability> capabilityMultiMap;
public DummyMonitoringService(final Set<Capability> capabilities) {
this.capabilities = new CapabilitiesBuilder().setCapability(
- Lists.newArrayList(Collections2.transform(capabilities, CAPABILITY_URI_FUNCTION))).build();
+ ImmutableSet.copyOf(Collections2.transform(capabilities, CAPABILITY_URI_FUNCTION))).build();
- Set<Capability> moduleCapabilities = Sets.newHashSet();
- this.capabilityMultiMap = ArrayListMultimap.create();
+ Set<Capability> moduleCapabilities = new HashSet<>();
+ capabilityMultiMap = ArrayListMultimap.create();
for (Capability cap : capabilities) {
if (cap.getModuleName().isPresent()) {
capabilityMultiMap.put(cap.getModuleName().get(), cap);
}
}
- this.schemas = new SchemasBuilder().setSchema(
- Lists.newArrayList(Collections2.transform(moduleCapabilities, CAPABILITY_SCHEMA_FUNCTION))).build();
+ schemas = new SchemasBuilder()
+ .setSchema(Maps.uniqueIndex(Collections2.transform(moduleCapabilities, CAPABILITY_SCHEMA_FUNCTION),
+ Schema::key))
+ .build();
}
@Override
@Override
public String getSchemaForCapability(final String moduleName, final Optional<String> revision) {
-
- for (Capability capability : capabilityMultiMap.get(moduleName)) {
- if (capability.getRevision().get().equals(revision.get())) {
- return capability.getCapabilitySchema().get();
+ final List<Capability> capabilityList = capabilityMultiMap.get(moduleName);
+ if (revision.isPresent()) {
+ for (Capability capability : capabilityList) {
+ if (capability.getRevision().orElseThrow().equals(revision.get())) {
+ return capability.getCapabilitySchema().orElseThrow();
+ }
}
+ } else {
+ checkState(capabilityList.size() == 1,
+ "Expected 1 capability for module %s, available revisions : %s", moduleName, capabilityList);
+ //Only one revision is present, so return it
+ return capabilityList.iterator().next().getCapabilitySchema().orElseThrow();
}
+
throw new IllegalArgumentException(
"Module with name: " + moduleName + " and revision: " + revision + " does not exist");
}