diff options
author | Sagar Srivastav <sagsriva@cisco.com> | 2016-12-19 01:42:11 -0800 |
---|---|---|
committer | Sagar Srivastav <sagsriva@cisco.com> | 2017-02-06 22:45:26 -0800 |
commit | fd88a38ef0c95452953deb2c52a07eb1232a82a5 (patch) | |
tree | 6e52e8ee92669a1924e1a6ff9fa25095f3cf219f /ioam/impl | |
parent | ffc4005f85536f9d19379cab26a1d8dd9b3c4b10 (diff) |
ioam: export plugin - support for ioam-export plugin in hc2vpp
Change-Id: I2314a2de3b9c2a1d9c2e5950659e44cb9bdda985
Signed-off-by: Sagar Srivastav <sagsriva@cisco.com>
Diffstat (limited to 'ioam/impl')
8 files changed, 408 insertions, 18 deletions
diff --git a/ioam/impl/pom.xml b/ioam/impl/pom.xml index 0b058a716..e88a040f6 100644 --- a/ioam/impl/pom.xml +++ b/ioam/impl/pom.xml @@ -99,6 +99,11 @@ <version>${ioam.version}</version> </dependency> <dependency> + <groupId>io.fd.vpp</groupId> + <artifactId>jvpp-ioamexport</artifactId> + <version>${ioam.version}</version> + </dependency> + <dependency> <groupId>io.fd.hc2vpp.common</groupId> <artifactId>vpp-translate-utils</artifactId> <version>${project.version}</version> diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java index 204d003c9..228653c1c 100755 --- a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java +++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java @@ -20,12 +20,15 @@ import com.google.inject.AbstractModule; import com.google.inject.Provider; import com.google.inject.Singleton; import com.google.inject.multibindings.Multibinder; -import io.fd.hc2vpp.vppioam.impl.oper.VppIoamReaderFactory; -import io.fd.honeycomb.translate.read.ReaderFactory; -import io.fd.honeycomb.translate.write.WriterFactory; import io.fd.hc2vpp.vppioam.impl.config.VppIoamWriterFactory; +import io.fd.hc2vpp.vppioam.impl.oper.VppIoamReaderFactory; +import io.fd.hc2vpp.vppioam.impl.util.JVppIoamExportProvider; import io.fd.hc2vpp.vppioam.impl.util.JVppIoamPotProvider; import io.fd.hc2vpp.vppioam.impl.util.JVppIoamTraceProvider; +import io.fd.honeycomb.translate.read.ReaderFactory; +import io.fd.honeycomb.translate.write.WriterFactory; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexportFacade; import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampot; import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampotFacade; import io.fd.vpp.jvpp.ioamtrace.future.FutureJVppIoamtrace; @@ -41,16 +44,19 @@ public final class VppIoamModule extends AbstractModule { private static final Logger LOG = LoggerFactory.getLogger(VppIoamModule.class); private final Class<? extends Provider<FutureJVppIoamtraceFacade>> jvppIoamTraceProviderClass; private final Class<? extends Provider<FutureJVppIoampotFacade>> jvppIoamPotProviderClass; + private final Class<? extends Provider<FutureJVppIoamexportFacade>> jvppIoamExportProviderClass; public VppIoamModule() { - this(JVppIoamTraceProvider.class, JVppIoamPotProvider.class); + this(JVppIoamTraceProvider.class, JVppIoamPotProvider.class, JVppIoamExportProvider.class); } @VisibleForTesting VppIoamModule(Class<? extends Provider<FutureJVppIoamtraceFacade>> jvppIoamTraceProvider, - Class<? extends Provider<FutureJVppIoampotFacade>> jvppIoamPotProviderClass) { + Class<? extends Provider<FutureJVppIoampotFacade>> jvppIoamPotProviderClass, + Class<? extends Provider<FutureJVppIoamexportFacade>> jvppIoamExportProviderClass) { this.jvppIoamTraceProviderClass = jvppIoamTraceProvider; this.jvppIoamPotProviderClass = jvppIoamPotProviderClass; + this.jvppIoamExportProviderClass = jvppIoamExportProviderClass; } @Override @@ -60,6 +66,7 @@ public final class VppIoamModule extends AbstractModule { // Bind to Plugin's JVPP. bind(FutureJVppIoamtrace.class).toProvider(jvppIoamTraceProviderClass).in(Singleton.class); bind(FutureJVppIoampot.class).toProvider(jvppIoamPotProviderClass).in(Singleton.class); + bind(FutureJVppIoamexport.class).toProvider(jvppIoamExportProviderClass).in(Singleton.class); // Below are classes picked up by HC framework Multibinder.newSetBinder(binder(), WriterFactory.class).addBinding().to(VppIoamWriterFactory.class); diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizer.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizer.java new file mode 100644 index 000000000..2172750d4 --- /dev/null +++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizer.java @@ -0,0 +1,95 @@ +/* + * Copyright (c) 2016 Cisco and its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.hc2vpp.vppioam.impl.config; + +import io.fd.hc2vpp.common.translate.util.Ipv4Translator; +import io.fd.hc2vpp.common.translate.util.JvppReplyConsumer; +import io.fd.hc2vpp.vppioam.impl.util.FutureJVppIoamexportCustomizer; +import io.fd.honeycomb.translate.spi.write.WriterCustomizer; +import io.fd.honeycomb.translate.write.WriteContext; +import io.fd.honeycomb.translate.write.WriteFailedException; +import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisable; +import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisableReply; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport; +import javax.annotation.Nonnull; +import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExport; +import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +public class IoamExportWriterCustomizer extends FutureJVppIoamexportCustomizer + implements WriterCustomizer<IoamExport>, JvppReplyConsumer, Ipv4Translator { + + private static final Logger LOG = LoggerFactory.getLogger(IoamExportWriterCustomizer.class); + + public IoamExportWriterCustomizer(FutureJVppIoamexport jVppIoamexport){ + super(jVppIoamexport); + } + + @Override + public void writeCurrentAttributes(@Nonnull final InstanceIdentifier<IoamExport> instanceIdentifier, + @Nonnull final IoamExport ioamExport, @Nonnull final WriteContext writeContext) + throws WriteFailedException { + addExportProfile(ioamExport,instanceIdentifier); + LOG.info("Export profile {} created, id: {}", ioamExport, instanceIdentifier); + } + + @Override + public void updateCurrentAttributes(@Nonnull final InstanceIdentifier<IoamExport> instanceIdentifier, + @Nonnull final IoamExport dataBefore, @Nonnull final IoamExport dataAfter, + @Nonnull final WriteContext writeContext) throws WriteFailedException { + addExportProfile(dataAfter,instanceIdentifier); + LOG.info("Export profile {} updated , id: {}", dataAfter, instanceIdentifier); + } + + @Override + public void deleteCurrentAttributes(@Nonnull final InstanceIdentifier<IoamExport> instanceIdentifier, + @Nonnull final IoamExport ioamExport, @Nonnull final WriteContext writeContext) + throws WriteFailedException { + deleteExportProfile(ioamExport,instanceIdentifier); + LOG.info("Export profile {} deleted, id: {}", ioamExport, instanceIdentifier); + } + + private IoamExportIp6EnableDisableReply addExportProfile(IoamExport ioamExport, InstanceIdentifier<IoamExport> id) + throws WriteFailedException { + IoamExportIp6EnableDisable request = new IoamExportIp6EnableDisable(); + if (ioamExport.isDisable()==null) + request.isDisable = 1; + else + request.isDisable = (byte)(ioamExport.isDisable()?1:0); + request.srcAddress = ipv4AddressNoZoneToArray(ioamExport.getSourceAddress().getValue()); + request.collectorAddress = ipv4AddressNoZoneToArray(ioamExport.getCollectorAddress().getValue()); + return getReplyForCreate(getFutureJVppIoamexport() + .ioamExportIp6EnableDisable(request) + .toCompletableFuture(),id,ioamExport); + + } + + private IoamExportIp6EnableDisableReply deleteExportProfile(IoamExport ioamExport, + InstanceIdentifier<IoamExport> id) + throws WriteFailedException { + IoamExportIp6EnableDisable request = new IoamExportIp6EnableDisable(); + request.isDisable = 1; //disable when deleted + request.srcAddress = ipv4AddressNoZoneToArray(ioamExport.getSourceAddress().getValue()); + request.collectorAddress = ipv4AddressNoZoneToArray(ioamExport.getCollectorAddress().getValue()); + + return getReplyForDelete(getFutureJVppIoamexport() + .ioamExportIp6EnableDisable(request) + .toCompletableFuture(),id); + } + +} diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java index 15d312ff7..ba105eb39 100755 --- a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java +++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java @@ -19,11 +19,14 @@ package io.fd.hc2vpp.vppioam.impl.config; import com.google.common.collect.Sets; import com.google.inject.Inject; import io.fd.honeycomb.translate.impl.write.GenericListWriter; +import io.fd.honeycomb.translate.impl.write.GenericWriter; import io.fd.honeycomb.translate.write.WriterFactory; import io.fd.honeycomb.translate.write.registry.ModifiableWriterRegistryBuilder; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport; import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampot; import io.fd.vpp.jvpp.ioamtrace.future.FutureJVppIoamtrace; import javax.annotation.Nonnull; +import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExport; import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.trace.rev160512.IoamTraceConfig; import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.trace.rev160512.ioam.trace.config.TraceConfig; import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.trace.rev160512.ioam.trace.config.trace.config.NodeInterfaces; @@ -38,12 +41,16 @@ public class VppIoamWriterFactory implements WriterFactory { private final FutureJVppIoamtrace jVppIoamtrace; @Nonnull private final FutureJVppIoampot jVppIoampot; + @Nonnull + private final FutureJVppIoamexport jVppIoamexport; @Inject public VppIoamWriterFactory(@Nonnull final FutureJVppIoamtrace jVppIoamtrace, - @Nonnull final FutureJVppIoampot jVppIoampot) { + @Nonnull final FutureJVppIoampot jVppIoampot, + @Nonnull final FutureJVppIoamexport jVppIoamexport) { this.jVppIoamtrace = jVppIoamtrace; this.jVppIoampot = jVppIoampot; + this.jVppIoamexport = jVppIoamexport; } @Override @@ -60,6 +67,10 @@ public class VppIoamWriterFactory implements WriterFactory { registry.subtreeAdd(Sets.newHashSet(InstanceIdentifier.create(PotProfileSet.class) .child(PotProfileList.class)), new GenericListWriter<>(potId, new IoamPotWriterCustomizer(jVppIoampot))); + //Export Config + final InstanceIdentifier<IoamExport> exportId = + InstanceIdentifier.create(IoamExport.class); + registry.add(new GenericWriter<>(exportId,new IoamExportWriterCustomizer(jVppIoamexport))); } } diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/FutureJVppIoamexportCustomizer.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/FutureJVppIoamexportCustomizer.java new file mode 100644 index 000000000..adc2ffc10 --- /dev/null +++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/FutureJVppIoamexportCustomizer.java @@ -0,0 +1,41 @@ +/* + * Copyright (c) 2016 Cisco and its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.hc2vpp.vppioam.impl.util; + +import com.google.common.base.Preconditions; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport; +import javax.annotation.Nonnull; + +public abstract class FutureJVppIoamexportCustomizer { + + private final FutureJVppIoamexport futureJVppIoamexport; + + public FutureJVppIoamexportCustomizer (@Nonnull final FutureJVppIoamexport futureJVppIoamexport) { + this.futureJVppIoamexport = Preconditions.checkNotNull(futureJVppIoamexport, + "futureJVppIoamexport should not be null"); + } + + /** + * Get Ioam POT Api reference + * + * @return Ioam POT Api reference + */ + public FutureJVppIoamexport getFutureJVppIoamexport() { + return futureJVppIoamexport; + } + +} diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/JVppIoamExportProvider.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/JVppIoamExportProvider.java new file mode 100644 index 000000000..fa4d20004 --- /dev/null +++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/JVppIoamExportProvider.java @@ -0,0 +1,55 @@ +/* + * Copyright (c) 2016 Cisco and its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.hc2vpp.vppioam.impl.util; + +import com.google.inject.Inject; +import io.fd.honeycomb.infra.distro.ProviderTrait; +import io.fd.vpp.jvpp.JVppRegistry; +import io.fd.vpp.jvpp.ioamexport.JVppIoamexportImpl; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexportFacade; +import java.io.IOException; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +public class JVppIoamExportProvider extends ProviderTrait<FutureJVppIoamexportFacade> { + + private static final Logger LOG = LoggerFactory.getLogger(JVppIoamExportProvider.class); + + @Inject + private JVppRegistry registry; + + @Override + protected FutureJVppIoamexportFacade create() { + try { + final JVppIoamexportImpl jVppIoamexport = new JVppIoamexportImpl(); + // Free jvpp-ioam-export plugin's resources on shutdown + Runtime.getRuntime().addShutdownHook(new Thread() { + @Override + public void run() { + LOG.info("Unloading jvpp-ioam-export plugin"); + jVppIoamexport.close(); + LOG.info("Successfully unloaded jvpp-ioam-export plugin"); + } + }); + + LOG.info("Successfully loaded jvpp-ioam-export plugin"); + return new FutureJVppIoamexportFacade(registry, jVppIoamexport); + } catch (IOException e) { + throw new IllegalStateException("Unable to open VPP management connection", e); + } + } +} diff --git a/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java index c11ee2dc4..45b673218 100644 --- a/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java +++ b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java @@ -16,6 +16,14 @@ package io.fd.hc2vpp.vppioam.impl; +import static org.hamcrest.CoreMatchers.is; +import static org.hamcrest.CoreMatchers.not; +import static org.hamcrest.Matchers.empty; +import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertThat; +import static org.mockito.Mockito.mock; +import static org.mockito.MockitoAnnotations.initMocks; + import com.google.inject.Guice; import com.google.inject.Inject; import com.google.inject.Provider; @@ -27,24 +35,16 @@ import io.fd.honeycomb.translate.impl.write.registry.FlatWriterRegistryBuilder; import io.fd.honeycomb.translate.read.ReaderFactory; import io.fd.honeycomb.translate.write.WriterFactory; import io.fd.vpp.jvpp.JVppRegistry; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexportFacade; import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampotFacade; import io.fd.vpp.jvpp.ioamtrace.future.FutureJVppIoamtraceFacade; +import java.util.HashSet; +import java.util.Set; import org.junit.Before; import org.junit.Test; import org.mockito.Mock; import org.opendaylight.controller.md.sal.binding.api.DataBroker; -import java.util.HashSet; -import java.util.Set; - -import static org.hamcrest.CoreMatchers.is; -import static org.hamcrest.CoreMatchers.not; -import static org.hamcrest.Matchers.empty; -import static org.junit.Assert.assertNotNull; -import static org.junit.Assert.assertThat; -import static org.mockito.Mockito.mock; -import static org.mockito.MockitoAnnotations.initMocks; - public class VppIoamModuleTest { @@ -73,7 +73,10 @@ public class VppIoamModuleTest { initMocks(this); - Guice.createInjector(new VppIoamModule(MockJVppIoamTraceProvider.class,MockJVppIoamPotProvider.class), BoundFieldModule.of(this)).injectMembers(this); + Guice.createInjector(new VppIoamModule(MockJVppIoamTraceProvider.class, + MockJVppIoamPotProvider.class, + MockJVppIoamExportProvider.class), + BoundFieldModule.of(this)).injectMembers(this); } @Test @@ -109,5 +112,13 @@ public class VppIoamModuleTest { return mock(FutureJVppIoampotFacade.class); } } + + private static final class MockJVppIoamExportProvider implements Provider<FutureJVppIoamexportFacade> { + + @Override + public FutureJVppIoamexportFacade get() { + return mock(FutureJVppIoamexportFacade.class); + } + } } diff --git a/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizerTest.java b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizerTest.java new file mode 100644 index 000000000..b2dfaf308 --- /dev/null +++ b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizerTest.java @@ -0,0 +1,165 @@ +/* + * Copyright (c) 2016 Cisco and/or its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.hc2vpp.vppioam.impl.config; + +import static org.junit.Assert.fail; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.Mockito.doReturn; +import static org.mockito.Mockito.verify; + +import io.fd.hc2vpp.common.test.write.WriterCustomizerTest; +import io.fd.hc2vpp.common.translate.util.Ipv4Translator; +import io.fd.honeycomb.translate.write.WriteFailedException; +import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisable; +import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisableReply; +import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport; +import org.junit.Test; +import org.mockito.Mock; +import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExport; +import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExportBuilder; +import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address; +import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + + +public class IoamExportWriterCustomizerTest extends WriterCustomizerTest + implements Ipv4Translator{ + + @Mock + FutureJVppIoamexport jVppIoamexport; + + private IoamExportWriterCustomizer customizer; + + private static final Logger LOG = LoggerFactory.getLogger(IoamExportWriterCustomizerTest.class); + + @Override + public void setUpTest() throws Exception { + customizer = new IoamExportWriterCustomizer(jVppIoamexport); + } + + private static IoamExport generateExportProfile(){ + IoamExportBuilder builder = new IoamExportBuilder(); + builder.setDisable(false); + builder.setSourceAddress(new Ipv4Address("127.0.0.1")); + builder.setCollectorAddress(new Ipv4Address("127.0.0.2")); + return builder.build(); + } + + private void whenExportAddThenSuccess() { + final IoamExportIp6EnableDisableReply reply = new IoamExportIp6EnableDisableReply(); + reply.context = 1; + doReturn(future(reply)).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class)); + } + + private void whenExportAddThenFailure() { + doReturn(failedFuture()).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class)); + } + + private void whenExportDelThenSuccess() { + final IoamExportIp6EnableDisableReply reply = new IoamExportIp6EnableDisableReply(); + reply.context = 1; + doReturn(future(reply)).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class)); + } + + private void whenExportDelThenFailure() { + doReturn(failedFuture()).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class)); + } + + private IoamExportIp6EnableDisable generateIoamExportIp6EnableDisable(boolean disable) { + IoamExportIp6EnableDisable request = new IoamExportIp6EnableDisable(); + request.isDisable = (byte)(disable?1:0); + request.srcAddress = ipv4AddressNoZoneToArray("127.0.0.1"); + request.collectorAddress = ipv4AddressNoZoneToArray("127.0.0.2"); + + return request; + } + + @Test + public void testCreate() throws Exception { + final IoamExport ioamExport = generateExportProfile(); + final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class); + + whenExportAddThenSuccess(); + + customizer.writeCurrentAttributes(id, ioamExport, writeContext); + + verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(false)); + } + + @Test + public void testCreateFailed() throws Exception { + final IoamExport ioamExport = generateExportProfile(); + final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class); + + whenExportAddThenFailure(); + + try { + customizer.writeCurrentAttributes(id, ioamExport, writeContext); + } catch (WriteFailedException e) { + verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(false)); + + return; + } + fail("WriteFailedException.CreateFailedException was expected"); + } + + @Test + public void testDelete() throws Exception { + final IoamExport ioamExport = generateExportProfile(); + final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class); + + whenExportDelThenSuccess(); + + customizer.deleteCurrentAttributes(id, ioamExport, writeContext); + + verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(true)); + } + + @Test + public void testDeleteFailed() throws Exception { + final IoamExport ioamExport = generateExportProfile(); + final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class); + + whenExportDelThenFailure(); + + try { + customizer.deleteCurrentAttributes(id, ioamExport, writeContext); + } catch (WriteFailedException e) { + verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(true)); + + return; + } + fail("WriteFailedException.CreateFailedException was expected"); + } + + @Test + public void testCreateWithMissingDisabledField() throws Exception { + IoamExportBuilder builder = new IoamExportBuilder(); + builder.setSourceAddress(new Ipv4Address("127.0.0.1")); + builder.setCollectorAddress(new Ipv4Address("127.0.0.2")); + final IoamExport ioamExport = builder.build(); + final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class); + + whenExportAddThenSuccess(); + + customizer.writeCurrentAttributes(id, ioamExport, writeContext); + + verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(true)); + } + +} |